80 lines
3.1 KiB
Python
80 lines
3.1 KiB
Python
# Copyright 2013 Mirantis, Inc.
|
|
#
|
|
# 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 requests_mock
|
|
|
|
from fuel_plugin.ostf_adapter import config
|
|
from fuel_plugin.ostf_adapter import mixins
|
|
from fuel_plugin.testing.tests import base
|
|
|
|
|
|
class TestDeplTagsGetter(base.BaseUnitTest):
|
|
|
|
def setUp(self):
|
|
config.init_config([])
|
|
|
|
def test_get_cluster_depl_tags(self):
|
|
expected = {
|
|
'cluster_id': 3,
|
|
'attrs': {
|
|
'deployment_tags': set(
|
|
['ha', 'rhel', 'additional_components',
|
|
'murano', 'nova_network', 'public_on_all_nodes',
|
|
'enable_without_ceph']),
|
|
'release_version': '2015.2-1.0'
|
|
}
|
|
}
|
|
|
|
with requests_mock.Mocker() as m:
|
|
cluster = base.CLUSTERS[expected['cluster_id']]
|
|
m.register_uri('GET', '/api/clusters/3',
|
|
json=cluster['cluster_meta'])
|
|
m.register_uri('GET', '/api/clusters/3/attributes',
|
|
json=cluster['cluster_attributes'])
|
|
m.register_uri('GET', '/api/releases/3',
|
|
json=cluster['release_data'])
|
|
m.register_uri('GET', '/api/nodes?cluster_id=3',
|
|
json=cluster['cluster_node'])
|
|
res = mixins._get_cluster_attrs(expected['cluster_id'])
|
|
|
|
self.assertEqual(res, expected['attrs'])
|
|
|
|
def test_sriov_deployment_tag(self):
|
|
expected = {
|
|
'cluster_id': 7,
|
|
'attrs': {
|
|
'deployment_tags': set(
|
|
['ha', 'rhel', 'additional_components',
|
|
'murano', 'nova_network', 'public_on_all_nodes',
|
|
'enable_without_ceph', 'sriov']),
|
|
'release_version': '2015.2-1.0'
|
|
}
|
|
}
|
|
|
|
with requests_mock.Mocker() as m:
|
|
cluster = base.CLUSTERS[expected['cluster_id']]
|
|
m.register_uri('GET', '/api/clusters/7',
|
|
json=cluster['cluster_meta'])
|
|
m.register_uri('GET', '/api/clusters/7/attributes',
|
|
json=cluster['cluster_attributes'])
|
|
m.register_uri('GET', '/api/releases/7',
|
|
json=cluster['release_data'])
|
|
m.register_uri('GET', '/api/nodes?cluster_id=7',
|
|
json=cluster['cluster_node'])
|
|
m.register_uri('GET', '/api/nodes/1/interfaces',
|
|
json=cluster['node_interfaces'])
|
|
res = mixins._get_cluster_attrs(expected['cluster_id'])
|
|
|
|
self.assertEqual(res, expected['attrs'])
|