Merge "Reuse class Manager from common code"

3 files changed, 32 insertions, 1 deletions

diff --git a/tests/test_base.py b/tests/test_base.py
index fb51ae6..4a97de8 100644
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -16,7 +16,7 @@
 
 import testtools
 
-from glanceclient.common import base
+from glanceclient.openstack.common.apiclient import base
 
 
 class TestBase(testtools.TestCase):
diff --git a/tests/utils.py b/tests/utils.py
index b7ca3af..2eb5f02 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -44,6 +44,32 @@ class FakeAPI(object):
         fixture = self._request(*args, **kwargs)
         return FakeResponse(fixture[0]), fixture[1]
 
+    def client_request(self, method, url, **kwargs):
+        if 'json' in kwargs and 'body' not in kwargs:
+            kwargs['body'] = kwargs.pop('json')
+        resp, body = self.json_request(method, url, **kwargs)
+        resp.json = lambda: body
+        resp.content = bool(body)
+        return resp
+
+    def head(self, url, **kwargs):
+        return self.client_request("HEAD", url, **kwargs)
+
+    def get(self, url, **kwargs):
+        return self.client_request("GET", url, **kwargs)
+
+    def post(self, url, **kwargs):
+        return self.client_request("POST", url, **kwargs)
+
+    def put(self, url, **kwargs):
+        return self.client_request("PUT", url, **kwargs)
+
+    def delete(self, url, **kwargs):
+        return self.raw_request("DELETE", url, **kwargs)
+
+    def patch(self, url, **kwargs):
+        return self.client_request("PATCH", url, **kwargs)
+
 
 class FakeResponse(object):
     def __init__(self, headers, body=None,
diff --git a/tests/v1/test_images.py b/tests/v1/test_images.py
index 63df002..6c0fde1 100644
--- a/tests/v1/test_images.py
+++ b/tests/v1/test_images.py
@@ -838,6 +838,7 @@ class ImageTest(testtools.TestCase):
         image.delete()
         expect = [
             ('HEAD', '/v1/images/1', {}, None),
+            ('HEAD', '/v1/images/1', {}, None),
             ('DELETE', '/v1/images/1', {}, None),
         ]
         self.assertEqual(expect, self.api.calls)
@@ -847,6 +848,7 @@ class ImageTest(testtools.TestCase):
         image.update(name='image-5')
         expect = [
             ('HEAD', '/v1/images/1', {}, None),
+            ('HEAD', '/v1/images/1', {}, None),
             ('PUT', '/v1/images/1', {'x-image-meta-name': 'image-5'}, None),
         ]
         self.assertEqual(expect, self.api.calls)
@@ -856,6 +858,7 @@ class ImageTest(testtools.TestCase):
         data = ''.join([b for b in image.data()])
         expect = [
             ('HEAD', '/v1/images/1', {}, None),
+            ('HEAD', '/v1/images/1', {}, None),
             ('GET', '/v1/images/1', {}, None),
         ]
         self.assertEqual(expect, self.api.calls)
@@ -871,6 +874,7 @@ class ImageTest(testtools.TestCase):
         data = ''.join([b for b in image.data(do_checksum=False)])
         expect = [
             ('HEAD', '/v1/images/2', {}, None),
+            ('HEAD', '/v1/images/2', {}, None),
             ('GET', '/v1/images/2', {}, None),
         ]
         self.assertEqual(expect, self.api.calls)
@@ -892,6 +896,7 @@ class ImageTest(testtools.TestCase):
         data = ''.join([b for b in image.data(do_checksum=False)])
         expect = [
             ('HEAD', '/v1/images/3', {}, None),
+            ('HEAD', '/v1/images/3', {}, None),
             ('GET', '/v1/images/3', {}, None),
         ]
         self.assertEqual(expect, self.api.calls)