|
@@ -1,8 +1,11 @@
|
|
|
from django.test import LiveServerTestCase
|
|
|
from selenium import webdriver
|
|
|
from selenium.webdriver.common.keys import Keys
|
|
|
+from selenium.common.exceptions import WebDriverException
|
|
|
import time
|
|
|
|
|
|
+MAX_WAIT = 10
|
|
|
+
|
|
|
class NewVisitorTest(LiveServerTestCase):
|
|
|
|
|
|
def setUp(self):
|
|
@@ -12,12 +15,20 @@ class NewVisitorTest(LiveServerTestCase):
|
|
|
def tearDown(self):
|
|
|
self.browser.quit()
|
|
|
|
|
|
- def check_for_row_in_list_table(self, row_text):
|
|
|
- table = self.browser.find_element_by_id('id_list_table')
|
|
|
- rows = table.find_elements_by_tag_name('tr')
|
|
|
- self.assertIn(row_text, [row.text for row in rows])
|
|
|
+ def wait_for_row_in_list_table(self, row_text):
|
|
|
+ start_time = time.time()
|
|
|
+ while True:
|
|
|
+ try:
|
|
|
+ table = self.browser.find_element_by_id('id_list_table')
|
|
|
+ rows = table.find_elements_by_tag_name('tr')
|
|
|
+ self.assertIn(row_text, [row.text for row in rows])
|
|
|
+ return
|
|
|
+ except (AssertionError, WebDriverException) as e:
|
|
|
+ if (time.time() - start_time) > MAX_WAIT:
|
|
|
+ raise e
|
|
|
+ time.sleep(0.1)
|
|
|
|
|
|
- def test_can_start_a_list_and_retrieve_it_later(self):
|
|
|
+ def test_can_start_a_list_for_one_user(self):
|
|
|
self.browser.get(self.live_server_url)
|
|
|
self.assertIn('To-Do', self.browser.title)
|
|
|
header_text = self.browser.find_element_by_tag_name('h1').text
|
|
@@ -30,14 +41,38 @@ class NewVisitorTest(LiveServerTestCase):
|
|
|
# ITEM 1
|
|
|
inputbox.send_keys('Buy rubber ducks')
|
|
|
inputbox.send_keys(Keys.ENTER)
|
|
|
- time.sleep(1)
|
|
|
- self.check_for_row_in_list_table('1: Buy rubber ducks')
|
|
|
+ self.wait_for_row_in_list_table('1: Buy rubber ducks')
|
|
|
# ITEM 2
|
|
|
inputbox = self.browser.find_element_by_id('id_new_item')
|
|
|
inputbox.send_keys('Buy squirrel suits')
|
|
|
inputbox.send_keys(Keys.ENTER)
|
|
|
- time.sleep(1)
|
|
|
- self.check_for_row_in_list_table('1: Buy rubber ducks')
|
|
|
- self.check_for_row_in_list_table('2: Buy squirrel suits')
|
|
|
+ self.wait_for_row_in_list_table('1: Buy rubber ducks')
|
|
|
+ self.wait_for_row_in_list_table('2: Buy squirrel suits')
|
|
|
# Done?
|
|
|
- self.fail('Finish the test!')
|
|
|
+ #self.fail('Finish the test!')
|
|
|
+
|
|
|
+ def test_multi_users_can_start_list_diff_urls(self):
|
|
|
+ self.browser.get(self.live_server_url)
|
|
|
+ ib = self.browser.find_element_by_id('id_new_item')
|
|
|
+ ib.send_keys('Buy super bugz outfit')
|
|
|
+ ib.send_keys(Keys.ENTER)
|
|
|
+ self.wait_for_row_in_list_table('1: Buy super bugz outfit')
|
|
|
+ list_url = self.browser.current_url
|
|
|
+ self.assertRegex(list_url, '/lists/.+')
|
|
|
+ # NEW USER
|
|
|
+ self.browser.quit()
|
|
|
+ self.browser = webdriver.Chrome()
|
|
|
+ self.browser.get(self.live_server_url)
|
|
|
+ page_text = self.browser.find_element_by_tag_name('body').text
|
|
|
+ self.assertNotIn('Buy peacock feathers', page_text)
|
|
|
+ self.assertNotIn('make a fly', page_text)
|
|
|
+ inputbox = self.browser.find_element_by_id('id_new_item')
|
|
|
+ inputbox.send_keys('Buy milk')
|
|
|
+ inputbox.send_keys(Keys.ENTER)
|
|
|
+ self.wait_for_row_in_list_table('1: Buy milk')
|
|
|
+ francis_list_url = self.browser.current_url
|
|
|
+ self.assertRegex(francis_list_url, '/lists/.+')
|
|
|
+ self.assertNotEqual(francis_list_url, list_url)
|
|
|
+ page_text = self.browser.find_element_by_tag_name('body').text
|
|
|
+ self.assertNotIn('Buy peacock feathers', page_text)
|
|
|
+ self.assertIn('Buy milk', page_text)
|