45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
from selenium import webdriver
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.common.keys import Keys
|
|
import time
|
|
import unittest
|
|
|
|
class NewVisitorTest(unittest.TestCase):
|
|
# Helper methods
|
|
def setUp(self):
|
|
self.browser = webdriver.Firefox()
|
|
|
|
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])
|
|
|
|
# Test methods
|
|
def test_can_start_a_todo_list(self):
|
|
self.browser.get("http://localhost:8000")
|
|
|
|
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')
|
|
|
|
inputbox.send_keys('Buy peacock feathers')
|
|
|
|
inputbox.send_keys(Keys.ENTER)
|
|
time.sleep(1)
|
|
self.check_for_row_in_list_table('1: Buy peacock feathers')
|
|
|
|
inputbox = self.browser.find_element(By.ID, 'id-new-item')
|
|
inputbox.send_keys('Use peacock feathers to make a fly')
|
|
inputbox.send_keys(Keys.ENTER)
|
|
time.sleep(1)
|
|
self.check_for_row_in_list_table('2: Use peacock feathers to make a fly')
|
|
self.check_for_row_in_list_table('1: Buy peacock feathers')
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main() |