285 lines
11 KiB
Python
285 lines
11 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.
|
|
"""Test Tasks API"""
|
|
from falcon import testing
|
|
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
import json
|
|
import logging
|
|
|
|
from drydock_provisioner import policy
|
|
from drydock_provisioner.control.api import start_api
|
|
import drydock_provisioner.objects as objects
|
|
import drydock_provisioner.objects.fields as hd_fields
|
|
|
|
import falcon
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class TestTasksApiUnit(object):
|
|
def test_get_tasks_id_resp(self, falcontest):
|
|
url = '/api/v1.0/tasks/11111111-1111-1111-1111-111111111111'
|
|
hdr = self.get_standard_header()
|
|
|
|
result = falcontest.simulate_get(url, headers=hdr)
|
|
|
|
assert result.status == falcon.HTTP_200
|
|
response_json = json.loads(result.text)
|
|
assert response_json[
|
|
'task_id'] == '11111111-1111-1111-1111-111111111111'
|
|
try:
|
|
response_json['build_data']
|
|
key_error = False
|
|
except KeyError:
|
|
key_error = True
|
|
assert key_error
|
|
try:
|
|
response_json['subtask_errors']
|
|
key_error = False
|
|
except KeyError:
|
|
key_error = True
|
|
assert key_error
|
|
|
|
def test_get_tasks_id_subtaskerror_noerrors_resp(self, falcontest):
|
|
url = '/api/v1.0/tasks/11111111-1111-1111-1111-111111111111'
|
|
hdr = self.get_standard_header()
|
|
|
|
result = falcontest.simulate_get(
|
|
url, headers=hdr, query_string='subtaskerrors=true')
|
|
|
|
assert result.status == falcon.HTTP_200
|
|
response_json = json.loads(result.text)
|
|
assert response_json[
|
|
'task_id'] == '11111111-1111-1111-1111-111111111111'
|
|
assert response_json['subtask_errors'] == {}
|
|
|
|
def test_get_tasks_id_subtaskerror_errors_resp(self, falcontest):
|
|
url = '/api/v1.0/tasks/11111111-1111-1111-1111-111111111113'
|
|
hdr = self.get_standard_header()
|
|
|
|
result = falcontest.simulate_get(
|
|
url, headers=hdr, query_string='subtaskerrors=true')
|
|
|
|
assert result.status == falcon.HTTP_200
|
|
response_json = json.loads(result.text)
|
|
assert response_json[
|
|
'task_id'] == '11111111-1111-1111-1111-111111111113'
|
|
assert response_json['subtask_errors'][
|
|
'11111111-1111-1111-1111-111111111116']['details'][
|
|
'errorCount'] == 1
|
|
|
|
def test_get_tasks_id_builddata_resp(self, falcontest):
|
|
url = '/api/v1.0/tasks/11111111-1111-1111-1111-111111111111'
|
|
hdr = self.get_standard_header()
|
|
|
|
result = falcontest.simulate_get(
|
|
url, headers=hdr, query_string='builddata=true')
|
|
|
|
LOG.debug(result.text)
|
|
assert result.status == falcon.HTTP_200
|
|
response_json = json.loads(result.text)
|
|
assert response_json['build_data']
|
|
try:
|
|
response_json['subtask_errors']
|
|
key_error = False
|
|
except KeyError:
|
|
key_error = True
|
|
assert key_error
|
|
|
|
def test_get_tasks_id_builddata_subtaskerrors_resp(self, falcontest):
|
|
url = '/api/v1.0/tasks/11111111-1111-1111-1111-111111111111'
|
|
hdr = self.get_standard_header()
|
|
|
|
result = falcontest.simulate_get(
|
|
url, headers=hdr, query_string='builddata=true&subtaskerrors=true')
|
|
|
|
LOG.debug(result.text)
|
|
assert result.status == falcon.HTTP_200
|
|
response_json = json.loads(result.text)
|
|
assert response_json['build_data']
|
|
assert response_json['subtask_errors'] == {}
|
|
|
|
def test_get_tasks_id_layers_resp(self, falcontest):
|
|
url = '/api/v1.0/tasks/11111111-1111-1111-1111-111111111113'
|
|
hdr = self.get_standard_header()
|
|
|
|
result = falcontest.simulate_get(
|
|
url, headers=hdr, query_string='layers=2')
|
|
|
|
LOG.debug(result.text)
|
|
assert result.status == falcon.HTTP_200
|
|
response_json = json.loads(result.text)
|
|
init_task_id = '11111111-1111-1111-1111-111111111113'
|
|
sub_task_id_1 = '11111111-1111-1111-1111-111111111114'
|
|
sub_task_id_2 = '11111111-1111-1111-1111-111111111115'
|
|
assert response_json['init_task_id'] == init_task_id
|
|
assert response_json[init_task_id]['task_id'] == init_task_id
|
|
assert response_json[sub_task_id_1]['task_id'] == sub_task_id_1
|
|
assert response_json[sub_task_id_2]['task_id'] == sub_task_id_2
|
|
try:
|
|
response_json['11111111-1111-1111-1111-111111111116']
|
|
key_error = False
|
|
except KeyError:
|
|
key_error = True
|
|
assert key_error
|
|
|
|
def test_get_tasks_id_layers_all_noerrors_resp(self, falcontest):
|
|
url = '/api/v1.0/tasks/11111111-1111-1111-1111-111111111113'
|
|
hdr = self.get_standard_header()
|
|
|
|
result = falcontest.simulate_get(
|
|
url, headers=hdr, query_string='layers=-1')
|
|
|
|
LOG.debug(result.text)
|
|
assert result.status == falcon.HTTP_200
|
|
response_json = json.loads(result.text)
|
|
init_task_id = '11111111-1111-1111-1111-111111111113'
|
|
sub_task_id_1 = '11111111-1111-1111-1111-111111111114'
|
|
sub_task_id_2 = '11111111-1111-1111-1111-111111111115'
|
|
assert response_json['init_task_id'] == init_task_id
|
|
assert response_json[init_task_id]['task_id'] == init_task_id
|
|
assert response_json[sub_task_id_1]['task_id'] == sub_task_id_1
|
|
assert response_json[sub_task_id_2]['task_id'] == sub_task_id_2
|
|
try:
|
|
response_json['11111111-1111-1111-1111-111111111116']
|
|
key_error = False
|
|
except KeyError:
|
|
key_error = True
|
|
assert key_error is False
|
|
try:
|
|
response_json['subtask_errors']
|
|
key_error = False
|
|
except KeyError:
|
|
key_error = True
|
|
assert key_error
|
|
|
|
def test_get_tasks_id_layers_all_errors_resp(self, falcontest):
|
|
url = '/api/v1.0/tasks/11111111-1111-1111-1111-111111111113'
|
|
hdr = self.get_standard_header()
|
|
|
|
result = falcontest.simulate_get(
|
|
url, headers=hdr, query_string='layers=-1&subtaskerrors=true')
|
|
|
|
LOG.debug(result.text)
|
|
assert result.status == falcon.HTTP_200
|
|
response_json = json.loads(result.text)
|
|
init_task_id = '11111111-1111-1111-1111-111111111113'
|
|
sub_task_id_1 = '11111111-1111-1111-1111-111111111114'
|
|
sub_task_id_2 = '11111111-1111-1111-1111-111111111115'
|
|
assert response_json['init_task_id'] == init_task_id
|
|
assert response_json[init_task_id]['task_id'] == init_task_id
|
|
assert response_json[sub_task_id_1]['task_id'] == sub_task_id_1
|
|
assert response_json[sub_task_id_2]['task_id'] == sub_task_id_2
|
|
try:
|
|
response_json['11111111-1111-1111-1111-111111111116']
|
|
key_error = False
|
|
except KeyError:
|
|
key_error = True
|
|
assert key_error is False
|
|
assert response_json['subtask_errors'][
|
|
'11111111-1111-1111-1111-111111111116']['details'][
|
|
'errorCount'] == 1
|
|
|
|
def test_input_not_found(self, falcontest):
|
|
url = '/api/v1.0/tasks/11111111-1111-1111-1111-111111111112'
|
|
hdr = self.get_standard_header()
|
|
|
|
result = falcontest.simulate_get(url, headers=hdr)
|
|
|
|
LOG.debug(result.text)
|
|
assert result.status == falcon.HTTP_404
|
|
|
|
@pytest.fixture()
|
|
def falcontest(self, drydock_state, deckhand_ingester,
|
|
deckhand_orchestrator, mock_get_build_data, mock_get_task):
|
|
"""Create a test harness for the Falcon API framework."""
|
|
policy.policy_engine = policy.DrydockPolicy()
|
|
policy.policy_engine.register_policy()
|
|
|
|
return testing.TestClient(
|
|
start_api(
|
|
state_manager=drydock_state,
|
|
ingester=deckhand_ingester,
|
|
orchestrator=deckhand_orchestrator))
|
|
|
|
def get_standard_header(self):
|
|
hdr = {
|
|
'Content-Type': 'application/json',
|
|
'X-IDENTITY-STATUS': 'Confirmed',
|
|
'X-USER-NAME': 'Test',
|
|
'X-ROLES': 'admin'
|
|
}
|
|
return hdr
|
|
|
|
|
|
@pytest.fixture()
|
|
def mock_get_task(drydock_state):
|
|
def side_effect(*args):
|
|
task_id = str(args[0])
|
|
LOG.debug(task_id)
|
|
# Basic task
|
|
if task_id == '11111111-1111-1111-1111-111111111111':
|
|
new_task = objects.Task()
|
|
new_task.task_id = '11111111-1111-1111-1111-111111111111'
|
|
new_task.result = objects.TaskStatus()
|
|
new_task.result.set_status(hd_fields.ActionResult.Failure)
|
|
new_task.result.add_status_msg(
|
|
msg='Test', error=True, ctx_type='N/A', ctx='N/A')
|
|
return new_task
|
|
# Task not found
|
|
if task_id == '11111111-1111-1111-1111-111111111112':
|
|
return None
|
|
# Task layers
|
|
if task_id == '11111111-1111-1111-1111-111111111113':
|
|
new_task = objects.Task()
|
|
new_task.task_id = '11111111-1111-1111-1111-111111111113'
|
|
new_task.subtask_id_list = [
|
|
'11111111-1111-1111-1111-111111111114',
|
|
'11111111-1111-1111-1111-111111111115'
|
|
]
|
|
return new_task
|
|
if task_id == '11111111-1111-1111-1111-111111111114':
|
|
new_task = objects.Task()
|
|
new_task.task_id = '11111111-1111-1111-1111-111111111114'
|
|
return new_task
|
|
if task_id == '11111111-1111-1111-1111-111111111115':
|
|
new_task = objects.Task()
|
|
new_task.task_id = '11111111-1111-1111-1111-111111111115'
|
|
new_task.subtask_id_list = [
|
|
'11111111-1111-1111-1111-111111111116',
|
|
'11111111-1111-1111-1111-111111111117'
|
|
]
|
|
return new_task
|
|
if task_id == '11111111-1111-1111-1111-111111111116':
|
|
new_task = objects.Task()
|
|
new_task.task_id = '11111111-1111-1111-1111-111111111116'
|
|
new_task.result = objects.TaskStatus()
|
|
new_task.result.set_status(hd_fields.ActionResult.Failure)
|
|
new_task.result.add_status_msg(
|
|
msg='Test', error=True, ctx_type='N/A', ctx='N/A')
|
|
LOG.debug('error_count')
|
|
LOG.debug(new_task.result.error_count)
|
|
return new_task
|
|
LOG.debug('returning None')
|
|
return None
|
|
|
|
drydock_state.real_get_task = drydock_state.get_task
|
|
drydock_state.get_task = Mock(side_effect=side_effect)
|
|
|
|
yield
|
|
drydock_state.get_task = Mock(wraps=None, side_effect=None)
|
|
drydock_state.get_task = drydock_state.real_get_task
|