168 lines
6.7 KiB
Python
168 lines
6.7 KiB
Python
import pylibyaml # noqa: F401 # patch pyyaml to use libyaml bindings
|
|
import yaml
|
|
import jsonschema
|
|
import pkg_resources
|
|
import os
|
|
import shutil
|
|
import pytest
|
|
|
|
from jsonschema.exceptions import ValidationError
|
|
|
|
|
|
class BaseSchemaValidationTest(object):
|
|
def _test_validate(self, schema, expect_failure, input_files, input):
|
|
"""validates input yaml against schema.
|
|
|
|
:param schema: schema yaml file
|
|
:param expect_failure: should the validation pass or fail.
|
|
:param input_files: pytest fixture used to access the test input files
|
|
:param input: test input yaml doc filename"""
|
|
schema_dir = pkg_resources.resource_filename('drydock_provisioner',
|
|
'schemas')
|
|
schema_filename = os.path.join(schema_dir, schema)
|
|
schema_file = open(schema_filename, 'r')
|
|
schema = yaml.safe_load(schema_file)
|
|
|
|
input_file = input_files.join(input)
|
|
instance_file = open(str(input_file), 'r')
|
|
instance = yaml.safe_load(instance_file)
|
|
|
|
if expect_failure:
|
|
with pytest.raises(ValidationError):
|
|
jsonschema.validate(instance['spec'], schema['data'])
|
|
else:
|
|
jsonschema.validate(instance['spec'], schema['data'])
|
|
|
|
|
|
class TestValidation(BaseSchemaValidationTest):
|
|
def test_validate_baremetalNode(self, input_files):
|
|
self._test_validate('baremetalNode.yaml', False, input_files,
|
|
"baremetalNode.yaml")
|
|
|
|
def test_validate_baremetalNode2(self, input_files):
|
|
self._test_validate('baremetalNode.yaml', False, input_files,
|
|
"baremetalNode2.yaml")
|
|
|
|
def test_invalidate_baremetalNode(self, input_files):
|
|
self._test_validate('baremetalNode.yaml', True, input_files,
|
|
"invalid_baremetalNode.yaml")
|
|
|
|
def test_invalidate_baremetalNode2(self, input_files):
|
|
self._test_validate('baremetalNode.yaml', True, input_files,
|
|
"invalid_baremetalNode2.yaml")
|
|
|
|
def test_validate_hardwareProfile(self, input_files):
|
|
self._test_validate('hardwareProfile.yaml', False, input_files,
|
|
"hardwareProfile.yaml")
|
|
|
|
def test_invalidate_hardwareProfile(self, input_files):
|
|
self._test_validate('hardwareProfile.yaml', True, input_files,
|
|
"invalid_hardwareProfile.yaml")
|
|
|
|
def test_validate_hostProfile(self, input_files):
|
|
self._test_validate('hostProfile.yaml', False, input_files,
|
|
"hostProfile.yaml")
|
|
|
|
def test_validate_hostProfile2(self, input_files):
|
|
self._test_validate('hostProfile.yaml', False, input_files,
|
|
"hostProfile2.yaml")
|
|
|
|
def test_invalidate_hostProfile(self, input_files):
|
|
self._test_validate('hostProfile.yaml', True, input_files,
|
|
"invalid_hostProfile.yaml")
|
|
|
|
def test_invalidate_hostProfile2(self, input_files):
|
|
self._test_validate('hostProfile.yaml', True, input_files,
|
|
"invalid_hostProfile2.yaml")
|
|
|
|
def test_validate_network(self, input_files):
|
|
self._test_validate('network.yaml', False, input_files, "network.yaml")
|
|
|
|
def test_validate_network2(self, input_files):
|
|
self._test_validate('network.yaml', False, input_files,
|
|
"network2.yaml")
|
|
|
|
def test_validate_network3(self, input_files):
|
|
self._test_validate('network.yaml', False, input_files,
|
|
"network3.yaml")
|
|
|
|
def test_validate_network4(self, input_files):
|
|
self._test_validate('network.yaml', False, input_files,
|
|
"network4.yaml")
|
|
|
|
def test_validate_network5(self, input_files):
|
|
self._test_validate('network.yaml', False, input_files,
|
|
"network5.yaml")
|
|
|
|
def test_invalidate_network(self, input_files):
|
|
self._test_validate('network.yaml', True, input_files,
|
|
"invalid_network.yaml")
|
|
|
|
def test_invalidate_network2(self, input_files):
|
|
self._test_validate('network.yaml', True, input_files,
|
|
"invalid_network2.yaml")
|
|
|
|
def test_invalidate_network3(self, input_files):
|
|
self._test_validate('network.yaml', True, input_files,
|
|
"invalid_network3.yaml")
|
|
|
|
def test_invalidate_network4(self, input_files):
|
|
self._test_validate('network.yaml', True, input_files,
|
|
"invalid_network4.yaml")
|
|
|
|
def test_invalidate_network5(self, input_files):
|
|
self._test_validate('network.yaml', True, input_files,
|
|
"invalid_network5.yaml")
|
|
|
|
def test_validate_networkLink(self, input_files):
|
|
self._test_validate('networkLink.yaml', False, input_files,
|
|
"networkLink.yaml")
|
|
|
|
def test_validate_networkLink2(self, input_files):
|
|
self._test_validate('networkLink.yaml', False, input_files,
|
|
"networkLink2.yaml")
|
|
|
|
def test_validate_networkLink3(self, input_files):
|
|
self._test_validate('networkLink.yaml', False, input_files,
|
|
"networkLink3.yaml")
|
|
|
|
def test_invalidate_networkLink(self, input_files):
|
|
self._test_validate('networkLink.yaml', True, input_files,
|
|
"invalid_networkLink.yaml")
|
|
|
|
def test_invalidate_networkLink2(self, input_files):
|
|
self._test_validate('networkLink.yaml', True, input_files,
|
|
"invalid_networkLink2.yaml")
|
|
|
|
def test_invalidate_networkLink3(self, input_files):
|
|
self._test_validate('networkLink.yaml', True, input_files,
|
|
"invalid_networkLink3.yaml")
|
|
|
|
def test_validate_region(self, input_files):
|
|
self._test_validate('region.yaml', False, input_files, "region.yaml")
|
|
|
|
def test_invalidate_region(self, input_files):
|
|
self._test_validate('region.yaml', True, input_files,
|
|
"invalid_region.yaml")
|
|
|
|
def test_validate_rack(self, input_files):
|
|
self._test_validate('rack.yaml', False, input_files, "rack.yaml")
|
|
|
|
def test_invalidate_rack(self, input_files):
|
|
self._test_validate('rack.yaml', True, input_files,
|
|
"invalid_rack.yaml")
|
|
|
|
@pytest.fixture(scope='module')
|
|
def input_files(self, tmpdir_factory, request):
|
|
tmpdir = tmpdir_factory.mktemp('data')
|
|
samples_dir = os.path.dirname(str(
|
|
request.fspath)) + "/" + "../yaml_samples"
|
|
samples = os.listdir(samples_dir)
|
|
|
|
for f in samples:
|
|
src_file = samples_dir + "/" + f
|
|
dst_file = str(tmpdir) + "/" + f
|
|
shutil.copyfile(src_file, dst_file)
|
|
|
|
return tmpdir
|