devicetype-library/tests/test_dupes.py
2020-11-24 13:10:31 -06:00

62 lines
1.8 KiB
Python

import glob
import json
import pytest
import yaml
KNOWN_MODELS = {}
def _get_definition_files():
"""
Return a list of all definition files.
"""
return [f for f in glob.glob("device-types/*/*", recursive=True)]
def test_environment():
"""
Run basic sanity checks on the environment to ensure tests are running correctly.
"""
# Validate that definition files exist
assert _get_definition_files(), "No definition files found!"
@pytest.mark.parametrize("file_path", _get_definition_files())
def test_dupes(file_path):
# Check file extension
assert file_path.split('.')[-1] in ('yaml', 'yml'), f"Invalid file extension: {file_path}"
# Read file
with open(file_path) as definition_file:
content = definition_file.read()
# Check for trailing newline
assert content[-1] == '\n', "Missing trailing newline"
# Load YAML data
definition = yaml.load(content, Loader=yaml.SafeLoader)
slug = definition.get('slug')
if KNOWN_MODELS.get(slug, None) is not None:
pytest.fail(f"{file_path} is a duplicate device_type for {slug}", False)
KNOWN_MODELS[slug] = definition.get('model')
def test_components(type):
KNOWN = []
components = definition.get(type, None)
if components is not None:
for idx,component in enumerate(components):
name = component.get('name')
if name in KNOWN:
pytest.fail(f'Duplicate {type} "{name}" in {file_path}', False)
KNOWN.append(name)
test_components('interfaces')
test_components('device-bays')
test_components('front-ports')
test_components('rear-ports')
test_components('power-ports')
test_components('power-outlets')
test_components('console-ports')
test_components('console-server-ports')