selenium unittest in python

To write Selenium unittests in Python, you need to follow these steps:

Import unittest and Selenium modules


main.py
import unittest
from selenium import webdriver
47 chars
3 lines

Create a class for your test case that inherits from Python's unittest.TestCase


main.py
class MyTestCase(unittest.TestCase):
37 chars
2 lines

In the setUp() method of the class, create an instance of the Selenium WebDriver


main.py
def setUp(self):
    self.driver = webdriver.Chrome()
54 chars
3 lines

Write test methods, each starting with the word test_


main.py
def test_search_in_google(self):
    self.driver.get("https://www.google.com")
    self.driver.find_element_by_name("q").send_keys("selenium")
    self.driver.find_element_by_name("btnK").click()
    assert "selenium" in self.driver.page_source
245 chars
6 lines

In the tearDown() method of the class, close the web driver instance


main.py
def tearDown(self):
    self.driver.close()
44 chars
3 lines

Finally, use Python's unittest.main() to run the tests in the command line


main.py
if __name__ == '__main__':
    unittest.main()
47 chars
3 lines

similar python code snippets

selenium in python

throw and catch errors in python

open a file in python

connect to secrets manager in python

send an email in python

find urls in a string in python

how to create a class in python

loop in python

loop from 1 to 10 in python

how to create a flask app in python

related categories