simple_selenium.py

'''
Dependecies :
selenium webdriver -> pip install selenium
'''
import unittest
from selenium import webdriver

class SimpleTest(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Safari()
    
    def test_case(self):
        web = "https://facebook.com"
        driver = self.driver
        driver.get(web)
        uname = driver.find_element_by_id("email")
        pwd = driver.find_element_by_id("pass")
        submit = driver.find_element_by_id("u_0_a")

        uname.send_keys("afif.hendrawan@polinema.ac.id")
        pwd.send_keys("polinemaJoss")
        submit.click()
        assert "loginbutton" in driver.page_source
    
    def take_down(self):
        self.driver.close()
    

if __name__ == "__main__":
    unittest.main()