Pages/start of tests for EventChecklists

RIGS/tests/test_functional.py

@@ -868,7 +868,6 @@ class TECEventAuthorisationTest(TestCase):
         self.assertIsNotNone(self.event.auth_request_at)
 
 
-
 @screenshot_failure_cls
 class TestHealthAndSafety(BaseRigboardTest):
     def setUp(self):
@@ -876,7 +875,7 @@ class TestHealthAndSafety(BaseRigboardTest):
         self.profile = models.Profile.objects.get_or_create(
             first_name='Test',
             last_name='TEC User',
-            username='eventauthtest',
+            username='eventtest',
             email='teccie@functional.test',
             is_superuser=True  # lazily grant all permissions
         )[0]
@@ -884,28 +883,10 @@ class TestHealthAndSafety(BaseRigboardTest):
                                                      start_date=date.today() + timedelta(days=6),
                                                      description="start future no end",
                                                      purchase_order='TESTPO',
-                                                     person=self.client,
-                                                     auth_request_by=self.profile,
-                                                     auth_request_at=base.create_datetime(2015, 0o6, 0o4, 10, 00),
-                                                     auth_request_to="some@email.address")
-
-        item1 = models.EventItem(
-            event=self.testEvent,
-            name="Test Item 1",
-            cost="10.00",
-            quantity="1",
-            order=1
-        ).save()
-        item2 = models.EventItem(
-            event=self.testEvent,
-            name="Test Item 2",
-            description="Foo",
-            cost="9.72",
-            quantity="3",
-            order=2,
-        ).save()
+                                                     person=self.client)
         self.page = pages.EventDetail(self.driver, self.live_server_url, event_id=self.testEvent.pk).open()
 
+    # TODO Can I loop through all the boolean fields and test them at once?
     def test_ra_creation(self):
         self.page = pages.CreateRiskAssessment(self.driver, self.live_server_url, event_id=self.testEvent.pk).open()
 
@@ -945,3 +926,58 @@ class TestHealthAndSafety(BaseRigboardTest):
 
         self.page.submit()
         self.assertTrue(self.page.success)
+
+        # Test that we can't make another one
+        self.page = pages.CreateRiskAssessment(self.driver, self.live_server_url, event_id=self.testEvent.pk).open()
+        self.assertIn('edit', self.driver.current_url)
+
+    def test_ra_edit(self):
+        # Create assessment to edit
+        self.page = pages.CreateRiskAssessment(self.driver, self.live_server_url, event_id=self.testEvent.pk).open()
+        self.page.nonstandard_equipment = False
+        self.page.nonstandard_use = False
+        self.page.contractors = False
+        self.page.other_companies = False
+        self.page.crew_fatigue = False
+        self.page.general_notes = "There are no notes."
+        self.page.big_power = False
+        self.page.power_mic.search(self.profile.name)
+        self.page.power_mic.toggle()
+        self.assertFalse(self.page.power_mic.is_open)
+        self.page.remove_all_required()
+        self.page.submit()
+        self.page = pages.EditRiskAssessment(self.driver, self.live_server_url, pk=models.RiskAssessment.objects.get(event=self.testEvent.pk).pk).open()
+        self.page.nonstandard_equipment = nse = True
+        self.page.general_notes = gn = "There are some notes, but I've not written them here as that would be helpful"
+        self.page.submit()
+        self.assertTrue(self.page.success)
+        # Check that data is right
+        ra = models.RiskAssessment.objects.get(event=self.testEvent.pk)
+        self.assertEqual(ra.general_notes, gn)
+        self.assertEqual(ra.nonstandard_equipment, nse)
+
+    def test_ec_create_small(self):
+        self.page = pages.CreateEventChecklist(self.driver, self.live_server_url, event_id=self.testEvent.pk).open()
+
+        self.page.safe_parking = True
+        self.page.safe_packing = True
+        self.page.exits = True
+        self.page.trip_hazard = True
+        self.page.warning_signs = True
+        self.page.ear_plugs = True
+        self.page.hs_location = "The Moon"
+        self.page.extinguishers_location = "With the rest of the fire"
+        # If we do this first the search fails, for ... reasons
+        self.page.power_mic.search(self.profile.name)
+        self.page.power_mic.toggle()
+        self.assertFalse(self.page.power_mic.is_open)
+
+        self.page.select_size('Small')
+        self.wait.until(animation_is_finished())
+        self.page.earthing = True
+        self.page.rcds = True
+        self.page.supply_test = True
+        self.page.pat = True
+
+        self.page.submit()
+        self.assertTrue(self.page.success)