Expanded browser control in .functional_tests

functional_tests.py

@@ -1,7 +1,17 @@
+import unittest
 from selenium import webdriver
 
-browser = webdriver.Firefox()
+class NewVisitorTest(unittest.TestCase):
-browser.get("http://localhost:8000")
+    def setUp(self):
+        self.browser = webdriver.Firefox()
+    
+    def tearDown(self):
+        self.browser.quit()
 
-assert "Congratulations!" in browser.title
+    def test_can_start_a_todo_list(self):
-print("OK")
+        self.browser.get("http://localhost:8000")
+        self.assertIn("To-Do", self.browser.title)
+        self.fail("Finish the test!")
+
+if __name__ == "__main__":
+    unittest.main()