3629245b0c
Bypass a failure due to the health of Drydock on update_site or deploy_site if the continue-on-fail param is enetered. Adds unit tests Changed operator to be more easily testable Removed the K8s Health Check for preflight checks. We will add the functionalities back once we have a clearer view on what should be checked and validated for the workflow. Change-Id: Idd6d6b18d762a0284f2041248faa4040c78def3f
102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
# Copyright 2018 AT&T Intellectual Property. All other 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
|
|
import pytest
|
|
from requests.models import Response
|
|
|
|
from airflow.exceptions import AirflowException
|
|
from shipyard_airflow.plugins.ucp_preflight_check_operator import (
|
|
UcpHealthCheckOperator)
|
|
|
|
ucp_components = [
|
|
'armada',
|
|
'deckhand',
|
|
'kubernetesprovisioner',
|
|
'physicalprovisioner',
|
|
'shipyard']
|
|
|
|
|
|
def test_drydock_health_skip_update_site():
|
|
"""
|
|
Ensure that an error is not thrown due to Drydock health failing during
|
|
update_site or deploy site
|
|
"""
|
|
|
|
expected_log = ('Drydock did not pass health check. Continuing '
|
|
'as "continue-on-fail" option is enabled.')
|
|
|
|
req = Response()
|
|
req.status_code = None
|
|
|
|
action_info = {
|
|
"dag_id": "update_site",
|
|
"parameters": {"continue-on-fail": "true"}
|
|
}
|
|
|
|
op = UcpHealthCheckOperator(task_id='test')
|
|
op.action_info = action_info
|
|
|
|
with mock.patch('logging.info', autospec=True) as mock_logger:
|
|
op.log_health('physicalprovisioner', req)
|
|
mock_logger.assert_called_with(expected_log)
|
|
|
|
action_info = {
|
|
"dag_id": "deploy_site",
|
|
"parameters": {"continue-on-fail": "true"}
|
|
}
|
|
|
|
with mock.patch('logging.info', autospec=True) as mock_logger:
|
|
op.log_health('physicalprovisioner', req)
|
|
mock_logger.assert_called_with(expected_log)
|
|
|
|
|
|
def test_failure_log_health():
|
|
""" Ensure an error is thrown on failure for all components.
|
|
"""
|
|
action_info = {
|
|
"dag_id": "update_site",
|
|
"parameters": {"something-else": "true"}
|
|
}
|
|
|
|
req = Response()
|
|
req.status_code = None
|
|
|
|
op = UcpHealthCheckOperator(task_id='test')
|
|
op.action_info = action_info
|
|
|
|
for i in ucp_components:
|
|
with pytest.raises(AirflowException) as expected_exc:
|
|
op.log_health(i, req)
|
|
assert "Health check failed" in str(expected_exc)
|
|
|
|
|
|
def test_success_log_health():
|
|
""" Ensure 204 gives correct response for all components
|
|
"""
|
|
action_info = {
|
|
"dag_id": "deploy_site",
|
|
"parameters": {"something-else": "true"}
|
|
}
|
|
|
|
req = Response()
|
|
req.status_code = 204
|
|
|
|
op = UcpHealthCheckOperator(task_id='test')
|
|
op.action_info = action_info
|
|
|
|
for i in ucp_components:
|
|
with mock.patch('logging.info', autospec=True) as mock_logger:
|
|
op.log_health(i, req)
|
|
mock_logger.assert_called_with('%s is alive and healthy', i)
|