forked from mirrors/thatmattlove-hyperglass
22 lines
692 B
Python
22 lines
692 B
Python
"""Test model utilities."""
|
|
|
|
# Third Party
|
|
import pytest
|
|
|
|
# Local
|
|
from ..util import check_legacy_fields
|
|
|
|
|
|
def test_check_legacy_fields():
|
|
test1 = {"name": "Device A", "nos": "juniper"}
|
|
test1_expected = {"name": "Device A", "platform": "juniper"}
|
|
test2 = {"name": "Device B", "platform": "juniper"}
|
|
test3 = {"name": "Device C"}
|
|
assert set(check_legacy_fields("Device", **test1).keys()) == set(
|
|
test1_expected.keys()
|
|
), "legacy field not replaced"
|
|
assert set(check_legacy_fields("Device", **test2).keys()) == set(
|
|
test2.keys()
|
|
), "new field not left unmodified"
|
|
with pytest.raises(ValueError):
|
|
check_legacy_fields("Device", **test3)
|