Initial work on event create test reimpl

RIGS/tests/test_rigs.py

@@ -14,14 +14,15 @@ from selenium.webdriver.support.ui import WebDriverWait
 from PyRIGS.tests.base import animation_is_finished
 from PyRIGS.tests import base
 from RIGS.tests import regions
-from datetime import date, time, datetime, timedelta
+import datetime
+from datetime import date, time, timedelta
 from django.utils import timezone
 
 
 class TestRigboard(AutoLoginTest):
     def setUp(self):
         super().setUp()
-        client = models.Person.objects.create(name='Duplicate Test Person', email='duplicate@functional.test')
+        client = models.Person.objects.create(name='Rigboard Test Person', email='rigboard@functional.test')
         self.vatrate = models.VatRate.objects.create(start_at='2014-03-05', rate=0.20, comment='test1')
         self.testEvent = models.Event.objects.create(name="TE E1", status=models.Event.PROVISIONAL,
                                                 start_date=date.today() + timedelta(days=6),
@@ -64,3 +65,56 @@ class TestRigboard(AutoLoginTest):
         self.page.add()
         self.assertIn('create', self.driver.current_url)
         # Ideally get a response object to assert 200 on
+
+
+class TestEventCreate(AutoLoginTest):
+    def setUp(self):
+        super().setUp()
+        self.client = models.Person.objects.create(name='Creation Test Person', email='god@functional.test')
+        self.vatrate = models.VatRate.objects.create(start_at='2014-03-05', rate=0.20, comment='test1')
+        self.page = pages.CreateEvent(self.driver, self.live_server_url).open()
+        self.wait = WebDriverWait(self.driver, 5)
+
+    def test_rig_creation(self):
+        self.wait.until(animation_is_finished())
+        self.assertFalse(self.page.is_expanded)
+        self.page.select_event_type("Rig")
+        self.wait.until(animation_is_finished())
+        self.assertTrue(self.page.is_expanded)
+
+        self.page.person_selector.toggle()
+        self.assertTrue(self.page.person_selector.is_open)
+        self.page.person_selector.search(self.client.name)
+        self.page.person_selector.set_option(self.client.name, True)
+        # TODO This should not be necessary, normally closes automatically
+        self.page.person_selector.toggle()
+        self.assertFalse(self.page.person_selector.is_open)
+
+        self.page.name = "Test Rig"
+        self.page.start_date = datetime.date(2015, 1, 1)
+        self.page.start_time = datetime.time(10)
+        self.page.end_date = datetime.date(2015, 1, 10)
+        self.page.access_at = datetime.datetime(2015, 1, 1, 9)
+        self.page.dry_hire = True
+        self.page.status = "Booked"
+        self.page.collected_by = "Fred"
+        self.page.po = "1234"
+        self.page.notes = "A note!"
+
+        # TODO Test validation with some wrong data
+        self.page.submit()
+        self.assertTrue(self.page.success)
+
+    # TODO Testing of internal rig (approval system interface)
+
+    def test_non_rig_creation(self):
+        self.wait.until(animation_is_finished())
+        self.assertFalse(self.page.is_expanded)
+        self.page.select_event_type("Non-Rig")
+        self.wait.until(animation_is_finished())
+        self.assertTrue(self.page.is_expanded)
+
+        # self.assertFalse(self.page.person_selector.is_displayed())
+
+    def test_subhire_creation(self):
+        pass