From b3f32d7f355a8bd0ac99f086ed36963b870535be Mon Sep 17 00:00:00 2001 From: Jeremy Stretch Date: Tue, 18 Aug 2020 14:43:21 -0400 Subject: [PATCH] Update test script ObjectVars --- netbox/extras/tests/test_scripts.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/netbox/extras/tests/test_scripts.py b/netbox/extras/tests/test_scripts.py index 482b86b9a..4518548d3 100644 --- a/netbox/extras/tests/test_scripts.py +++ b/netbox/extras/tests/test_scripts.py @@ -152,10 +152,7 @@ class ScriptVariablesTest(TestCase): def test_objectvar(self): class TestScript(Script): - - var1 = ObjectVar( - queryset=DeviceRole.objects.all() - ) + var1 = ObjectVar(model=DeviceRole) # Populate some objects for i in range(1, 6): @@ -173,10 +170,7 @@ class ScriptVariablesTest(TestCase): def test_multiobjectvar(self): class TestScript(Script): - - var1 = MultiObjectVar( - queryset=DeviceRole.objects.all() - ) + var1 = MultiObjectVar(model=DeviceRole) # Populate some objects for i in range(1, 6):