NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-17 21:18:16 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

14438 fix tests

Browse Source
This commit is contained in:
Arthur 2024-02-09 14:42:35 -08:00
parent b4c9f93432
commit d9afaf9052
1 changed files with 22 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
netbox/extras/tests/test_api.py
View File

@ -11,7 +11,7 @@ from dcim.models import Device, DeviceRole, DeviceType, Manufacturer, Rack, Loca
from extras.choices import * from extras.choices import *
from extras.models import * from extras.models import *
from extras.reports import Report from extras.reports import Report
from extras.scripts import BooleanVar, IntegerVar, Script, StringVar from extras.scripts import BooleanVar, IntegerVar, Script as PythonClass, StringVar
from utilities.testing import APITestCase, APIViewTestCases from utilities.testing import APITestCase, APIViewTestCases
User = get_user_model() User = get_user_model()
@ -748,7 +748,7 @@ class ConfigTemplateTest(APIViewTestCases.APIViewTestCase):
class ScriptTest(APITestCase): class ScriptTest(APITestCase):
class TestScript(Script): class TestScriptClass(PythonClass):
class Meta: class Meta:
name = "Test script" name = "Test script"
@ -767,27 +767,40 @@ class ScriptTest(APITestCase):
@classmethod @classmethod
def setUpTestData(cls): def setUpTestData(cls):
ScriptModule.objects.create( module = ScriptModule.objects.create(
file_root=ManagedFileRootPathChoices.SCRIPTS, file_root=ManagedFileRootPathChoices.SCRIPTS,
file_path='/var/tmp/script.py' file_path='/var/tmp/script.py'
) )
Script.objects.create(
module=module,
name="Test script",
is_valid=True,
)
def get_test_script(self, *args): # def get_test_script(self, *args):
return ScriptModule.objects.first(), self.TestScript # return ScriptModule.objects.first(), self.TestScriptClass
def python_class(self):
return self.TestScriptClass
def setUp(self): def setUp(self):
super().setUp() super().setUp()
# Monkey-patch the API viewset's _get_script() method to return our test Script above # Monkey-patch the API viewset's _get_script() method to return our test Script above
from extras.api.views import ScriptViewSet from extras.api.views import ScriptViewSet
ScriptViewSet._get_script = self.get_test_script # ScriptViewSet._get_script = self.get_test_script
Script.python_class = self.python_class
def test_get_script(self): def test_get_script(self):
module = ScriptModule.objects.get(
url = reverse('extras-api:script-detail', kwargs={'pk': None}) file_root=ManagedFileRootPathChoices.SCRIPTS,
file_path='/var/tmp/script.py'
)
script = module.scripts.all().first()
url = reverse('extras-api:script-detail', kwargs={'pk': script.pk})
response = self.client.get(url, **self.header) response = self.client.get(url, **self.header)
self.assertEqual(response.data['name'], self.TestScript.Meta.name) self.assertEqual(response.data['name'], self.TestScriptClass.Meta.name)
self.assertEqual(response.data['vars']['var1'], 'StringVar') self.assertEqual(response.data['vars']['var1'], 'StringVar')
self.assertEqual(response.data['vars']['var2'], 'IntegerVar') self.assertEqual(response.data['vars']['var2'], 'IntegerVar')
self.assertEqual(response.data['vars']['var3'], 'BooleanVar') self.assertEqual(response.data['vars']['var3'], 'BooleanVar')