python-tripleoclient/tripleoclient/tests/workflows/test_validations.py
Carlos Camacho c85b3de515 Add unit tests for running and listing validations from the CLI
This submission add unit tests for both listing and running validations
from the CLI.

Change-Id: If728b7535e3aef9439e022ae28e70ccffb97e0dc
Implements: blueprint validation-framework
2019-03-05 15:05:15 +01:00

87 lines
3.7 KiB
Python

# Copyright 2017 Red Hat, 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 osc_lib.tests import utils
from tripleoclient.workflows import validations
class TestValidationsList(utils.TestCommand):
def setUp(self):
super(TestValidationsList, self).setUp()
self.app.client_manager = mock.Mock()
self.app.client_manager.workflow_engine = self.workflow = mock.Mock()
self.tripleoclient = mock.Mock()
self.websocket = mock.Mock()
self.websocket.__enter__ = lambda s: self.websocket
self.websocket.__exit__ = lambda s, *exc: None
self.tripleoclient.messaging_websocket.return_value = self.websocket
self.app.client_manager.tripleoclient = self.tripleoclient
@mock.patch('tripleoclient.workflows.base.wait_for_messages')
@mock.patch('tripleoclient.workflows.base.start_workflow')
def test_list_validations(self, start_wf_mock, messages_mock):
messages_mock.return_value = []
fetch_name = 'tripleo.validations.v1.list'
fetch_input = {
'group_names': ['pre-deployment']
}
validations.list_validations(self.app.client_manager, fetch_input)
start_wf_mock.assert_called_once_with(self.workflow,
fetch_name,
workflow_input=fetch_input)
class TestValidationsRun(utils.TestCommand):
def setUp(self):
super(TestValidationsRun, self).setUp()
self.app.client_manager = mock.Mock()
self.app.client_manager.workflow_engine = self.workflow = mock.Mock()
self.tripleoclient = mock.Mock()
self.websocket = mock.Mock()
self.websocket.__enter__ = lambda s: self.websocket
self.websocket.__exit__ = lambda s, *exc: None
self.tripleoclient.messaging_websocket.return_value = self.websocket
self.app.client_manager.tripleoclient = self.tripleoclient
@mock.patch('tripleoclient.workflows.base.wait_for_messages')
@mock.patch('tripleoclient.workflows.base.start_workflow')
def test_run_validations_group(self, start_wf_mock, messages_mock):
messages_mock.return_value = []
fetch_name = 'tripleo.validations.v1.run_groups'
fetch_input = {
'group_names': ['pre-deployment']
}
validations.run_validations(self.app.client_manager, fetch_input)
start_wf_mock.assert_called_once_with(self.workflow,
fetch_name,
workflow_input=fetch_input)
@mock.patch('tripleoclient.workflows.base.wait_for_messages')
@mock.patch('tripleoclient.workflows.base.start_workflow')
def test_run_validations(self, start_wf_mock, messages_mock):
messages_mock.return_value = []
fetch_name = 'tripleo.validations.v1.run_validations'
fetch_input = {
'validation_names': ['check-ftype']
}
validations.run_validations(self.app.client_manager, fetch_input)
start_wf_mock.assert_called_once_with(self.workflow,
fetch_name,
workflow_input=fetch_input)