Tests: sheet person / organization

1 files changed, 14 insertions, 0 deletions

diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index 8b0b81571..f2bb390bd 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -920,16 +920,30 @@ class OperationTest(TestCase, OperationInitTest):
 
     def test_show(self):
         operation = self.operations[0]
+        source = models.OperationSource.objects.create(
+            operation=operation, title="Source title",
+            source_type=models.SourceType.objects.all()[0]
+        )
         c = Client()
+
         response = c.get(reverse('show-operation', kwargs={'pk': operation.pk}))
         self.assertEqual(response.status_code, 200)
         # empty content when not allowed
         self.assertEqual(response.content, "")
+        response = c.get(reverse('show-operationsource',
+                                 kwargs={'pk': source.pk}))
+        self.assertEqual(response.status_code, 200)
+        # empty content when not allowed
+        self.assertEqual(response.content, "")
 
         c.login(username=self.username, password=self.password)
         response = c.get(reverse('show-operation', kwargs={'pk': operation.pk}))
         self.assertEqual(response.status_code, 200)
         self.assertIn('class="sheet"', response.content)
+        response = c.get(reverse('show-operationsource',
+                                 kwargs={'pk': source.pk}))
+        self.assertEqual(response.status_code, 200)
+        self.assertIn('class="sheet"', response.content)
 
 
 class OperationSearchTest(TestCase, OperationInitTest):