67 lines
2.7 KiB
Python
67 lines
2.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 pegleg.engine.util import definition
|
|
from tests.unit.fixtures import create_tmp_deployment_files
|
|
|
|
|
|
class TestSiteDefinitionHelpers(object):
|
|
def _test_documents_for_site(self, sitename):
|
|
documents = definition.documents_for_site(sitename)
|
|
global_documents = []
|
|
site_documents = []
|
|
|
|
for document in documents:
|
|
name = document["metadata"]["name"]
|
|
# Assert that the document is either global level or a relevant
|
|
# site document.
|
|
assert name.startswith("global") or name.startswith(sitename)
|
|
|
|
if name.startswith("global"):
|
|
global_documents.append(document)
|
|
elif name.startswith(sitename):
|
|
site_documents.append(document)
|
|
else:
|
|
raise AssertionError("Unexpected document retrieved by "
|
|
"`documents_for_site`: %s" % document)
|
|
|
|
# Assert that documents from both levels appear.
|
|
assert global_documents
|
|
assert site_documents
|
|
|
|
return global_documents + site_documents
|
|
|
|
def test_documents_for_site(self, create_tmp_deployment_files):
|
|
self._test_documents_for_site("cicd")
|
|
self._test_documents_for_site("lab")
|
|
|
|
def test_documents_for_each_site(self, create_tmp_deployment_files):
|
|
documents_by_site = definition.documents_for_each_site()
|
|
sort_func = lambda x: x['metadata']['name']
|
|
|
|
# Validate that both expected site documents are found.
|
|
assert 2 == len(documents_by_site)
|
|
assert "cicd" in documents_by_site
|
|
assert "lab" in documents_by_site
|
|
|
|
cicd_documents = self._test_documents_for_site("cicd")
|
|
lab_documents = self._test_documents_for_site("lab")
|
|
|
|
# Validate that each set of site documents matches the same set of
|
|
# documents returned by ``documents_for_site`` for that site.
|
|
assert (sorted(cicd_documents, key=sort_func) == sorted(
|
|
documents_by_site["cicd"], key=sort_func))
|
|
assert (sorted(lab_documents, key=sort_func) == sorted(
|
|
documents_by_site["lab"], key=sort_func))
|