77 lines
3.3 KiB
Python
77 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2015 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.
|
|
|
|
from nailgun import consts
|
|
from nailgun.extensions.network_manager.manager import NetworkManager
|
|
from nailgun import objects
|
|
from nailgun.task.task import CheckRepoAvailability
|
|
from nailgun.task.task import CheckRepoAvailabilityWithSetup
|
|
from nailgun.test.base import BaseTestCase
|
|
|
|
|
|
class TestRepoAvailability(BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestRepoAvailability, self).setUp()
|
|
self.cluster = self.env.create(
|
|
cluster_kwargs={
|
|
'net_provider': 'neutron',
|
|
'net_segment_type': 'gre'
|
|
},
|
|
nodes_kwargs=[{'roles': ['controller']},
|
|
{'roles': ['controller']},
|
|
{'roles': ['compute']},
|
|
{'roles': ['compute'], 'online': False}])
|
|
|
|
self.public_ng = next(ng for ng in self.cluster.network_groups
|
|
if ng.name == 'public')
|
|
self.free_ips = NetworkManager.get_free_ips(self.public_ng, 2)
|
|
self.repo_urls = objects.Cluster.get_repo_urls(self.cluster)
|
|
self.controllers = [n for n in self.cluster.nodes
|
|
if 'controller' in n.all_roles]
|
|
self.online_uids = [n.uid for n in self.cluster.nodes if n.online]
|
|
|
|
def test_repo_with_setup_generate_config(self):
|
|
config, errors = CheckRepoAvailabilityWithSetup.get_config(
|
|
self.cluster)
|
|
self.assertEqual(len(config), 2)
|
|
# resulted list of nodes contains only online uids
|
|
self.assertTrue(
|
|
set([n['uid'] for n in config]) <= set(self.online_uids))
|
|
|
|
control_1, control_2 = self.controllers
|
|
control_1_conf = next(c for c in config if c['uid'] == control_1.uid)
|
|
control_2_conf = next(c for c in config if c['uid'] == control_2.uid)
|
|
|
|
self.assertNotEqual(control_1_conf['addr'], control_2_conf['addr'])
|
|
self.assertEqual(control_1_conf['gateway'], control_2_conf['gateway'])
|
|
self.assertEqual(control_1_conf['gateway'], self.public_ng.gateway)
|
|
self.assertIn(control_1_conf['addr'].split('/')[0], self.free_ips)
|
|
self.assertIn(control_2_conf['addr'].split('/')[0], self.free_ips)
|
|
|
|
self.assertEqual(control_1_conf['urls'], control_2_conf['urls'])
|
|
self.assertEqual(control_1_conf['urls'], self.repo_urls)
|
|
|
|
self.assertEqual(control_1_conf['vlan'], control_2_conf['vlan'])
|
|
|
|
def test_nodes_to_check(self):
|
|
task = objects.Task.create({
|
|
'cluster': self.cluster,
|
|
'name': consts.TASK_NAMES.check_repo_availability})
|
|
repo_check = CheckRepoAvailability(task, {})
|
|
self.assertItemsEqual(
|
|
[consts.MASTER_NODE_UID] + self.online_uids,
|
|
[str(n['uid']) for n in repo_check._get_nodes_to_check()])
|