48 |
48 |
self.client = network.NetworkRestClient(self.url, self.token)
|
49 |
49 |
|
50 |
50 |
def tearDown(self):
|
51 |
|
pass
|
|
51 |
del self.client
|
52 |
52 |
|
53 |
53 |
def _assert(self, method_call, path, set_param=None, params=(), **kwargs):
|
54 |
54 |
"""Assert the REST method call is called as expected"""
|
... | ... | |
260 |
260 |
data=dumps(json_data), **kwargs)
|
261 |
261 |
|
262 |
262 |
|
|
263 |
class FakeObject(object):
|
|
264 |
|
|
265 |
json = None
|
|
266 |
|
|
267 |
|
|
268 |
class NetworkClient(TestCase):
|
|
269 |
|
|
270 |
"""Set up a ComputesRest thorough test"""
|
|
271 |
def setUp(self):
|
|
272 |
self.url = 'http://network.example.com'
|
|
273 |
self.token = 'n2tw0rk70k3n'
|
|
274 |
self.client = network.NetworkClient(self.url, self.token)
|
|
275 |
|
|
276 |
def tearDown(self):
|
|
277 |
del self.client
|
|
278 |
|
|
279 |
@patch(
|
|
280 |
'kamaki.clients.network.NetworkClient.networks_get',
|
|
281 |
return_value=FakeObject())
|
|
282 |
def test_list_networks(self, networks_get):
|
|
283 |
FakeObject.json = dict(networks='ret val')
|
|
284 |
self.assertEqual(self.client.list_networks(), 'ret val')
|
|
285 |
networks_get.assert_called_once_with(success=200)
|
|
286 |
|
|
287 |
@patch(
|
|
288 |
'kamaki.clients.network.NetworkClient.networks_post',
|
|
289 |
return_value=FakeObject())
|
|
290 |
def test_create_network(self, networks_post):
|
|
291 |
FakeObject.json = dict(network='ret val')
|
|
292 |
req = dict()
|
|
293 |
for kwargs in (dict(shared=None), dict(shared=True)):
|
|
294 |
for body_params in product(
|
|
295 |
(('name', None, ''), ('name', 'some name', 'some name')),
|
|
296 |
(
|
|
297 |
('admin_state_up', None, False),
|
|
298 |
('admin_state_up', True, True))):
|
|
299 |
fullargs = dict(kwargs)
|
|
300 |
for k, v, exp in body_params:
|
|
301 |
fullargs[k] = v
|
|
302 |
req[k] = exp
|
|
303 |
self.assertEqual(
|
|
304 |
self.client.create_network(**fullargs), 'ret val')
|
|
305 |
expargs = dict(kwargs)
|
|
306 |
expargs = dict(json_data=dict(network=req), success=201)
|
|
307 |
expargs.update(kwargs)
|
|
308 |
self.assertEqual(networks_post.mock_calls[-1], call(**expargs))
|
|
309 |
|
|
310 |
|
263 |
311 |
if __name__ == '__main__':
|
264 |
312 |
from sys import argv
|
265 |
313 |
from kamaki.clients.test import runTestCase
|
266 |
314 |
not_found = True
|
267 |
|
#if not argv[1:] or argv[1] == 'NetworkClient':
|
268 |
|
# not_found = False
|
269 |
|
# runTestCase(NetworkClient, 'Network Client', argv[2:])
|
|
315 |
if not argv[1:] or argv[1] == 'NetworkClient':
|
|
316 |
not_found = False
|
|
317 |
runTestCase(NetworkClient, 'Network Client', argv[2:])
|
270 |
318 |
if not argv[1:] or argv[1] == 'NetworkRest':
|
271 |
319 |
not_found = False
|
272 |
320 |
runTestCase(NetworkRestClient, 'NetworkRest Client', argv[2:])
|