1
0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00

Update test script ObjectVars

This commit is contained in:
Jeremy Stretch
2020-08-18 14:43:21 -04:00
parent c9c79dabef
commit 0d9fc309d5

View File

@@ -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):