Admin: test json import/export

1 files changed, 44 insertions, 0 deletions

diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py
index 34b25a6ec..a4014db18 100644
--- a/ishtar_common/tests.py
+++ b/ishtar_common/tests.py
@@ -1566,6 +1566,50 @@ class AdminGenTypeTest(TestCase):
             finally:
                 f.close()
 
+    def test_json_export(self):
+        for model in self.gen_models:
+            url = '/admin/{}/{}/'.format(self.module_name,
+                                         model.__name__.lower())
+            q = model.objects
+            if not q.count():
+                continue
+            response = self.client.post(
+                url, {'action': '_serialize_action',
+                      '_selected_action': [str(o.pk) for o in q.all()]})
+            self.assertEqual(
+                response.status_code, 200,
+                msg="Can not export as JSON for {}.".format(model))
+            # json content already tested on full export
+
+    def test_json_import(self):
+        for model in self.gen_models:
+            q = model.objects
+            nb = q.count()
+            if not nb:
+                continue
+            base_url = '/admin/{}/{}/'.format(self.module_name,
+                                              model.__name__.lower())
+            response = self.client.post(
+                base_url, {'action': '_serialize_action',
+                           '_selected_action': [str(o.pk) for o in q.all()]})
+            self.assertEqual(
+                response.status_code, 200,
+                msg="Can not export as CSV for {}.".format(model))
+
+            for obj in q.all():
+                obj.delete()
+
+            url = base_url + 'import-from-json/'
+            try:
+                f = io.BytesIO(response.content)
+                response = self.client.post(url, {'json_file': f,
+                                                  'apply': True})
+                self.assertEqual(response.status_code, 302)
+                self.assertRedirects(response, base_url)
+                self.assertEqual(nb, model.objects.count())
+            finally:
+                f.close()
+
     def test_importer_type_duplicate(self):
         model = models.ImporterType
         base_url = '/admin/{}/{}/'.format(self.module_name,