def setUp(self):
super(TestNovaNetwork, self).setUp()
self.nova_mock_client = mock.MagicMock()
self.nova_client_patch = mockpatch.PatchObject(
nova_client,
'Client',
new=self.nova_mock_client)
self.fake_cloud = mock.Mock()
self.useFixture(self.nova_client_patch)
self.nova_network_client = NovaNetwork(FAKE_CONFIG, self.fake_cloud)
self.sg1 = mock.Mock()
self.sg1.name = 'fake_name_1'
self.sg1.description = 'fake_description_1'
self.sg2 = mock.Mock()
self.sg2.name = 'fake_name_2'
self.sg2.description = 'fake_description_2'
self.sg2.rules = []
self.fake_instance = mock.Mock()
class TestNovaNetwork(test.TestCase):
def setUp(self):
super(TestNovaNetwork, self).setUp()
self.nova_mock_client = mock.MagicMock()
self.nova_client_patch = mockpatch.PatchObject(
nova_client,
'Client',
new=self.nova_mock_client)
self.fake_cloud = mock.Mock()
self.useFixture(self.nova_client_patch)
self.nova_network_client = NovaNetwork(FAKE_CONFIG, self.fake_cloud)
self.sg1 = mock.Mock()
self.sg1.name = 'fake_name_1'
self.sg1.description = 'fake_description_1'
self.sg2 = mock.Mock()
self.sg2.name = 'fake_name_2'
self.sg2.description = 'fake_description_2'
self.sg2.rules = []
self.fake_instance = mock.Mock()
def test_get_client(self):
self.nova_mock_client.reset_mock()
client = self.nova_network_client.get_client()
self.nova_mock_client.assert_called_once_with(
FAKE_CONFIG.cloud.user,
FAKE_CONFIG.cloud.password,
FAKE_CONFIG.cloud.tenant,
FAKE_CONFIG.cloud.auth_url,
cacert=FAKE_CONFIG.cloud.cacert,
insecure=FAKE_CONFIG.cloud.insecure,
region_name=FAKE_CONFIG.cloud.region)
self.assertEquals(self.nova_mock_client(), client)
def test_get_security_groups(self):
fake_security_groups = [self.sg1, self.sg2]
self.nova_mock_client().security_groups.list.return_value = (
fake_security_groups)
security_groups = self.nova_network_client.get_security_groups()
result = [x.__dict__ for x in fake_security_groups]
self.assertEquals(result, security_groups)
@mock.patch('cloudferry.lib.os.network.nova_network.NovaNetwork.'
'get_security_groups')
def test_upload_security_groups(self, mock_get):
fake_existing_groups = [self.sg1]
fake_security_groups = [self.sg1, self.sg2]
mock_get.return_value = fake_existing_groups
self.nova_network_client.upload_security_groups(fake_security_groups)
kwargs = {'name': 'fake_name_2',
'description': 'fake_description_2'}
self.nova_mock_client().security_groups.create.assert_called_once_with(
**kwargs)
