271 lines
8.2 KiB
Python
271 lines
8.2 KiB
Python
import copy
|
|
import json
|
|
import uuid
|
|
import time
|
|
import urlparse
|
|
|
|
import mox
|
|
import requests
|
|
import testtools
|
|
|
|
from keystoneclient.v3 import client
|
|
|
|
|
|
def parameterize(ref):
|
|
"""Rewrites attributes to match the kwarg naming convention in client.
|
|
|
|
>>> paramterize({'project_id': 0})
|
|
{'project': 0}
|
|
|
|
"""
|
|
params = ref.copy()
|
|
for key in ref:
|
|
if key[-3:] == '_id':
|
|
params.setdefault(key[:-3], params.pop(key))
|
|
return params
|
|
|
|
|
|
class TestClient(client.Client):
|
|
|
|
def serialize(self, entity):
|
|
return json.dumps(entity, sort_keys=True)
|
|
|
|
|
|
class TestCase(testtools.TestCase):
|
|
TEST_TENANT_NAME = 'aTenant'
|
|
TEST_TOKEN = 'aToken'
|
|
TEST_USER = 'test'
|
|
TEST_ROOT_URL = 'http://127.0.0.1:5000/'
|
|
TEST_URL = '%s%s' % (TEST_ROOT_URL, 'v3')
|
|
TEST_ROOT_ADMIN_URL = 'http://127.0.0.1:35357/'
|
|
TEST_ADMIN_URL = '%s%s' % (TEST_ROOT_ADMIN_URL, 'v3')
|
|
TEST_REQUEST_BASE = {
|
|
'config': {'danger_mode': False},
|
|
'verify': True,
|
|
}
|
|
|
|
def setUp(self):
|
|
super(TestCase, self).setUp()
|
|
self.mox = mox.Mox()
|
|
self._original_time = time.time
|
|
time.time = lambda: 1234
|
|
requests.request = self.mox.CreateMockAnything()
|
|
self.client = TestClient(username=self.TEST_USER,
|
|
token=self.TEST_TOKEN,
|
|
tenant_name=self.TEST_TENANT_NAME,
|
|
auth_url=self.TEST_URL,
|
|
endpoint=self.TEST_URL)
|
|
|
|
def tearDown(self):
|
|
time.time = self._original_time
|
|
self.mox.UnsetStubs()
|
|
self.mox.VerifyAll()
|
|
super(TestCase, self).tearDown()
|
|
|
|
|
|
class UnauthenticatedTestCase(testtools.TestCase):
|
|
""" Class used as base for unauthenticated calls """
|
|
TEST_ROOT_URL = 'http://127.0.0.1:5000/'
|
|
TEST_URL = '%s%s' % (TEST_ROOT_URL, 'v3')
|
|
TEST_ROOT_ADMIN_URL = 'http://127.0.0.1:35357/'
|
|
TEST_ADMIN_URL = '%s%s' % (TEST_ROOT_ADMIN_URL, 'v3')
|
|
TEST_REQUEST_BASE = {
|
|
'config': {'danger_mode': False},
|
|
'verify': True,
|
|
}
|
|
|
|
def setUp(self):
|
|
super(UnauthenticatedTestCase, self).setUp()
|
|
self.mox = mox.Mox()
|
|
self._original_time = time.time
|
|
time.time = lambda: 1234
|
|
requests.request = self.mox.CreateMockAnything()
|
|
|
|
def tearDown(self):
|
|
time.time = self._original_time
|
|
self.mox.UnsetStubs()
|
|
self.mox.VerifyAll()
|
|
super(UnauthenticatedTestCase, self).tearDown()
|
|
|
|
|
|
class CrudTests(testtools.TestCase):
|
|
key = None
|
|
collection_key = None
|
|
model = None
|
|
manager = None
|
|
|
|
def new_ref(self, **kwargs):
|
|
kwargs.setdefault('id', uuid.uuid4().hex)
|
|
return kwargs
|
|
|
|
def additionalSetUp(self):
|
|
self.headers = {
|
|
'GET': {
|
|
'X-Auth-Token': 'aToken',
|
|
'User-Agent': 'python-keystoneclient',
|
|
}
|
|
}
|
|
|
|
self.headers['HEAD'] = self.headers['GET'].copy()
|
|
self.headers['DELETE'] = self.headers['GET'].copy()
|
|
self.headers['PUT'] = self.headers['GET'].copy()
|
|
self.headers['POST'] = self.headers['GET'].copy()
|
|
self.headers['POST']['Content-Type'] = 'application/json'
|
|
self.headers['PATCH'] = self.headers['POST'].copy()
|
|
|
|
def serialize(self, entity):
|
|
if isinstance(entity, dict):
|
|
return json.dumps({self.key: entity}, sort_keys=True)
|
|
if isinstance(entity, list):
|
|
return json.dumps({self.collection_key: entity}, sort_keys=True)
|
|
raise NotImplementedError('Are you sure you want to serialize that?')
|
|
|
|
def test_create(self, ref=None):
|
|
ref = ref or self.new_ref()
|
|
resp = TestResponse({
|
|
"status_code": 201,
|
|
"text": self.serialize(ref),
|
|
})
|
|
|
|
method = 'POST'
|
|
req_ref = ref.copy()
|
|
req_ref.pop('id')
|
|
kwargs = copy.copy(self.TEST_REQUEST_BASE)
|
|
kwargs['headers'] = self.headers[method]
|
|
kwargs['data'] = self.serialize(req_ref)
|
|
requests.request(
|
|
method,
|
|
urlparse.urljoin(
|
|
self.TEST_URL,
|
|
'v3/%s' % self.collection_key),
|
|
**kwargs).AndReturn((resp))
|
|
self.mox.ReplayAll()
|
|
|
|
returned = self.manager.create(**parameterize(req_ref))
|
|
self.assertTrue(isinstance(returned, self.model))
|
|
for attr in ref:
|
|
self.assertEqual(
|
|
getattr(returned, attr),
|
|
ref[attr],
|
|
'Expected different %s' % attr)
|
|
|
|
def test_get(self, ref=None):
|
|
ref = ref or self.new_ref()
|
|
resp = TestResponse({
|
|
"status_code": 200,
|
|
"text": self.serialize(ref),
|
|
})
|
|
|
|
method = 'GET'
|
|
kwargs = copy.copy(self.TEST_REQUEST_BASE)
|
|
kwargs['headers'] = self.headers[method]
|
|
requests.request(
|
|
method,
|
|
urlparse.urljoin(
|
|
self.TEST_URL,
|
|
'v3/%s/%s' % (self.collection_key, ref['id'])),
|
|
**kwargs).AndReturn((resp))
|
|
self.mox.ReplayAll()
|
|
|
|
returned = self.manager.get(ref['id'])
|
|
self.assertTrue(isinstance(returned, self.model))
|
|
for attr in ref:
|
|
self.assertEqual(
|
|
getattr(returned, attr),
|
|
ref[attr],
|
|
'Expected different %s' % attr)
|
|
|
|
def test_list(self, ref_list=None, expected_path=None, **filter_kwargs):
|
|
ref_list = ref_list or [self.new_ref(), self.new_ref()]
|
|
resp = TestResponse({
|
|
"status_code": 200,
|
|
"text": self.serialize(ref_list),
|
|
})
|
|
|
|
method = 'GET'
|
|
kwargs = copy.copy(self.TEST_REQUEST_BASE)
|
|
kwargs['headers'] = self.headers[method]
|
|
requests.request(
|
|
method,
|
|
urlparse.urljoin(
|
|
self.TEST_URL,
|
|
expected_path or 'v3/%s' % self.collection_key),
|
|
**kwargs).AndReturn((resp))
|
|
self.mox.ReplayAll()
|
|
|
|
returned_list = self.manager.list(**filter_kwargs)
|
|
self.assertTrue(len(returned_list))
|
|
[self.assertTrue(isinstance(r, self.model)) for r in returned_list]
|
|
|
|
def test_update(self, ref=None):
|
|
ref = ref or self.new_ref()
|
|
req_ref = ref.copy()
|
|
del req_ref['id']
|
|
resp = TestResponse({
|
|
"status_code": 200,
|
|
"text": self.serialize(ref),
|
|
})
|
|
|
|
method = 'PATCH'
|
|
kwargs = copy.copy(self.TEST_REQUEST_BASE)
|
|
kwargs['headers'] = self.headers[method]
|
|
kwargs['data'] = self.serialize(req_ref)
|
|
requests.request(
|
|
method,
|
|
urlparse.urljoin(
|
|
self.TEST_URL,
|
|
'v3/%s/%s' % (self.collection_key, ref['id'])),
|
|
**kwargs).AndReturn((resp))
|
|
self.mox.ReplayAll()
|
|
|
|
returned = self.manager.update(ref['id'], **parameterize(req_ref))
|
|
self.assertTrue(isinstance(returned, self.model))
|
|
for attr in ref:
|
|
self.assertEqual(
|
|
getattr(returned, attr),
|
|
ref[attr],
|
|
'Expected different %s' % attr)
|
|
|
|
def test_delete(self, ref=None):
|
|
ref = ref or self.new_ref()
|
|
resp = TestResponse({
|
|
"status_code": 204,
|
|
"text": '',
|
|
})
|
|
|
|
method = 'DELETE'
|
|
kwargs = copy.copy(self.TEST_REQUEST_BASE)
|
|
kwargs['headers'] = self.headers[method]
|
|
requests.request(
|
|
method,
|
|
urlparse.urljoin(
|
|
self.TEST_URL,
|
|
'v3/%s/%s' % (self.collection_key, ref['id'])),
|
|
**kwargs).AndReturn((resp))
|
|
self.mox.ReplayAll()
|
|
|
|
self.manager.delete(ref['id'])
|
|
|
|
|
|
class TestResponse(requests.Response):
|
|
""" Class used to wrap requests.Response and provide some
|
|
convenience to initialize with a dict """
|
|
|
|
def __init__(self, data):
|
|
self._text = None
|
|
super(TestResponse, self)
|
|
if isinstance(data, dict):
|
|
self.status_code = data.get('status_code', None)
|
|
self.headers = data.get('headers', None)
|
|
# Fake the text attribute to streamline Response creation
|
|
self._text = data.get('text', None)
|
|
else:
|
|
self.status_code = data
|
|
|
|
def __eq__(self, other):
|
|
return self.__dict__ == other.__dict__
|
|
|
|
@property
|
|
def text(self):
|
|
return self._text
|