rally/tests/unit/objects/test_task.py
temujin e9f5ceeb33 add 'rally task list' filters
Add ability to filter tasks by status and/or deployment.

Change-Id: I663bddf73a4d5be1b24e712ddcde9efdf494f005
2014-12-02 13:16:53 +02:00

156 lines
6.2 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.
"""Tests for db.task layer."""
import json
import mock
from rally import consts
from rally import objects
from tests.unit import test
class TaskTestCase(test.TestCase):
def setUp(self):
super(TaskTestCase, self).setUp()
self.task = {
'uuid': '00ef46a2-c5b8-4aea-a5ca-0f54a10cbca1',
'status': consts.TaskStatus.INIT,
'failed': False,
'verification_log': '',
}
@mock.patch('rally.objects.task.db.task_create')
def test_init_with_create(self, mock_create):
mock_create.return_value = self.task
task = objects.Task(failed=True)
mock_create.assert_called_once_with({'failed': True})
self.assertEqual(task['uuid'], self.task['uuid'])
@mock.patch('rally.objects.task.db.task_create')
def test_init_without_create(self, mock_create):
task = objects.Task(task=self.task)
self.assertFalse(mock_create.called)
self.assertEqual(task['uuid'], self.task['uuid'])
@mock.patch('rally.objects.task.db.task_get')
def test_get(self, mock_get):
mock_get.return_value = self.task
task = objects.Task.get(self.task['uuid'])
mock_get.assert_called_once_with(self.task['uuid'])
self.assertEqual(task['uuid'], self.task['uuid'])
@mock.patch('rally.objects.task.db.task_delete')
@mock.patch('rally.objects.task.db.task_create')
def test_create_and_delete(self, mock_create, mock_delete):
mock_create.return_value = self.task
task = objects.Task()
task.delete()
mock_delete.assert_called_once_with(self.task['uuid'], status=None)
@mock.patch('rally.objects.task.db.task_delete')
@mock.patch('rally.objects.task.db.task_create')
def test_create_and_delete_status(self, mock_create, mock_delete):
mock_create.return_value = self.task
task = objects.Task()
task.delete(status=consts.TaskStatus.FINISHED)
mock_delete.assert_called_once_with(self.task['uuid'],
status=consts.TaskStatus.FINISHED)
@mock.patch('rally.objects.task.db.task_delete')
def test_delete_by_uuid(self, mock_delete):
objects.Task.delete_by_uuid(self.task['uuid'])
mock_delete.assert_called_once_with(self.task['uuid'], status=None)
@mock.patch('rally.objects.task.db.task_delete')
def test_delete_by_uuid_status(self, mock_delete):
objects.Task.delete_by_uuid(self.task['uuid'],
consts.TaskStatus.FINISHED)
mock_delete.assert_called_once_with(self.task['uuid'],
status=consts.TaskStatus.FINISHED)
@mock.patch("rally.objects.task.db.task_list",
return_value=[{"uuid": "a",
"created_at": "b",
"status": "c",
"failed": True,
"tag": "d",
"deployment_name": "some_name"}])
def list(self, mock_db_task_list):
tasks = objects.Task.list(status="somestatus")
mock_db_task_list.assert_called_once_with("somestatus", None)
self.assertIs(type(tasks), list)
self.assertIsInstance(tasks[0], objects.Task)
self.assertEqual(mock_db_task_list.return_value["uuis"],
tasks[0]["uuid"])
@mock.patch('rally.objects.deploy.db.task_update')
@mock.patch('rally.objects.task.db.task_create')
def test_update(self, mock_create, mock_update):
mock_create.return_value = self.task
mock_update.return_value = {'opt': 'val2'}
deploy = objects.Task(opt='val1')
deploy._update({'opt': 'val2'})
mock_update.assert_called_once_with(self.task['uuid'],
{'opt': 'val2'})
self.assertEqual(deploy['opt'], 'val2')
@mock.patch('rally.objects.task.db.task_update')
def test_update_status(self, mock_update):
mock_update.return_value = self.task
task = objects.Task(task=self.task)
task.update_status(consts.TaskStatus.FINISHED)
mock_update.assert_called_once_with(
self.task['uuid'],
{'status': consts.TaskStatus.FINISHED},
)
@mock.patch('rally.objects.task.db.task_update')
def test_update_verification_log(self, mock_update):
mock_update.return_value = self.task
task = objects.Task(task=self.task)
task.update_verification_log({"a": "fake"})
mock_update.assert_called_once_with(
self.task['uuid'],
{'verification_log': json.dumps({"a": "fake"})}
)
@mock.patch("rally.objects.task.db.task_result_get_all_by_uuid",
return_value="foo_results")
def test_get_results(self, mock_get):
task = objects.Task(task=self.task)
results = task.get_results()
mock_get.assert_called_once_with(self.task["uuid"])
self.assertEqual(results, "foo_results")
@mock.patch('rally.objects.task.db.task_result_create')
def test_append_results(self, mock_append_results):
task = objects.Task(task=self.task)
task.append_results('opt', 'val')
mock_append_results.assert_called_once_with(self.task['uuid'],
'opt', 'val')
@mock.patch('rally.objects.task.db.task_update')
def test_set_failed(self, mock_update):
mock_update.return_value = self.task
task = objects.Task(task=self.task)
task.set_failed()
mock_update.assert_called_once_with(
self.task['uuid'],
{'failed': True, 'status': 'failed', 'verification_log': '""'},
)