Fix iCal tests

This commit is contained in:
David Taylor
2017-06-21 12:04:30 +01:00
committed by Tom Price
parent ee930aa86a
commit 0aec836b1a

View File

@@ -941,16 +941,18 @@ class IcalTest(LiveServerTestCase):
response = c.get(icalUrl)
self.assertEqual(200, response.status_code)
# content = response.content.decode('utf-8')
# Check has entire file
self.assertIn("BEGIN:VCALENDAR", response.content)
self.assertIn("END:VCALENDAR", response.content)
self.assertContains(response, "BEGIN:VCALENDAR")
self.assertContains(response, "END:VCALENDAR")
expectedIn = [1, 2, 3, 5, 6, 7, 10, 11, 12, 13, 15, 16, 17]
for test in range(1, 18):
if test in expectedIn:
self.assertIn("TE E" + str(test) + " ", response.content)
self.assertContains(response, "TE E" + str(test) + " ")
else:
self.assertNotIn("TE E" + str(test) + " ", response.content)
self.assertNotContains(response, "TE E" + str(test) + " ")
# Only dry hires
self.browser.find_element_by_xpath("//input[@value='rig']").click()
@@ -963,9 +965,9 @@ class IcalTest(LiveServerTestCase):
expectedIn = [10, 11, 12, 13]
for test in range(1, 18):
if test in expectedIn:
self.assertIn("TE E" + str(test) + " ", response.content)
self.assertContains(response, "TE E" + str(test) + " ")
else:
self.assertNotIn("TE E" + str(test) + " ", response.content)
self.assertNotContains(response, "TE E" + str(test) + " ")
# Only provisional rigs
self.browser.find_element_by_xpath("//input[@value='rig']").click()
@@ -979,9 +981,9 @@ class IcalTest(LiveServerTestCase):
expectedIn = [1, 2]
for test in range(1, 18):
if test in expectedIn:
self.assertIn("TE E" + str(test) + " ", response.content)
self.assertContains(response, "TE E" + str(test) + " ")
else:
self.assertNotIn("TE E" + str(test) + " ", response.content)
self.assertNotContains(response, "TE E" + str(test) + " ")
# Only cancelled non-rigs
self.browser.find_element_by_xpath("//input[@value='rig']").click()
@@ -996,9 +998,9 @@ class IcalTest(LiveServerTestCase):
expectedIn = [18]
for test in range(1, 18):
if test in expectedIn:
self.assertIn("TE E" + str(test) + " ", response.content)
self.assertContains(response, "TE E" + str(test) + " ")
else:
self.assertNotIn("TE E" + str(test) + " ", response.content)
self.assertNotContains(response, "TE E" + str(test) + " ")
# Nothing selected
self.browser.find_element_by_xpath("//input[@value='non-rig']").click()
@@ -1011,9 +1013,9 @@ class IcalTest(LiveServerTestCase):
expectedIn = []
for test in range(1, 18):
if test in expectedIn:
self.assertIn("TE E" + str(test) + " ", response.content)
self.assertContains(response, "TE E" + str(test) + " ")
else:
self.assertNotIn("TE E" + str(test) + " ", response.content)
self.assertNotContains(response, "TE E" + str(test) + " ")
# Wow - that was a lot of tests