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

Add InterfaceTestCase for virtual machines

Browse Source
This commit is contained in:
Jeremy Stretch
2020-02-04 16:40:18 -05:00
parent c1639b7781
commit 3e79b9d26a
2 changed files with 87 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
netbox/utilities/testing/testcases.py
View File

@@ -104,15 +104,22 @@ class StandardTestCases:
if self.model is None:
raise Exception("Test case requires model to be defined")
def _get_base_url(self):
"""
Return the base format for a URL for the test's model. Override this to test for a model which belongs
to a different app (e.g. testing Interfaces within the virtualization app).
"""
return '{}:{}_{{}}'.format(
self.model._meta.app_label,
self.model._meta.model_name
)
def _get_url(self, action, instance=None):
"""
Return the URL name for a specific action. An instance must be specified for
get/edit/delete views.
"""
url_format = '{}:{}_{{}}'.format(
self.model._meta.app_label,
self.model._meta.model_name
)
url_format = self._get_base_url()
if action in ('list', 'add', 'import', 'bulk_edit', 'bulk_delete'):
return reverse(url_format.format(action))