def test_update_limits_for_ng(self, nova_mock): flavor_mock = mock.Mock() type(flavor_mock).ram = mock.PropertyMock(return_value=4) type(flavor_mock).vcpus = mock.PropertyMock(return_value=2) flavor_get_mock = mock.Mock() flavor_get_mock.get.return_value = flavor_mock type(nova_mock.return_value).flavors = mock.PropertyMock( return_value=flavor_get_mock) ng = mock.Mock() type(ng).flavor_id = mock.PropertyMock(return_value=3) type(ng).floating_ip_pool = mock.PropertyMock(return_value='pool') type(ng).volumes_per_node = mock.PropertyMock(return_value=4) type(ng).volumes_size = mock.PropertyMock(return_value=5) type(ng).auto_security_group = mock.PropertyMock(return_value=True) type(ng).open_ports = mock.PropertyMock(return_value=[1111, 2222]) limits = quotas._get_zero_limits() quotas._update_limits_for_ng(limits, ng, 3) self.assertEqual(3, limits['instances']) self.assertEqual(12, limits['ram']) self.assertEqual(6, limits['cpu']) self.assertEqual(3, limits['floatingips']) self.assertEqual(12, limits['volumes']) self.assertEqual(60, limits['volume_gbs']) self.assertEqual(1, limits['security_groups']) self.assertEqual(5, limits['security_group_rules']) self.assertEqual(3, limits['ports'])
def test_get_zero_limits(self): res = quotas._get_zero_limits() self.assertEqual(9, len(res)) for key in self.LIST_LIMITS: self.assertEqual(0, res[key])