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