|
|
@@ -624,8 +624,11 @@ class ExportTemplateTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
|
|
|
export_templates = (
|
|
|
ExportTemplate(name='Export Template 1', template_code='TESTING', description='foobar1'),
|
|
|
- ExportTemplate(name='Export Template 2', template_code='TESTING', description='foobar2'),
|
|
|
- ExportTemplate(name='Export Template 3', template_code='TESTING'),
|
|
|
+ ExportTemplate(
|
|
|
+ name='Export Template 2', template_code='TESTING', description='foobar2',
|
|
|
+ file_name='export_template_2', file_extension='nagios',
|
|
|
+ ),
|
|
|
+ ExportTemplate(name='Export Template 3', template_code='TESTING', file_name='export_filename'),
|
|
|
)
|
|
|
ExportTemplate.objects.bulk_create(export_templates)
|
|
|
for i, et in enumerate(export_templates):
|
|
|
@@ -635,6 +638,9 @@ class ExportTemplateTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
params = {'q': 'foobar1'}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
|
|
|
+ params = {'q': 'export_filename'}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
+
|
|
|
def test_name(self):
|
|
|
params = {'name': ['Export Template 1', 'Export Template 2']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
@@ -649,6 +655,20 @@ class ExportTemplateTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
params = {'description': ['foobar1', 'foobar2']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_file_name(self):
|
|
|
+ params = {'file_name': ['export_filename']}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
+
|
|
|
+ def test_file_extension(self):
|
|
|
+ params = {'file_extension': ['nagios']}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
+
|
|
|
+ params = {'file_name': ['export_template_2'], 'file_extension': ['nagios']}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
+
|
|
|
+ params = {'file_name': 'export_filename', 'file_extension': ['nagios']}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 0)
|
|
|
+
|
|
|
|
|
|
class ImageAttachmentTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
queryset = ImageAttachment.objects.all()
|