| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778 |
- 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):
- #self.browser = webdriver.Firefox()
- self.browser = webdriver.Chrome()
- def tearDown(self):
- self.browser.quit()
- 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_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
- self.assertIn('To-Do', header_text)
- inputbox = self.browser.find_element_by_id('id_new_item')
- self.assertEqual(
- inputbox.get_attribute('placeholder'),
- 'Enter a to-do item'
- )
- # ITEM 1
- inputbox.send_keys('Buy rubber ducks')
- inputbox.send_keys(Keys.ENTER)
- 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)
- 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!')
- 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 super bugz outfit', 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)
|
