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)