screenshotting temporary

Author: sarinali

modules/browser_object_tracker_panel.py

@@ -1,3 +1,4 @@
+import datetime
 import logging
 from typing import List
 
@@ -16,7 +17,7 @@ class TrackerPanel(BasePage):
     URL_TEMPLATE = ""
 
     def wait_for_blocked_tracking_icon(
-        self, nav: Navigation, page: BasePage
+        self, nav: Navigation, page: BasePage, screenshot
     ) -> BasePage:
         """
         Waits for the shield icon to indicate that cookies/trackers are being blocked by continuously refresing the page
@@ -33,6 +34,7 @@ def shield_active() -> bool:
             with self.driver.context(self.driver.CONTEXT_CONTENT):
                 page.wait_for_page_to_load()
             shield_icon = nav.get_element("shield-icon")
+            screenshot(str(datetime.datetime.now()))
             if (
                 shield_icon.get_attribute("data-l10n-id")
                 == "tracking-protection-icon-active-container"

tests/security_and_privacy/test_third_party_content_blocked_private_browsing.py

@@ -17,7 +17,7 @@
 SECOND_TRACKER_WEBSITE = "https://www.itisatrap.org/firefox/its-a-tracker.html"
 
 
-def test_third_party_content_blocked_private_browsing_cross_site(driver: Firefox):
+def test_third_party_content_blocked_private_browsing_cross_site(driver: Firefox, screenshot):
     """
     C446323.1: Ensure that third party content is blocked correctly
     """
@@ -33,7 +33,7 @@ def test_third_party_content_blocked_private_browsing_cross_site(driver: Firefox
 
     # open the website, ensure the blocking is taking place by continuously refreshing website until indicated
     tracker_website.open()
-    tracker_panel.wait_for_blocked_tracking_icon(nav, tracker_website)
+    tracker_panel.wait_for_blocked_tracking_icon(nav, tracker_website, screenshot)
 
     # verify the indicator
     driver.set_context(driver.CONTEXT_CHROME)
@@ -58,7 +58,7 @@ def test_third_party_content_blocked_private_browsing_cross_site(driver: Firefox
     assert found_tracker
 
 
-def test_third_party_content_blocked_private_browsing_allowed_tracking(driver: Firefox):
+def test_third_party_content_blocked_private_browsing_allowed_tracking(driver: Firefox, screenshot):
     """
     C446323.2: Ensure that some third party content is allowed
     """
@@ -74,7 +74,7 @@ def test_third_party_content_blocked_private_browsing_allowed_tracking(driver: F
 
     # open the website, ensure the blocking is taking place by continuously refreshing website until indicated
     tracker_website.open()
-    tracker_panel.wait_for_blocked_tracking_icon(nav, tracker_website)
+    tracker_panel.wait_for_blocked_tracking_icon(nav, tracker_website, screenshot)
 
     # verify the indicator
     driver.set_context(driver.CONTEXT_CHROME)
@@ -96,7 +96,7 @@ def test_third_party_content_blocked_private_browsing_allowed_tracking(driver: F
         assert item.get_attribute("value") in ALLOWED_TRACKING_URLS
 
 
-def test_third_party_content_private_browsing_tracking_statuses(driver: Firefox):
+def test_third_party_content_private_browsing_tracking_statuses(driver: Firefox, screenshot):
     """
     C446323.3: Ensure that the statuses of some third party content are loaded properly
     """
@@ -112,7 +112,7 @@ def test_third_party_content_private_browsing_tracking_statuses(driver: Firefox)
 
     # open the tracker website
     tracker_website.open()
-    tracker_panel.wait_for_blocked_tracking_icon(nav, tracker_website)
+    tracker_panel.wait_for_blocked_tracking_icon(nav, tracker_website, screenshot)
 
     # verify the indicator
     tracker_panel.verify_tracker_shield_indicator(nav)