tripleo-validations/tripleo_validations/tests/library/test_check_flavors.py

68 lines
2.1 KiB
Python

# -*- coding: utf-8 -*-
# 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.
"""
test_check_flavors
--------------
Tests for the `check_flavors` module.
"""
try:
from unittest import mock
except ImportError:
import mock
from tripleo_validations.tests import base
from tripleo_validations.tests import fakes
from library import check_flavors
class TestCheckFlavors(base.TestCase):
def setUp(self):
self.tested_module = check_flavors
return super().setUp()
def test_module_init(self):
expepect_attrs = set(
[
'DOCUMENTATION',
'EXAMPLES',
'main'])
actual_attrs = set(dir(self.tested_module))
self.assertTrue(expepect_attrs.issubset(actual_attrs))
def test_validate_roles_and_flavors(self):
expected_values = fakes.MOCK_FLAVORS_CHECK_EXPECTED['ok']
return_value = self.tested_module.validate_roles_and_flavors(
fakes.MOCK_ROLES_INFO,
fakes.MOCK_FLAVORS['ok'])
self.assertEqual(expected_values, return_value)
def test_validate_roles_and_flavors_nocpu(self):
"""Tests situation when the 'VCPU' key doesn't have associated value.
'DISK_GB' and 'MEMORY_MB' behave the same way,
so we don't need to test them. For now at least.
"""
expected_values = fakes.MOCK_FLAVORS_CHECK_EXPECTED['fail_NOVCPU']
return_value = self.tested_module.validate_roles_and_flavors(
fakes.MOCK_ROLES_INFO,
fakes.MOCK_FLAVORS['fail_NOVCPU'])
self.assertEqual(expected_values, return_value)