
Pytest includes a fixture that can be used to generate temporary directories. Previously Pegleg had implemented a hombrewed version of a temporary directory fixture. This change removes the homebrewed version and replaces it with the tmpdir fixture. Implement tmpdir fixture in tests Upgrade all testing packages to use the latest features Removes unused imports and organizes import lists Removes mock package requirement and uses unittest.mock, included in python >3.3 Implements a slightly cleaner method to get proxy info Change-Id: If66e1cfba858d5fb8948529deb8fb2d32345f630
242 lines
8.7 KiB
Python
242 lines
8.7 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.
|
|
|
|
from unittest import mock
|
|
|
|
from pegleg.engine import lint
|
|
from pegleg.engine.errorcodes import DECKHAND_DUPLICATE_SCHEMA
|
|
from pegleg.engine.errorcodes import DECKHAND_RENDER_EXCEPTION
|
|
from pegleg.engine.util import deckhand
|
|
from pegleg.engine.util import files
|
|
from pegleg.engine.util.pegleg_managed_document \
|
|
import PeglegManagedSecretsDocument
|
|
|
|
|
|
def test_verify_deckhand_render_site_documents_separately(
|
|
temp_deployment_files):
|
|
expected_documents = {
|
|
'cicd': [
|
|
'global-common', 'global-v1.0', 'cicd-type-common',
|
|
'cicd-type-v1.0', 'cicd-chart', 'cicd-passphrase'
|
|
],
|
|
'lab': [
|
|
'global-common', 'global-v1.0', 'lab-type-common', 'lab-type-v1.0',
|
|
'lab-chart', 'lab-passphrase'
|
|
],
|
|
}
|
|
|
|
with mock.patch('pegleg.engine.util.deckhand.deckhand_render',
|
|
autospec=True) as mock_render:
|
|
mock_render.return_value = (None, [])
|
|
|
|
result = lint._verify_deckhand_render()
|
|
assert result == []
|
|
|
|
expected_sitenames = ['cicd', 'lab']
|
|
sites = files.list_sites()
|
|
|
|
# Verify that both expected site types are listed.
|
|
assert sorted(sites) == expected_sitenames
|
|
# Verify that Deckhand called render twice, once for each site.
|
|
assert mock_render.call_count == 2
|
|
|
|
expected_documents = []
|
|
for sitename in expected_sitenames:
|
|
documents = [
|
|
{
|
|
'data': 'global-common-password',
|
|
'metadata': {
|
|
'layeringDefinition': {
|
|
'abstract': False,
|
|
'layer': 'global'
|
|
},
|
|
'name': 'global-common',
|
|
'schema': 'metadata/Document/v1',
|
|
'storagePolicy': 'cleartext'
|
|
},
|
|
'schema': 'deckhand/Passphrase/v1'
|
|
}, {
|
|
'data': 'global-v1.0-password',
|
|
'metadata': {
|
|
'layeringDefinition': {
|
|
'abstract': False,
|
|
'layer': 'global'
|
|
},
|
|
'name': 'global-v1.0',
|
|
'schema': 'metadata/Document/v1',
|
|
'storagePolicy': 'cleartext'
|
|
},
|
|
'schema': 'deckhand/Passphrase/v1'
|
|
}, {
|
|
'data': '%s-type-common-password' % sitename,
|
|
'metadata': {
|
|
'layeringDefinition': {
|
|
'abstract': False,
|
|
'layer': 'type'
|
|
},
|
|
'name': '%s-type-common' % sitename,
|
|
'schema': 'metadata/Document/v1',
|
|
'storagePolicy': 'cleartext'
|
|
},
|
|
'schema': 'deckhand/Passphrase/v1'
|
|
}, {
|
|
'data': '%s-type-v1.0-password' % sitename,
|
|
'metadata': {
|
|
'layeringDefinition': {
|
|
'abstract': False,
|
|
'layer': 'type'
|
|
},
|
|
'name': '%s-type-v1.0' % sitename,
|
|
'schema': 'metadata/Document/v1',
|
|
'storagePolicy': 'cleartext'
|
|
},
|
|
'schema': 'deckhand/Passphrase/v1'
|
|
}, {
|
|
'data': '%s-chart-password' % sitename,
|
|
'metadata': {
|
|
'layeringDefinition': {
|
|
'abstract': False,
|
|
'layer': 'site'
|
|
},
|
|
'name': '%s-chart' % sitename,
|
|
'schema': 'metadata/Document/v1',
|
|
'storagePolicy': 'cleartext'
|
|
},
|
|
'schema': 'deckhand/Passphrase/v1'
|
|
}, {
|
|
'data': '%s-passphrase-password' % sitename,
|
|
'metadata': {
|
|
'layeringDefinition': {
|
|
'abstract': False,
|
|
'layer': 'site'
|
|
},
|
|
'name': '%s-passphrase' % sitename,
|
|
'schema': 'metadata/Document/v1',
|
|
'storagePolicy': 'cleartext'
|
|
},
|
|
'schema': 'deckhand/Passphrase/v1'
|
|
}
|
|
]
|
|
expected_documents.extend(documents)
|
|
|
|
mock_calls = list(mock_render.mock_calls)
|
|
actual_documents = []
|
|
for mock_call in mock_calls:
|
|
documents = mock_call[2]['documents']
|
|
actual_documents.extend(documents)
|
|
|
|
sort_func = lambda x: x['metadata']['name']
|
|
assert sorted(
|
|
expected_documents, key=sort_func) == sorted(
|
|
actual_documents, key=sort_func)
|
|
|
|
|
|
@mock.patch('pegleg.engine.util.deckhand.deckhand_render')
|
|
def test_verify_deckhand_render_error_handling(mock_render):
|
|
"""
|
|
Verifying deckhand render error handling.
|
|
|
|
Parameters:
|
|
mock_render: Mock render object.
|
|
"""
|
|
exp_dict = {
|
|
'exp1': DECKHAND_DUPLICATE_SCHEMA + ": Duplicate schema specified.\n",
|
|
'exp2': DECKHAND_RENDER_EXCEPTION +
|
|
": An unknown Deckhand exception occurred while trying to render documents\n",
|
|
'exp3': "Generic Error\n"
|
|
}
|
|
# No exception raised
|
|
mock_render.return_value = _return_deckhand_render_errors()
|
|
errors = deckhand.deckhand_render()
|
|
assert errors == []
|
|
# check errors object type
|
|
mock_render.return_value = _return_deckhand_render_errors(1)
|
|
errors = deckhand.deckhand_render()
|
|
assert isinstance(errors, list)
|
|
# check single exception handling
|
|
assert _deckhand_render_exception_msg(errors) == exp_dict['exp1']
|
|
# check multiple exception with tuple type
|
|
mock_render.return_value = _return_deckhand_render_errors(2)
|
|
errors = deckhand.deckhand_render()
|
|
assert _deckhand_render_exception_msg(
|
|
errors) == exp_dict['exp1'] + exp_dict['exp2']
|
|
# check multiple exception with mixed type
|
|
mock_render.return_value = _return_deckhand_render_errors(3)
|
|
errors = deckhand.deckhand_render()
|
|
assert _deckhand_render_exception_msg(
|
|
errors) == exp_dict['exp1'] + exp_dict['exp2'] + exp_dict['exp3']
|
|
|
|
|
|
def test_handle_managed_document():
|
|
not_managed = {
|
|
"schema": "pegleg/FakeSchema/v1",
|
|
"metadata": {
|
|
"schema": "metadata/Document/v1",
|
|
"layeringDefinition": {
|
|
"abstract": "false",
|
|
"layer": "site"
|
|
},
|
|
"name": "fakesite",
|
|
"storagePolicy": "cleartext"
|
|
},
|
|
"data": "None"
|
|
}
|
|
|
|
managed = PeglegManagedSecretsDocument(not_managed).pegleg_document
|
|
|
|
assert lint._handle_managed_document(not_managed) == not_managed
|
|
assert lint._handle_managed_document(managed) == not_managed
|
|
|
|
|
|
def _deckhand_render_exception_msg(errors):
|
|
"""
|
|
Helper function to create deckhand render exception msg.
|
|
|
|
Parameters:
|
|
errors: List of errors provided by deckhand render.
|
|
Returns:
|
|
string: formulated error message.
|
|
"""
|
|
err_msg = ''
|
|
for err in errors:
|
|
if isinstance(err, tuple) and len(err) > 1:
|
|
err_msg += ': '.join(err) + '\n'
|
|
else:
|
|
err_msg += str(err) + '\n'
|
|
return err_msg
|
|
|
|
|
|
def _return_deckhand_render_errors(error_count=0):
|
|
"""
|
|
Helper function to mock deckhand render errors.
|
|
|
|
Parameters:
|
|
error_count: Number of mock errors to return.
|
|
Returns:
|
|
List: List of mock errors as per error_count.
|
|
"""
|
|
errors = []
|
|
if error_count >= 1:
|
|
errors.append(
|
|
(DECKHAND_DUPLICATE_SCHEMA, 'Duplicate schema specified.'))
|
|
if error_count >= 2:
|
|
errors.append(
|
|
(
|
|
DECKHAND_RENDER_EXCEPTION,
|
|
'An unknown Deckhand exception occurred while '
|
|
'trying to render documents'))
|
|
if error_count >= 3:
|
|
errors.append(('Generic Error'))
|
|
return errors
|