Jelajahi Sumber

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

Jeremy Stretch 6 tahun lalu
induk
melakukan
5386ed438e
1 mengubah file dengan 6 tambahan dan 0 penghapusan
  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'))
             response = self.client.get(self._get_url('list'))
             self.assertHttpStatus(response, 200)
             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=[])
         @override_settings(EXEMPT_VIEW_PERMISSIONS=[])
         def test_get_object(self):
         def test_get_object(self):
             instance = self.model.objects.first()
             instance = self.model.objects.first()