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

Clean up filterset tests

Browse Source
This commit is contained in:
Jeremy Stretch 2023-11-29 08:55:12 -05:00
parent f880b8f924
commit afc1e789d2
1 changed files with 31 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
netbox/extras/tests/test_filtersets.py
View File

@ -210,7 +210,9 @@ class EventRuleTestCase(TestCase, BaseFilterSetTests):
@classmethod @classmethod
def setUpTestData(cls): def setUpTestData(cls):
content_types = ContentType.objects.filter(model__in=['region', 'site', 'rack', 'location', 'device', 'circuit']) content_types = ContentType.objects.filter(
model__in=['region', 'site', 'rack', 'location', 'device']
)
webhooks = ( webhooks = (
Webhook( Webhook(
@ -219,23 +221,30 @@ class EventRuleTestCase(TestCase, BaseFilterSetTests):
), ),
Webhook( Webhook(
name='Webhook 2', name='Webhook 2',
payload_url='http://example.com/?1', payload_url='http://example.com/?2',
), ),
Webhook( Webhook(
name='Webhook 3', name='Webhook 3',
payload_url='http://example.com/?1', payload_url='http://example.com/?3',
), ),
) )
Webhook.objects.bulk_create(webhooks) Webhook.objects.bulk_create(webhooks)
module = ScriptModule.objects.create( scripts = (
file_root=ManagedFileRootPathChoices.SCRIPTS, ScriptModule(
file_path='/var/tmp/script.py' file_root=ManagedFileRootPathChoices.SCRIPTS,
file_path='/var/tmp/script1.py'
),
ScriptModule(
file_root=ManagedFileRootPathChoices.SCRIPTS,
file_path='/var/tmp/script2.py'
),
) )
ScriptModule.objects.bulk_create(scripts)
event_rules = ( event_rules = (
EventRule( EventRule(
name='EventRule 1', name='Event Rule 1',
action_object=webhooks[0], action_object=webhooks[0],
enabled=True, enabled=True,
type_create=True, type_create=True,
@ -246,8 +255,8 @@ class EventRuleTestCase(TestCase, BaseFilterSetTests):
action_type=EventRuleActionChoices.WEBHOOK, action_type=EventRuleActionChoices.WEBHOOK,
), ),
EventRule( EventRule(
name='EventRule 2', name='Event Rule 2',
action_object=webhooks[0], action_object=webhooks[1],
enabled=True, enabled=True,
type_create=False, type_create=False,
type_update=True, type_update=True,
@ -257,8 +266,8 @@ class EventRuleTestCase(TestCase, BaseFilterSetTests):
action_type=EventRuleActionChoices.WEBHOOK, action_type=EventRuleActionChoices.WEBHOOK,
), ),
EventRule( EventRule(
name='EventRule 3', name='Event Rule 3',
action_object=webhooks[0], action_object=webhooks[2],
enabled=False, enabled=False,
type_create=False, type_create=False,
type_update=False, type_update=False,
@ -268,36 +277,25 @@ class EventRuleTestCase(TestCase, BaseFilterSetTests):
action_type=EventRuleActionChoices.WEBHOOK, action_type=EventRuleActionChoices.WEBHOOK,
), ),
EventRule( EventRule(
name='EventRule 4', name='Event Rule 4',
action_object=webhooks[0], action_object=scripts[0],
enabled=False, enabled=False,
type_create=False, type_create=False,
type_update=False, type_update=False,
type_delete=False, type_delete=False,
type_job_start=True, type_job_start=True,
type_job_end=False, type_job_end=False,
action_type=EventRuleActionChoices.WEBHOOK, action_type=EventRuleActionChoices.SCRIPT,
), ),
EventRule( EventRule(
name='EventRule 5', name='Event Rule 5',
action_object=webhooks[0], action_object=scripts[1],
enabled=False, enabled=False,
type_create=False, type_create=False,
type_update=False, type_update=False,
type_delete=False, type_delete=False,
type_job_start=False, type_job_start=False,
type_job_end=True, type_job_end=True,
action_type=EventRuleActionChoices.WEBHOOK,
),
EventRule(
name='EventRule 6',
action_object=webhooks[0],
enabled=False,
type_create=False,
type_update=False,
type_delete=False,
type_job_start=False,
type_job_end=False,
action_type=EventRuleActionChoices.SCRIPT, action_type=EventRuleActionChoices.SCRIPT,
), ),
) )
@ -307,10 +305,9 @@ class EventRuleTestCase(TestCase, BaseFilterSetTests):
event_rules[2].content_types.add(content_types[2]) event_rules[2].content_types.add(content_types[2])
event_rules[3].content_types.add(content_types[3]) event_rules[3].content_types.add(content_types[3])
event_rules[4].content_types.add(content_types[4]) event_rules[4].content_types.add(content_types[4])
event_rules[5].content_types.add(content_types[5])
def test_name(self): def test_name(self):
params = {'name': ['EventRule 1', 'EventRule 2']} params = {'name': ['Event Rule 1', 'Event Rule 2']}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2) self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_content_types(self): def test_content_types(self):
@ -320,14 +317,16 @@ class EventRuleTestCase(TestCase, BaseFilterSetTests):
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1) self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
def test_action_type(self): def test_action_type(self):
params = {'action_type': EventRuleActionChoices.SCRIPT} params = {'action_type': [EventRuleActionChoices.WEBHOOK]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1) self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)
params = {'action_type': [EventRuleActionChoices.SCRIPT]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_enabled(self): def test_enabled(self):
params = {'enabled': True} params = {'enabled': True}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2) self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
params = {'enabled': False} params = {'enabled': False}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4) self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)
def test_type_create(self): def test_type_create(self):
params = {'type_create': True} params = {'type_create': True}