Revision 3c50df2e kamaki/clients/cyclades/test.py
b/kamaki/clients/cyclades/test.py | ||
---|---|---|
30 | 30 |
# documentation are those of the authors and should not be |
31 | 31 |
# interpreted as representing official policies, either expressed |
32 | 32 |
# or implied, of GRNET S.A. |
33 |
|
|
33 | 34 |
from mock import patch, call |
34 | 35 |
from unittest import TestCase |
35 | 36 |
from itertools import product |
36 | 37 |
|
37 |
from kamaki.clients import ClientError |
|
38 |
from kamaki.clients.cyclades import CycladesClient, CycladesRestClient |
|
38 |
from kamaki.clients import ClientError, cyclades |
|
39 | 39 |
|
40 | 40 |
img_ref = "1m4g3-r3f3r3nc3" |
41 | 41 |
vm_name = "my new VM" |
... | ... | |
94 | 94 |
cyclades_pkg = 'kamaki.clients.cyclades.CycladesClient' |
95 | 95 |
|
96 | 96 |
|
97 |
class CycladesRest(TestCase): |
|
97 |
class CycladesRestClient(TestCase):
|
|
98 | 98 |
|
99 | 99 |
"""Set up a Cyclades thorough test""" |
100 | 100 |
def setUp(self): |
101 | 101 |
self.url = 'http://cyclades.example.com' |
102 | 102 |
self.token = 'cyc14d3s70k3n' |
103 |
self.client = CycladesRestClient(self.url, self.token) |
|
103 |
self.client = cyclades.CycladesRestClient(self.url, self.token)
|
|
104 | 104 |
|
105 | 105 |
def tearDown(self): |
106 | 106 |
FR.json = vm_recv |
... | ... | |
208 | 208 |
**kwargs)) |
209 | 209 |
|
210 | 210 |
|
211 |
class Cyclades(TestCase): |
|
211 |
class CycladesClient(TestCase):
|
|
212 | 212 |
|
213 | 213 |
def assert_dicts_are_equal(self, d1, d2): |
214 | 214 |
for k, v in d1.items(): |
... | ... | |
222 | 222 |
def setUp(self): |
223 | 223 |
self.url = 'http://cyclades.example.com' |
224 | 224 |
self.token = 'cyc14d3s70k3n' |
225 |
self.client = CycladesClient(self.url, self.token) |
|
225 |
self.client = cyclades.CycladesClient(self.url, self.token)
|
|
226 | 226 |
|
227 | 227 |
def tearDown(self): |
228 | 228 |
FR.status_code = 200 |
... | ... | |
272 | 272 |
vm_id = vm_recv['server']['id'] |
273 | 273 |
v = firewalls['attachments']['values'][0]['firewallProfile'] |
274 | 274 |
with patch.object( |
275 |
CycladesClient, 'get_server_details', |
|
275 |
cyclades.CycladesClient, 'get_server_details',
|
|
276 | 276 |
return_value=firewalls) as GSD: |
277 | 277 |
r = self.client.get_firewall_profile(vm_id) |
278 | 278 |
GSD.assert_called_once_with(vm_id) |
279 | 279 |
self.assertEqual(r, v) |
280 | 280 |
with patch.object( |
281 |
CycladesClient, 'get_server_details', |
|
281 |
cyclades.CycladesClient, 'get_server_details',
|
|
282 | 282 |
return_value=dict()): |
283 | 283 |
self.assertRaises( |
284 | 284 |
ClientError, |
... | ... | |
344 | 344 |
dict(id='another-nic-id', network_id='another-net-id'), |
345 | 345 |
dict(id=nic_id * 2, network_id=net_id * 2)] |
346 | 346 |
with patch.object( |
347 |
CycladesClient, |
|
347 |
cyclades.CycladesClient,
|
|
348 | 348 |
'list_server_nics', |
349 | 349 |
return_value=vm_nics) as LSN: |
350 | 350 |
r = self.client.disconnect_server(vm_id, nic_id) |
... | ... | |
392 | 392 |
net_id = net_recv['network']['id'] |
393 | 393 |
nics = ['nic1', 'nic2', 'nic3'] |
394 | 394 |
with patch.object( |
395 |
CycladesClient, |
|
395 |
cyclades.CycladesClient,
|
|
396 | 396 |
'list_network_nics', |
397 | 397 |
return_value=nics) as LNN: |
398 | 398 |
self.client.disconnect_network_nics(net_id) |
... | ... | |
422 | 422 |
def test_delete_network(self): |
423 | 423 |
net_id = net_recv['network']['id'] |
424 | 424 |
with patch.object( |
425 |
CycladesClient, 'networks_delete', |
|
425 |
cyclades.CycladesClient, 'networks_delete',
|
|
426 | 426 |
return_value=FR()) as ND: |
427 | 427 |
self.client.delete_network(net_id) |
428 | 428 |
ND.assert_called_once_with(net_id) |
429 | 429 |
with patch.object( |
430 |
CycladesClient, 'networks_delete', |
|
430 |
cyclades.CycladesClient, 'networks_delete',
|
|
431 | 431 |
side_effect=ClientError('A 421 Error', 421)): |
432 | 432 |
try: |
433 | 433 |
self.client.delete_network(421) |
... | ... | |
441 | 441 |
from sys import argv |
442 | 442 |
from kamaki.clients.test import runTestCase |
443 | 443 |
not_found = True |
444 |
if not argv[1:] or argv[1] == 'Cyclades': |
|
444 |
if not argv[1:] or argv[1] == 'CycladesClient':
|
|
445 | 445 |
not_found = False |
446 |
runTestCase(Cyclades, 'Cyclades Client', argv[2:]) |
|
447 |
if not argv[1:] or argv[1] == 'CycladesRest': |
|
446 |
runTestCase(CycladesClient, 'Cyclades Client', argv[2:])
|
|
447 |
if not argv[1:] or argv[1] == 'CycladesRestClient':
|
|
448 | 448 |
not_found = False |
449 |
runTestCase(CycladesRest, 'CycladesRest Client', argv[2:]) |
|
449 |
runTestCase(CycladesRestClient, 'CycladesRest Client', argv[2:])
|
|
450 | 450 |
if not_found: |
451 | 451 |
print('TestCase %s not found' % argv[1]) |
Also available in: Unified diff