Moved from unittest to pytest as a test framework

Author: Gaurav Singh

Pipfile

@@ -7,6 +7,7 @@ verify_ssl = true
 
 [packages]
 eyes-selenium = "*"
+pytest = "*"
 
 [requires]
-python_version = "3.7"
+python_version = "3.7.4"

Pipfile.lock

@@ -1,13 +1,8 @@
-from automation.tests.base_test import BaseTest
+from automation.tests.conftest import *
 
 
-class HelloWorldTest(BaseTest):
-
-    def test_hello_world(self):
-        self.setup(test_name=self._testMethodName)
-
-        self.validate_window(tag='hello_world')
-        self.driver.find_element_by_css_selector('button').click()
-        self.validate_window('click_me')
-
-        self.teardown()
+def test_hello_world(driver, eyes):
+    open_eyes(driver, eyes)
+    validate_window(eyes, tag='hello_world')
+    driver.find_element_by_css_selector('button').click()
+    validate_window(eyes, 'click_me')

automation/tests/base_test.py

@@ -0,0 +1,41 @@
+import pytest
+from applitools.selenium import Eyes
+from selenium import webdriver
+
+from automation.config.base import APPLITOOLS_API_KEY
+
+APP_NAME = 'hello_world'
+
+
+@pytest.fixture(scope='function')
+def driver():
+    driver = webdriver.Chrome()
+    driver.get('https://applitools.com/helloworld')
+    yield driver
+    driver.quit()
+
+
+@pytest.fixture(scope='function')
+def eyes(driver):
+    eyes = initialize_eyes()
+    yield eyes
+    eyes.close()
+
+
+def initialize_eyes():
+    eyes = Eyes()
+    eyes.api_key = APPLITOOLS_API_KEY
+    return eyes
+
+
+def open_eyes(driver, eyes):
+    eyes.open(driver, APP_NAME, test_name=get_test_name())
+
+
+def validate_window(eyes, tag):
+    eyes.check_window(tag)
+
+
+def get_test_name():
+    import inspect
+    return inspect.stack()[2].function