Files
deb-python-fuelclient/fuelclient/tests/test_environment.py
Dmitry Shulyak c64c3c659c Add start parameter for tasks execution
--start parameter will be used to specify start endpoint
for tasks execution based on graph traversal

Examples:
  fuel node --node 2,3 --start netconfig
  fuel node --node 2,3 --start netconfig --end neutron

Patch-Set in nailgun: I4d7c9ec825d9534ce3aafd218716e6045915e5a0

implements blueprint granular-deployment-based-on-tasks

Change-Id: Iccc9555f532858e9df67d4dbd10856c1e7f8afd9
2015-02-10 13:44:33 +02:00

64 lines
2.2 KiB
Python

# -*- coding: utf-8 -*-
#
# Copyright 2014 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 mock
from fuelclient.objects.environment import Environment
from fuelclient.tests import base
class TestEnvironmentOstf(base.UnitTestCase):
def setUp(self):
super(TestEnvironmentOstf, self).setUp()
self.env = Environment(None)
@mock.patch.object(Environment.connection, 'post_request', mock.Mock(
return_value=[
{'id': 1},
{'id': 2}, ]))
def test_run_test_sets(self):
self.assertEqual(self.env._testruns_ids, [])
testruns = self.env.run_test_sets(['sanity', 'ha'])
self.assertEqual(len(testruns), 2)
self.assertIn(1, self.env._testruns_ids)
self.assertIn(2, self.env._testruns_ids)
@mock.patch.object(Environment.connection, 'get_request', mock.Mock(
side_effect=[
{'id': 1, 'status': 'running'},
{'id': 2, 'status': 'finished'}, ]))
def test_get_state_of_tests(self):
self.env._testruns_ids.extend([1, 2])
tests = self.env.get_state_of_tests()
self.env.connection.get_request.assert_has_calls([
mock.call('testruns/1', ostf=True),
mock.call('testruns/2', ostf=True)])
self.assertEqual(tests, [
{'id': 1, 'status': 'running'},
{'id': 2, 'status': 'finished'}])
@mock.patch('fuelclient.client.requests')
def test_get_deployment_tasks_with_end(self, mrequests):
end = 'task1'
self.env.get_deployment_tasks(end=end)
kwargs = mrequests.get.call_args[1]
self.assertEqual(kwargs['params'], {'start': None, 'end': 'task1'})