From 6bc6cf0e1c343fbb4988f67249c1b041012c4dfb Mon Sep 17 00:00:00 2001 From: thatmattlove Date: Fri, 17 Sep 2021 01:11:18 -0700 Subject: [PATCH] Fix missing items from `Device.platform` refactor --- hyperglass/plugins/_output.py | 2 +- hyperglass/plugins/tests/test_bgp_route_juniper.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/hyperglass/plugins/_output.py b/hyperglass/plugins/_output.py index 42e0a8e..06e5115 100644 --- a/hyperglass/plugins/_output.py +++ b/hyperglass/plugins/_output.py @@ -8,7 +8,7 @@ from hyperglass.log import log from hyperglass.types import Series # Local -from ._base import DirectivePlugin, PlatformPlugin, HyperglassPlugin +from ._base import PlatformPlugin, DirectivePlugin, HyperglassPlugin if TYPE_CHECKING: # Project diff --git a/hyperglass/plugins/tests/test_bgp_route_juniper.py b/hyperglass/plugins/tests/test_bgp_route_juniper.py index 8d409ea..405e6b7 100644 --- a/hyperglass/plugins/tests/test_bgp_route_juniper.py +++ b/hyperglass/plugins/tests/test_bgp_route_juniper.py @@ -33,9 +33,9 @@ def _tester(sample: str): address="127.0.0.1", network={"name": "Test Network", "display_name": "Test Network"}, credential={"username": "", "password": ""}, - type="juniper", + platform="juniper", structured_output=True, - commands=[{"id": "test", "name": "Test", "rules": []}], + directives=[{"id": "test", "name": "Test", "rules": []}], ) # Override has_directives method for testing.