209 lines
8.1 KiB
Python
209 lines
8.1 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import copy
|
|
|
|
from keystoneauth1 import adapter
|
|
import mock
|
|
from openstack.tests.unit import base
|
|
|
|
from openstack.compute.v2 import limits
|
|
|
|
ABSOLUTE_LIMITS = {
|
|
"maxImageMeta": 128,
|
|
"maxPersonality": 5,
|
|
"maxPersonalitySize": 10240,
|
|
"maxSecurityGroupRules": 20,
|
|
"maxSecurityGroups": 10,
|
|
"maxServerMeta": 128,
|
|
"maxTotalCores": 20,
|
|
"maxTotalFloatingIps": 10,
|
|
"maxTotalInstances": 10,
|
|
"maxTotalKeypairs": 100,
|
|
"maxTotalRAMSize": 51200,
|
|
"maxServerGroups": 10,
|
|
"maxServerGroupMembers": 10,
|
|
"totalFloatingIpsUsed": 1,
|
|
"totalSecurityGroupsUsed": 2,
|
|
"totalRAMUsed": 4,
|
|
"totalInstancesUsed": 5,
|
|
"totalServerGroupsUsed": 6,
|
|
"totalCoresUsed": 7
|
|
}
|
|
|
|
RATE_LIMIT = {
|
|
"limit": [
|
|
{
|
|
"next-available": "2012-11-27T17:22:18Z",
|
|
"remaining": 120,
|
|
"unit": "MINUTE",
|
|
"value": 120,
|
|
"verb": "POST"
|
|
},
|
|
],
|
|
"regex": ".*",
|
|
"uri": "*"
|
|
}
|
|
|
|
LIMITS_BODY = {
|
|
"limits": {
|
|
"absolute": ABSOLUTE_LIMITS,
|
|
"rate": [RATE_LIMIT]
|
|
}
|
|
}
|
|
|
|
|
|
class TestAbsoluteLimits(base.TestCase):
|
|
|
|
def test_basic(self):
|
|
sot = limits.AbsoluteLimits()
|
|
self.assertIsNone(sot.resource_key)
|
|
self.assertIsNone(sot.resources_key)
|
|
self.assertEqual("", sot.base_path)
|
|
self.assertFalse(sot.allow_create)
|
|
self.assertFalse(sot.allow_fetch)
|
|
self.assertFalse(sot.allow_commit)
|
|
self.assertFalse(sot.allow_delete)
|
|
self.assertFalse(sot.allow_list)
|
|
|
|
def test_make_it(self):
|
|
sot = limits.AbsoluteLimits(**ABSOLUTE_LIMITS)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxImageMeta"], sot.image_meta)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxPersonality"], sot.personality)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxPersonalitySize"],
|
|
sot.personality_size)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxSecurityGroupRules"],
|
|
sot.security_group_rules)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxSecurityGroups"],
|
|
sot.security_groups)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxServerMeta"], sot.server_meta)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxTotalCores"], sot.total_cores)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxTotalFloatingIps"],
|
|
sot.floating_ips)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxTotalInstances"],
|
|
sot.instances)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxTotalKeypairs"],
|
|
sot.keypairs)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxTotalRAMSize"],
|
|
sot.total_ram)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxServerGroups"], sot.server_groups)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxServerGroupMembers"],
|
|
sot.server_group_members)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalFloatingIpsUsed"],
|
|
sot.floating_ips_used)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalSecurityGroupsUsed"],
|
|
sot.security_groups_used)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalRAMUsed"], sot.total_ram_used)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalInstancesUsed"],
|
|
sot.instances_used)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalServerGroupsUsed"],
|
|
sot.server_groups_used)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalCoresUsed"],
|
|
sot.total_cores_used)
|
|
|
|
|
|
class TestRateLimit(base.TestCase):
|
|
|
|
def test_basic(self):
|
|
sot = limits.RateLimit()
|
|
self.assertIsNone(sot.resource_key)
|
|
self.assertIsNone(sot.resources_key)
|
|
self.assertEqual("", sot.base_path)
|
|
self.assertIsNone(sot.service)
|
|
self.assertFalse(sot.allow_create)
|
|
self.assertFalse(sot.allow_fetch)
|
|
self.assertFalse(sot.allow_commit)
|
|
self.assertFalse(sot.allow_delete)
|
|
self.assertFalse(sot.allow_list)
|
|
|
|
def test_make_it(self):
|
|
sot = limits.RateLimit(**RATE_LIMIT)
|
|
self.assertEqual(RATE_LIMIT["regex"], sot.regex)
|
|
self.assertEqual(RATE_LIMIT["uri"], sot.uri)
|
|
self.assertEqual(RATE_LIMIT["limit"], sot.limits)
|
|
|
|
|
|
class TestLimits(base.TestCase):
|
|
|
|
def test_basic(self):
|
|
sot = limits.Limits()
|
|
self.assertEqual("limits", sot.resource_key)
|
|
self.assertEqual("/limits", sot.base_path)
|
|
self.assertTrue(sot.allow_fetch)
|
|
self.assertFalse(sot.allow_create)
|
|
self.assertFalse(sot.allow_commit)
|
|
self.assertFalse(sot.allow_delete)
|
|
self.assertFalse(sot.allow_list)
|
|
|
|
def test_get(self):
|
|
sess = mock.Mock(spec=adapter.Adapter)
|
|
sess.default_microversion = None
|
|
resp = mock.Mock()
|
|
sess.get.return_value = resp
|
|
resp.json.return_value = copy.deepcopy(LIMITS_BODY)
|
|
resp.headers = {}
|
|
resp.status_code = 200
|
|
|
|
sot = limits.Limits().fetch(sess)
|
|
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxImageMeta"],
|
|
sot.absolute.image_meta)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxPersonality"],
|
|
sot.absolute.personality)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxPersonalitySize"],
|
|
sot.absolute.personality_size)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxSecurityGroupRules"],
|
|
sot.absolute.security_group_rules)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxSecurityGroups"],
|
|
sot.absolute.security_groups)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxServerMeta"],
|
|
sot.absolute.server_meta)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxTotalCores"],
|
|
sot.absolute.total_cores)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxTotalFloatingIps"],
|
|
sot.absolute.floating_ips)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxTotalInstances"],
|
|
sot.absolute.instances)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxTotalKeypairs"],
|
|
sot.absolute.keypairs)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxTotalRAMSize"],
|
|
sot.absolute.total_ram)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxServerGroups"],
|
|
sot.absolute.server_groups)
|
|
self.assertEqual(ABSOLUTE_LIMITS["maxServerGroupMembers"],
|
|
sot.absolute.server_group_members)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalFloatingIpsUsed"],
|
|
sot.absolute.floating_ips_used)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalSecurityGroupsUsed"],
|
|
sot.absolute.security_groups_used)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalRAMUsed"],
|
|
sot.absolute.total_ram_used)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalInstancesUsed"],
|
|
sot.absolute.instances_used)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalServerGroupsUsed"],
|
|
sot.absolute.server_groups_used)
|
|
self.assertEqual(ABSOLUTE_LIMITS["totalCoresUsed"],
|
|
sot.absolute.total_cores_used)
|
|
|
|
self.assertEqual(RATE_LIMIT["uri"], sot.rate[0].uri)
|
|
self.assertEqual(RATE_LIMIT["regex"], sot.rate[0].regex)
|
|
self.assertEqual(RATE_LIMIT["limit"], sot.rate[0].limits)
|
|
|
|
dsot = sot.to_dict()
|
|
|
|
self.assertIsInstance(dsot['rate'][0], dict)
|
|
self.assertIsInstance(dsot['absolute'], dict)
|
|
self.assertEqual(RATE_LIMIT["uri"], dsot['rate'][0]['uri'])
|
|
self.assertEqual(
|
|
ABSOLUTE_LIMITS["totalSecurityGroupsUsed"],
|
|
dsot['absolute']['security_groups_used'])
|