class EraserTest(unittest.TestCase):
def setUp(self):
self.eraser = Eraser(email, password)
def test_setup(self):
self.assertIsInstance(self.eraser.driver, webdriver.Firefox)
def test_login_function(self):
"""
Login and check the title of the page
:return:
"""
self.eraser.login()
self.assertTrue('Facebook' in self.eraser.driver.title)
def tearDown(self):
self.eraser.quit()
def setUp(self):
self.eraser = Eraser(email, password)