Archaeological tests: fix REST test by adding authentification

1 files changed, 23 insertions, 1 deletions

diff --git a/archaeological_files/tests.py b/archaeological_files/tests.py
index b4ad7d4b5..b30c39dc1 100644
--- a/archaeological_files/tests.py
+++ b/archaeological_files/tests.py
@@ -36,9 +36,15 @@ class FileTest(TestCase):
                 '../ishtar_common/fixtures/initial_data-fr.json']
     model = models.File
 
+    def login_as_superuser(self):
+        self.client.login(username='username', password='tralala')
+
     def setUp(self):
         self.extra_models, self.model_list = {}, []
-        self.user, created = User.objects.get_or_create(username='username')
+        self.user, created = User.objects.get_or_create(username='username',
+                                                        is_superuser=True)
+        self.user.set_password('tralala')
+        self.user.save()
         self.o_user, created = User.objects.get_or_create(username='ousername')
         person_type = PersonType(label=u'Test person type',
                                  txt_idx='test_person', available=True)
@@ -136,6 +142,14 @@ class FileTest(TestCase):
                              {'numeric_reference':self.item.numeric_reference})
         self.assertEqual(response.status_code, 200)
         data = json.loads(response.content)
+        # not allowed -> no data
+        self.assertTrue(not data)
+
+        self.login_as_superuser()
+        response = self.client.post('/get-file/',
+                             {'numeric_reference':self.item.numeric_reference})
+        self.assertEqual(response.status_code, 200)
+        data = json.loads(response.content)
         self.assertTrue('records' in data)
         self.assertTrue(data['records'] == 1)
 
@@ -150,6 +164,14 @@ class FileTest(TestCase):
                      {'numeric_reference':self.item.numeric_reference, 'old':1})
         self.assertEqual(response.status_code, 200)
         data = json.loads(response.content)
+        # not allowed -> no data
+        self.assertTrue(not data)
+
+        self.login_as_superuser()
+        response = self.client.post('/get-file/',
+                     {'numeric_reference':self.item.numeric_reference, 'old':1})
+        self.assertEqual(response.status_code, 200)
+        data = json.loads(response.content)
         self.assertTrue('records' in data)
         self.assertTrue(data['records'] == 1)
         self.assertEqual(data['rows'][0]['internal_reference'], initial_ref)