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_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])
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])