Files
rally/tests/benchmark/test_utils.py
Mikhail Dubov 9bd4a1e8f2 Start refactoring scenario runner
We should remove spaghetti from code and make it safer.

The current implementation has several issues:
1) If something went wrong (which can be the case) we won't cleanup the cloud completely;
2) There is a lot of spaghetti around catching exceptions and preparing the cloud;
3) There will be even more spaghetti when we try to extend current functionality
with (stress exectution/non admin users/another loads and cloud pre configuration);
4) We have a lot of IFs that switch what kind of exectuion we will use.
This should be done via classes and inheritance IMO.

This patch fixes some of the issues mentioned above:
1) Add Context class that allows to create/delete tmp users and tenants;
2) Add Context class that makes all cleanups. For users if they are present
and for admins if they are present;
3) Changed algorithm of deleting resources (try-catch) e.g. if 1 user
wasn't deleted => we should try to continue and delete others;
4) Add some todos;
5) Rewrite ScenarioRunner.run() method to use these context classes;
6) Rewrite UTs accordingly and also use the correct @mock.patch(...) syntax.

Implements: Blueprint scenario-runner-refactoring
Implements: Blueprint unit-tests-refactoring

Change-Id: I5c6c082ed30390bc000a2371f0dfca31994910cb
2014-02-05 22:32:58 +04:00

104 lines
3.6 KiB
Python

# Copyright 2013: Mirantis 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.
import mock
from tests import fakes
from tests import test
from rally.benchmark import utils
from rally import exceptions
class BenchmarkUtilsTestCase(test.TestCase):
def test_resource_is(self):
is_active = utils.resource_is("ACTIVE")
self.assertTrue(is_active(fakes.FakeResource(status="active")))
self.assertTrue(is_active(fakes.FakeResource(status="aCtIvE")))
self.assertFalse(is_active(fakes.FakeResource(status="ERROR")))
def test_is_none(self):
self.assertTrue(utils.is_none(None))
self.assertFalse(utils.is_none(0))
self.assertFalse(utils.is_none(""))
self.assertFalse(utils.is_none("afafa"))
def test_false(self):
self.assertFalse(utils.false(None))
def test_infinite_run_args(self):
args = ("a", "b", "c", "d", 123)
for i, real_args in enumerate(utils.infinite_run_args(args)):
self.assertEqual((i,) + args, real_args)
if i == 10:
break
def test_create_openstack_clients(self):
#TODO(boris-42): Implement this method
pass
def test_manager_list_sizes(self):
manager = fakes.FakeManager()
def lst():
return [1] * 10
manager.list = lst
manager_list_size = utils.manager_list_size([5])
self.assertFalse(manager_list_size(manager))
manager_list_size = utils.manager_list_size([10])
self.assertTrue(manager_list_size(manager))
def test_get_from_manager(self):
get_from_manager = utils.get_from_manager()
manager = fakes.FakeManager()
resource = fakes.FakeResource(manager=manager)
manager._cache(resource)
self.assertEqual(get_from_manager(resource), resource)
def test_get_from_manager_in_error_state(self):
get_from_manager = utils.get_from_manager()
manager = fakes.FakeManager()
resource = fakes.FakeResource(manager=manager, status="ERROR")
manager._cache(resource)
self.assertRaises(exceptions.GetResourceFailure,
get_from_manager, resource)
def test_get_from_manager_not_found(self):
get_from_manager = utils.get_from_manager()
manager = mock.MagicMock()
resource = fakes.FakeResource(manager=manager, status="ERROR")
class NotFoundException(Exception):
http_status = 404
manager.get = mock.MagicMock(side_effect=NotFoundException)
self.assertRaises(exceptions.GetResourceFailure,
get_from_manager, resource)
def test_get_from_manager_http_exception(self):
get_from_manager = utils.get_from_manager()
manager = mock.MagicMock()
resource = fakes.FakeResource(manager=manager, status="ERROR")
class HTTPException(Exception):
pass
manager.get = mock.MagicMock(side_effect=HTTPException)
self.assertRaises(exceptions.GetResourceFailure,
get_from_manager, resource)