def setUp(self):
"""Setup Environment"""
super(NovaClientTestCase, self).setUp()
NovaClient.create_nova_client = mock.MagicMock()
self.client = NovaClient(dummy_context())
self.client._client = mock.MagicMock()
class NovaClientTestCase(ValidatorTestCase):
def setUp(self):
"""Setup Environment"""
super(NovaClientTestCase, self).setUp()
NovaClient.create_nova_client = mock.MagicMock()
self.client = NovaClient(dummy_context())
self.client._client = mock.MagicMock()
def test_list(self):
"""Test list function"""
dummy_image = mock.MagicMock()
dummy_image.id = "myid"
dummy_image.name = "myname"
exp = [{'id': 'myid', 'name': 'myname'}]
self.client._client.images.list.return_value = [dummy_image]
self.assertEqual(exp, self.client.list())
def test_get_image_by_name(self):
"""Test get_image_by_name function"""
dummy_image = mock.MagicMock()
dummy_image.id = "myid"
dummy_image.name = "myname"
exp = {'id': 'myid', 'name': 'myname'}
self.client._client.images.list.return_value = [dummy_image]
self.assertEqual(exp, self.client.get_image_by_name("myname"))
def test_get_ip(self):
"""Test get_ip function"""
addresses = {
'public': [{'version': 4,
'addr': '4.5.6.7'},
{'version': 6,
'addr': '2401:1801:7800:0101:c058:dd33:ff18:04e6'}],
'private': [{'version': 4,
'addr': '10.13.12.13'}]}
expected = '4.5.6.7'
self.client._client.servers.ips.return_value = addresses
observed = self.client.get_ip()
self.assertEqual(expected, observed)
def test_get_machine(self):
"""Test get_machine function"""
self.client._client.servers.find(name="mymachine").return_value = True
expected = True
observed = self.client.get_machine("mymachine")
self.assertEqual(expected, observed)
def test_deploy_machine(self):
"""Test deploy_machine function"""
machine = mock.MagicMock()
self.client._client.servers.create.return_value = machine
expected = machine
self.client.deploy_machine("mymachine", "myimage")
observed = self.client._machine
self.assertEqual(expected, observed)
def test_delete_machine(self):
"""Test delete_machine function"""
machine = mock.MagicMock()
self.client._client.servers.find.return_value = machine
self.client.delete_machine("mymachine")
machine.delete.assert_called_once_with()
def tearDown(self):
"""Cleanup environment"""
super(NovaClientTestCase, self).tearDown()
self.m.UnsetStubs()
self.m.ResetAll()
