Chapter 4 + 5: Initializing eyes in your tests and running your first hello world test.

Author: Gaurav Singh

automation/tests/base_test.py

@@ -1,12 +1,34 @@
 from unittest import TestCase
 
+from applitools.common import MatchLevel
+from applitools.selenium import Eyes
 from selenium import webdriver
 
+from automation.config.base import APPLITOOLS_API_KEY
+
 
 class BaseTest(TestCase):
-    def setUp(self):
+    def setup(self, test_name):
         self.driver = webdriver.Chrome()
         self.driver.get('https://applitools.com/helloworld')
+        self.eyes = self.open_eyes(test_name)
 
-    def tearDown(self):
+    def teardown(self):
         self.driver.quit()
+        self.close_eyes()
+
+    def open_eyes(self, test_name):
+        self.eyes = Eyes()
+        self.eyes.api_key = APPLITOOLS_API_KEY
+
+        self.eyes.match_level = MatchLevel.STRICT
+        self.eyes.open(self.driver, app_name='hello_world',
+                       test_name=test_name)
+
+        return self.eyes
+
+    def close_eyes(self):
+        self.eyes.close()
+
+    def validate_window(self, tag):
+        self.eyes.check_window(tag)

automation/tests/chapter_04/hello_world.py

@@ -4,4 +4,10 @@
 class HelloWorldTest(BaseTest):
 
     def test_hello_world(self):
-        pass
+        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()