Просмотр исходного кода

Extend standard view test case to validate built-in CSV export

Jeremy Stretch 6 лет назад
Родитель
Сommit
5386ed438e
1 измененных файлов с 6 добавлено и 0 удалено
  1. 6 0
      netbox/utilities/testing/testcases.py

+ 6 - 0
netbox/utilities/testing/testcases.py

@@ -137,6 +137,12 @@ class StandardTestCases:
             response = self.client.get(self._get_url('list'))
             self.assertHttpStatus(response, 200)
 
+            # Built-in CSV export
+            if hasattr(self.model, 'csv_headers'):
+                response = self.client.get('{}?export'.format(self._get_url('list')))
+                self.assertHttpStatus(response, 200)
+                self.assertEqual(response.get('Content-Type'), 'text/csv')
+
         @override_settings(EXEMPT_VIEW_PERMISSIONS=[])
         def test_get_object(self):
             instance = self.model.objects.first()