Skip to content

Commit fe50aa1

Browse files
committed
fixed import to avoid content type does not exist
1 parent 899b240 commit fe50aa1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

netbox/extras/tests/test_api.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88

99
from core.choices import ManagedFileRootPathChoices
1010
from dcim.models import Device, DeviceRole, DeviceType, Manufacturer, Rack, Location, RackRole, Site
11-
from extras.api.views import ReportViewSet, ScriptViewSet
1211
from extras.models import *
1312
from extras.reports import Report
1413
from extras.scripts import BooleanVar, IntegerVar, Script, StringVar
@@ -579,6 +578,7 @@ def setUp(self):
579578
super().setUp()
580579

581580
# Monkey-patch the API viewset's _get_report() method to return our test Report above
581+
from extras.api.views import ReportViewSet
582582
ReportViewSet._get_report = self.get_test_report
583583

584584
def test_get_report(self):
@@ -621,6 +621,7 @@ def setUp(self):
621621
super().setUp()
622622

623623
# Monkey-patch the API viewset's _get_script() method to return our test Script above
624+
from extras.api.views import ScriptViewSet
624625
ScriptViewSet._get_script = self.get_test_script
625626

626627
def test_get_script(self):

0 commit comments

Comments
 (0)