- def test_get_firewall_profile(self):
- vm_id = vm_recv['server']['id']
- v = firewalls['attachments'][0]['firewallProfile']
- with patch.object(
- cyclades.CycladesClient, 'get_server_details',
- return_value=firewalls) as GSD:
- r = self.client.get_firewall_profile(vm_id)
- GSD.assert_called_once_with(vm_id)
- self.assertEqual(r, v)
- with patch.object(
- cyclades.CycladesClient, 'get_server_details',
- return_value=dict()):
- self.assertRaises(
- ClientError,
- self.client.get_firewall_profile,
- vm_id)
-
- @patch('%s.servers_action_post' % cyclades_pkg, return_value=FR())
- def test_set_firewall_profile(self, SP):
- vm_id = vm_recv['server']['id']
- v = firewalls['attachments'][0]['firewallProfile']
- self.client.set_firewall_profile(vm_id, v)
- SP.assert_called_once_with(vm_id, json_data=dict(
- firewallProfile=dict(profile=v)), success=202)
-
- @patch('%s.networks_post' % cyclades_pkg, return_value=FR())
- def test_create_network(self, NP):
- net_name = net_send['network']['name']
- FR.json = net_recv
- full_args = dict(
- cidr='192.168.0.0/24',
- gateway='192.168.0.1',
- type='MAC_FILTERED',
- dhcp=True)
- test_args = dict(full_args)
- test_args.update(dict(empty=None, full=None))
- net_exp = dict(dhcp=False, name=net_name, type='MAC_FILTERED')
- for arg, val in test_args.items():
- kwargs = {} if arg == 'empty' else full_args if (
- arg == 'full') else {arg: val}
- expected = dict(network=dict(net_exp))
- expected['network'].update(kwargs)
- r = self.client.create_network(net_name, **kwargs)
- self.assertEqual(
- NP.mock_calls[-1],
- call(json_data=expected, success=202))
- self.assert_dicts_are_equal(r, net_recv['network'])
-
- @patch('%s.networks_post' % cyclades_pkg, return_value=FR())
- def test_connect_server(self, NP):
- vm_id = vm_recv['server']['id']
- net_id = net_recv['network']['id']
- self.client.connect_server(vm_id, net_id)
- NP.assert_called_once_with(
- net_id, 'action',
- json_data=dict(add=dict(serverRef=vm_id)))
-
- @patch('%s.networks_post' % cyclades_pkg, return_value=FR())
- def test_disconnect_server(self, NP):
- net_id, vm_id = net_recv['network']['id'], vm_recv['server']['id']
- nic_id = 'nic-%s-%s' % (net_id, vm_id)
- vm_nics = [
- dict(id=nic_id, network_id=net_id),
- dict(id='another-nic-id', network_id='another-net-id'),
- dict(id=nic_id * 2, network_id=net_id * 2)]
- with patch.object(
- cyclades.CycladesClient,
- 'list_server_nics',
- return_value=vm_nics) as LSN:
- r = self.client.disconnect_server(vm_id, nic_id)
- LSN.assert_called_once_with(vm_id)
- NP.assert_called_once_with(
- net_id, 'action',
- json_data=dict(remove=dict(attachment=nic_id)))
- self.assertEqual(r, 1)
-
- @patch('%s.servers_ips_get' % cyclades_pkg, return_value=FR())
- def test_list_server_nics(self, SG):
- vm_id = vm_recv['server']['id']
- nics = dict(attachments=[dict(id='nic1'), dict(id='nic2')])
- FR.json = nics
- r = self.client.list_server_nics(vm_id)
- SG.assert_called_once_with(vm_id)
- expected = nics['attachments']
- for i in range(len(r)):
- self.assert_dicts_are_equal(r[i], expected[i])
- self.assertEqual(i + 1, len(r))
-
- @patch('%s.networks_get' % cyclades_pkg, return_value=FR())
- def test_list_networks(self, NG):
- FR.json = net_list
- expected = net_list['networks']
- for detail in ('', 'detail'):
- r = self.client.list_networks(detail=True if detail else False)
- self.assertEqual(NG.mock_calls[-1], call(command=detail))
- for i, net in enumerate(expected):
- self.assert_dicts_are_equal(r[i], net)
- self.assertEqual(i + 1, len(r))
-
- @patch('%s.networks_get' % cyclades_pkg, return_value=FR())
- def test_list_network_nics(self, NG):
- net_id = net_recv['network']['id']
- FR.json = net_recv
- r = self.client.list_network_nics(net_id)
- NG.assert_called_once_with(network_id=net_id)
- expected = net_recv['network']['attachments']
- for i in range(len(r)):
- self.assert_dicts_are_equal(r[i], expected[i])
-
- @patch('%s.networks_post' % cyclades_pkg, return_value=FR())
- def test_disconnect_network_nics(self, NP):
- net_id = net_recv['network']['id']
- nics = ['nic1', 'nic2', 'nic3']
- with patch.object(
- cyclades.CycladesClient,
- 'list_network_nics',
- return_value=nics) as LNN:
- self.client.disconnect_network_nics(net_id)
- LNN.assert_called_once_with(net_id)
- for i in range(len(nics)):
- expected = call(net_id, 'action', json_data=dict(
- remove=dict(attachment=nics[i])))
- self.assertEqual(expected, NP.mock_calls[i])
-
- @patch('%s.networks_get' % cyclades_pkg, return_value=FR())
- def test_get_network_details(self, NG):
- FR.json = net_recv
- net_id = net_recv['network']['id']
- r = self.client.get_network_details(net_id)
- NG.assert_called_once_with(network_id=net_id)
- self.assert_dicts_are_equal(r, net_recv['network'])
-
- @patch('%s.networks_put' % cyclades_pkg, return_value=FR())
- def test_update_network_name(self, NP):
- net_id = net_recv['network']['id']
- new_name = '%s_new' % net_id
- self.client.update_network_name(net_id, new_name)
- NP.assert_called_once_with(
- network_id=net_id,
- json_data=dict(network=dict(name=new_name)))
-
- def test_delete_network(self):
- net_id = net_recv['network']['id']
- with patch.object(
- cyclades.CycladesClient, 'networks_delete',
- return_value=FR()) as ND:
- self.client.delete_network(net_id)
- ND.assert_called_once_with(net_id)
- with patch.object(
- cyclades.CycladesClient, 'networks_delete',
- side_effect=ClientError('A 421 Error', 421)):
- try:
- self.client.delete_network(421)
- except ClientError as err:
- self.assertEqual(err.status, 421)
- self.assertEqual(err.details, [
- 'Network may be still connected to at least one server'])
-
- @patch('%s.floating_ip_pools_get' % cyclades_pkg, return_value=FR())
- def test_get_floating_ip_pools(self, get):
- r = self.client.get_floating_ip_pools()
- self.assert_dicts_are_equal(r, FR.json)
- self.assertEqual(get.mock_calls[-1], call())
-
- @patch('%s.floating_ips_get' % cyclades_pkg, return_value=FR())
- def test_get_floating_ips(self, get):
- r = self.client.get_floating_ips()
- self.assert_dicts_are_equal(r, FR.json)
- self.assertEqual(get.mock_calls[-1], call())
-
- @patch('%s.floating_ips_post' % cyclades_pkg, return_value=FR())
- def test_alloc_floating_ip(self, post):
- FR.json = dict(floating_ip=dict(
- fixed_ip='fip',
- id=1,
- instance_id='lala',
- ip='102.0.0.1',
- pool='pisine'))
- for args in product(
- (None, 'pisine'),
- (None, 'Iwannanip')):
- r = self.client.alloc_floating_ip(*args)
- pool, address = args
- self.assert_dicts_are_equal(r, FR.json['floating_ip'])
- json_data = dict()
- if pool:
- json_data['pool'] = pool
- if address:
- json_data['address'] = address
- self.assertEqual(post.mock_calls[-1], call(json_data))
-
- @patch('%s.floating_ips_get' % cyclades_pkg, return_value=FR())
- def test_get_floating_ip(self, get):
- FR.json = dict(floating_ip=dict(
- fixed_ip='fip',
- id=1,
- instance_id='lala',
- ip='102.0.0.1',
- pool='pisine'))
- self.assertRaises(AssertionError, self.client.get_floating_ip, None)
- fip = 'fip'
- r = self.client.get_floating_ip(fip)
- self.assert_dicts_are_equal(r, FR.json['floating_ip'])
- self.assertEqual(get.mock_calls[-1], call(fip))
-
- @patch('%s.floating_ips_delete' % cyclades_pkg, return_value=FR())
- def test_delete_floating_ip(self, delete):
- self.assertRaises(AssertionError, self.client.delete_floating_ip, None)
- fip = 'fip'
- r = self.client.delete_floating_ip(fip)
- self.assert_dicts_are_equal(r, FR.headers)
- self.assertEqual(delete.mock_calls[-1], call(fip))
-
- @patch('%s.servers_action_post' % cyclades_pkg, return_value=FR())
- def test_attach_floating_ip(self, spost):
- vmid, addr = 42, 'anIpAddress'
- for err, args in {
- ValueError: ['not a server id', addr],
- TypeError: [None, addr],
- AssertionError: [vmid, None],
- AssertionError: [vmid, '']}.items():
- self.assertRaises(
- err, self.client.attach_floating_ip, *args)
- r = self.client.attach_floating_ip(vmid, addr)
- self.assert_dicts_are_equal(r, FR.headers)
- expected = dict(addFloatingIp=dict(address=addr))
- self.assertEqual(spost.mock_calls[-1], call(vmid, json_data=expected))
-
- @patch('%s.servers_action_post' % cyclades_pkg, return_value=FR())
- def test_detach_floating_ip(self, spost):
- vmid, addr = 42, 'anIpAddress'
- for err, args in {
- ValueError: ['not a server id', addr],
- TypeError: [None, addr],
- AssertionError: [vmid, None],
- AssertionError: [vmid, '']}.items():
- self.assertRaises(
- err, self.client.detach_floating_ip, *args)
- r = self.client.detach_floating_ip(vmid, addr)
- self.assert_dicts_are_equal(r, FR.headers)
- expected = dict(removeFloatingIp=dict(address=addr))
- self.assertEqual(spost.mock_calls[-1], call(vmid, json_data=expected))
-
projects / kamaki / commitdiff