118 lines
4.0 KiB
Python
118 lines
4.0 KiB
Python
# Copyright (c) 2019 Red Hat, Inc.
|
|
#
|
|
# All Rights Reserved.
|
|
#
|
|
# 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.
|
|
from __future__ import absolute_import
|
|
|
|
import os
|
|
import random
|
|
import string
|
|
import typing
|
|
|
|
import testtools
|
|
|
|
import tobiko
|
|
from tobiko.openstack import heat
|
|
from tobiko.openstack import neutron
|
|
from tobiko.openstack import nova
|
|
|
|
|
|
TEMPLATE_DIRS = [os.path.dirname(__file__)]
|
|
|
|
|
|
def random_string(length, letters=string.ascii_lowercase):
|
|
return ''.join(random.choice(letters) for _ in range(length))
|
|
|
|
|
|
class MyStack(heat.HeatStackFixture):
|
|
|
|
template = heat.heat_template_file(template_file='hot/my_stack.yaml',
|
|
template_dirs=TEMPLATE_DIRS)
|
|
|
|
input_text = random_string(8)
|
|
|
|
|
|
class HeatStackFixtureTest(testtools.TestCase):
|
|
|
|
stack = tobiko.required_setup_fixture(MyStack)
|
|
|
|
def test_get_stack(self):
|
|
self.stack.wait_for_create_complete()
|
|
stack = self.stack.get_stack()
|
|
self.assertIsNotNone(stack)
|
|
self.assertEqual(tobiko.get_fixture_name(MyStack), stack.stack_name)
|
|
self.assertIsInstance(stack.id, str)
|
|
self.assertIsInstance(stack.stack_status, str)
|
|
|
|
def test_get_fixture_with_fixture_id_0(self):
|
|
fixture_0 = tobiko.get_fixture(MyStack, fixture_id=0)
|
|
self.assertIs(fixture_0, self.stack)
|
|
|
|
def test_get_fixture_with_fixture_id_1(self):
|
|
fixture_0 = tobiko.get_fixture(MyStack)
|
|
fixture_1 = tobiko.get_fixture(MyStack, fixture_id=1)
|
|
self.assertIsNot(fixture_0, fixture_1)
|
|
stack_0 = tobiko.setup_fixture(fixture_0).get_stack()
|
|
stack_1 = tobiko.setup_fixture(fixture_1).get_stack()
|
|
self.assertNotEqual(stack_0.id, stack_1.id)
|
|
self.assertEqual(tobiko.get_fixture_name(MyStack), stack_0.stack_name)
|
|
self.assertEqual(tobiko.get_fixture_name(MyStack) + '-1',
|
|
stack_1.stack_name)
|
|
|
|
|
|
class EnsureNeutronQuotaLimitsFixture(MyStack):
|
|
|
|
requirements = {'network': 100, 'subnet': 100, 'router': 10}
|
|
|
|
@property
|
|
def neutron_required_quota_set(self) -> typing.Dict[str, int]:
|
|
return self.requirements
|
|
|
|
|
|
class EnsureNovaQuotaLimitsFixture(MyStack):
|
|
|
|
requirements = {'cores': 20, 'instances': 10}
|
|
|
|
@property
|
|
def nova_required_quota_set(self) -> typing.Dict[str, int]:
|
|
return self.requirements
|
|
|
|
|
|
class EnsureQuotaLimitsTest(testtools.TestCase):
|
|
|
|
def test_ensure_neutron_quota_limits(self):
|
|
stack = EnsureNeutronQuotaLimitsFixture(stack_name=self.id())
|
|
self.useFixture(stack)
|
|
quota_set = neutron.get_neutron_quota_set(detail=True)
|
|
for name, requirement in stack.requirements.items():
|
|
quota = quota_set[name]
|
|
self.assertGreaterEqual(quota['limit'],
|
|
requirement +
|
|
max(0, quota['used']) -
|
|
max(0, quota['reserved']))
|
|
|
|
def test_ensure_nova_quota_limits(self):
|
|
stack = EnsureNovaQuotaLimitsFixture(stack_name=self.id())
|
|
self.useFixture(stack)
|
|
quota_set = nova.get_nova_quota_set(detail=True)
|
|
for name, requirement in stack.requirements.items():
|
|
quota = getattr(quota_set, name)
|
|
if quota['limit'] > 0:
|
|
self.assertGreaterEqual(quota['limit'],
|
|
requirement +
|
|
max(0, quota['in_use']) -
|
|
max(0, quota['reserved']))
|
|
else:
|
|
self.assertEqual(quota['limit'], -1)
|