1 files changed, 64 insertions, 0 deletions
diff --git a/archaeological_files/tests.py b/archaeological_files/tests.py
index 51a09acd2..fc9483830 100644
--- a/archaeological_files/tests.py
+++ b/archaeological_files/tests.py
@@ -246,8 +246,68 @@ class FileTest(TestCase, FileInit):
         self.assertEqual(response.status_code, 200)
         self.assertIn('class="card sheet"', response.content.decode())
 
+    def test_preventive_price_agreement(self):
+        pk = self.item.pk
+        self.login_as_superuser()
+        url = reverse("file-edit-preventive", kwargs={"pk": pk})
+        response = self.client.get(url)
+        price_url = reverse("file-edit-preventive-price", kwargs={"pk": pk})
+        self.assertRedirects(
+            response,
+            price_url,
+            status_code=302,
+            target_status_code=200,
+            fetch_redirect_response=True,
+        )
+        self.item.price_agreement_id = models.PriceAgreement.objects.all()[0]
+        self.item.save()
+        response = self.client.get(url)
+        self.assertEqual(response.status_code, 200)
+
+        job1 = models.Job.objects.all()[0]
+        models.PreventiveFileJob.objects.create(
+            file=self.item,
+            job=job1
+        )
+
+        cost1 = models.EquipmentServiceCost.objects.filter(
+            equipment_service_type=models.EquipmentServiceType.objects.all()[0]
+        ).all()[0]
+        models.PreventiveFileEquipmentServiceCost.objects.create(
+            file=self.item,
+            equipment_service_cost=cost1,
+            quantity_by_day_planned=1,
+            days_planned=1
+        )
+        q_equip = models.PreventiveFileEquipmentServiceCost.objects.filter(
+            file=self.item,
+        )
+        q_job = models.PreventiveFileJob.objects.filter(
+            file=self.item,
+        )
+        self.assertEqual(q_equip.count(), 1)
+        self.assertEqual(q_job.count(), 1)
+
+        p = models.PriceAgreement.objects.create(label="New price", txt_idx="new-price")
+
+        # test remove of associated jobs
+        response = self.client.post(price_url, {"price_agreement": p.pk})
+
+        self.assertRedirects(
+            response,
+            url,
+            status_code=302,
+            target_status_code=200,
+            fetch_redirect_response=True,
+        )
+        self.assertEqual(q_equip.count(), 0)
+        self.assertEqual(q_job.count(), 0)
+
+
     def test_preventive(self):
         pk = self.item.pk
+        self.item.price_agreement_id = models.PriceAgreement.objects.all()[0]
+        self.item.save()
         url = reverse("file-edit-preventive", kwargs={"pk": pk})
         response = self.client.get(url)
         self.assertEqual(response.status_code, 302)
@@ -346,6 +406,8 @@ class FileTest(TestCase, FileInit):
 
     def test_preventive_add_default(self):
         pk = self.item.pk
+        self.item.price_agreement_id = models.PriceAgreement.objects.all()[0]
+        self.item.save()
         url = reverse("file-edit-preventive", kwargs={"pk": pk})
         self.login_as_superuser()
         for job in models.Job.objects.all()[:5]:
@@ -378,6 +440,8 @@ class FileTest(TestCase, FileInit):
 
     def test_preventive_copy_planned(self):
         pk = self.item.pk
+        self.item.price_agreement_id = models.PriceAgreement.objects.all()[0]
+        self.item.save()
         url = reverse("file-edit-preventive", kwargs={"pk": pk})
         self.login_as_superuser()