62570525ad
Some Python class and packages need to be renamed for a better compliance with the shared terminology which provides a better understanding of Watcher objects and components by every contributor. This patchset is there to change the code structure by adding the folder "strategies" and "loading". Partially implements: blueprint glossary-related-refactoring Change-Id: I56fb24ee6762b3186eccde5983233e17bb227cc1
139 lines
5.0 KiB
Python
139 lines
5.0 KiB
Python
# -*- encoding: utf-8 -*-
|
|
# Copyright (c) 2015 b<>com
|
|
#
|
|
# Authors: Jean-Emile DARTOIS <jean-emile.dartois@b-com.com>
|
|
#
|
|
# 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 uuid
|
|
|
|
from watcher.common import exception
|
|
from watcher.common.exception import IllegalArgumentException
|
|
from watcher.decision_engine.model.hypervisor import Hypervisor
|
|
from watcher.decision_engine.model.hypervisor_state import HypervisorState
|
|
from watcher.decision_engine.model.model_root import ModelRoot
|
|
from watcher.tests import base
|
|
from watcher.tests.decision_engine.strategy.strategies.faker_cluster_state import \
|
|
FakerModelCollector
|
|
|
|
|
|
class TestModel(base.BaseTestCase):
|
|
def test_model(self):
|
|
fake_cluster = FakerModelCollector()
|
|
model = fake_cluster.generate_scenario_1()
|
|
|
|
self.assertEqual(len(model._hypervisors), 5)
|
|
self.assertEqual(len(model._vms), 35)
|
|
self.assertEqual(len(model.get_mapping().get_mapping()), 5)
|
|
|
|
def test_add_hypervisor(self):
|
|
model = ModelRoot()
|
|
id = "{0}".format(uuid.uuid4())
|
|
hypervisor = Hypervisor()
|
|
hypervisor.uuid = id
|
|
model.add_hypervisor(hypervisor)
|
|
self.assertEqual(model.get_hypervisor_from_id(id), hypervisor)
|
|
|
|
def test_delete_hypervisor(self):
|
|
model = ModelRoot()
|
|
id = "{0}".format(uuid.uuid4())
|
|
hypervisor = Hypervisor()
|
|
hypervisor.uuid = id
|
|
model.add_hypervisor(hypervisor)
|
|
self.assertEqual(model.get_hypervisor_from_id(id), hypervisor)
|
|
model.remove_hypervisor(hypervisor)
|
|
self.assertRaises(exception.HypervisorNotFound,
|
|
model.get_hypervisor_from_id, id)
|
|
|
|
def test_get_all_hypervisors(self):
|
|
model = ModelRoot()
|
|
for i in range(10):
|
|
id = "{0}".format(uuid.uuid4())
|
|
hypervisor = Hypervisor()
|
|
hypervisor.uuid = id
|
|
model.add_hypervisor(hypervisor)
|
|
all_hypervisors = model.get_all_hypervisors()
|
|
for id in all_hypervisors:
|
|
hyp = model.get_hypervisor_from_id(id)
|
|
model.assert_hypervisor(hyp)
|
|
|
|
def test_set_get_state_hypervisors(self):
|
|
model = ModelRoot()
|
|
id = "{0}".format(uuid.uuid4())
|
|
hypervisor = Hypervisor()
|
|
hypervisor.uuid = id
|
|
model.add_hypervisor(hypervisor)
|
|
|
|
self.assertIsInstance(hypervisor.state, HypervisorState)
|
|
|
|
hyp = model.get_hypervisor_from_id(id)
|
|
hyp.state = HypervisorState.OFFLINE
|
|
self.assertIsInstance(hyp.state, HypervisorState)
|
|
|
|
# /watcher/decision_engine/framework/model/hypervisor.py
|
|
# set_state accept any char chain.
|
|
# verification (IsInstance) should be used in the function
|
|
# hyp.set_state('blablabla')
|
|
# self.assertEqual(hyp.get_state(), 'blablabla')
|
|
# self.assertIsInstance(hyp.get_state(), HypervisorState)
|
|
|
|
# def test_get_all_vms(self):
|
|
# model = ModelRoot()
|
|
# vms = model.get_all_vms()
|
|
# self.assert(len(model._vms))
|
|
def test_hypervisor_from_id_raise(self):
|
|
model = ModelRoot()
|
|
id = "{0}".format(uuid.uuid4())
|
|
hypervisor = Hypervisor()
|
|
hypervisor.uuid = id
|
|
model.add_hypervisor(hypervisor)
|
|
|
|
id2 = "{0}".format(uuid.uuid4())
|
|
self.assertRaises(exception.HypervisorNotFound,
|
|
model.get_hypervisor_from_id, id2)
|
|
|
|
def test_remove_hypervisor_raise(self):
|
|
model = ModelRoot()
|
|
id = "{0}".format(uuid.uuid4())
|
|
hypervisor = Hypervisor()
|
|
hypervisor.uuid = id
|
|
model.add_hypervisor(hypervisor)
|
|
|
|
id2 = "{0}".format(uuid.uuid4())
|
|
hypervisor2 = Hypervisor()
|
|
hypervisor2.uuid = id2
|
|
|
|
self.assertRaises(exception.HypervisorNotFound,
|
|
model.remove_hypervisor, hypervisor2)
|
|
|
|
def test_assert_hypervisor_raise(self):
|
|
model = ModelRoot()
|
|
id = "{0}".format(uuid.uuid4())
|
|
hypervisor = Hypervisor()
|
|
hypervisor.uuid = id
|
|
model.add_hypervisor(hypervisor)
|
|
self.assertRaises(IllegalArgumentException,
|
|
model.assert_hypervisor, "objet_qcq")
|
|
|
|
def test_vm_from_id_raise(self):
|
|
fake_cluster = FakerModelCollector()
|
|
model = fake_cluster.generate_scenario_1()
|
|
self.assertRaises(exception.VMNotFound,
|
|
model.get_vm_from_id, "valeur_qcq")
|
|
|
|
def test_assert_vm_raise(self):
|
|
model = ModelRoot()
|
|
self.assertRaises(IllegalArgumentException,
|
|
model.assert_vm, "valeur_qcq")
|