Fix test_screenshot_remove

CustomEnv · Dec 24, 2024 · 9ddefaf · 9ddefaf
1 parent 00abbc8
commit 9ddefaf
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 3 deletions.
diff --git a/tests/adata/pages/colored_blocks_page.py b/tests/adata/pages/colored_blocks_page.py
@@ -1,10 +1,17 @@
 from typing import List
 
 from dyatel.base.element import Element
+from dyatel.base.group import Group
 from dyatel.base.page import Page
 from tests.settings import domain_name, repo_name
 
 
+class Row(Group):
+    def __init__(self, locator, name: str):
+        self.card = Element('.card', name=f'any colored block of {name}')
+        super().__init__(locator, name=name)
+
+
 class ColoredBlocksPage(Page):
 
     def __init__(self):
@@ -13,6 +20,8 @@ def __init__(self):
     url = f'{domain_name}/{repo_name}/colored_blocks.html'
 
     blocks_container = Element('.container', name='colored blocks container')
+    row1 = Row('row-1', name='first blocks row')
+    row2 = Row('row-2', name='second blocks row')
     card = Element('.card', name='any colored block')
     navbar = Element('.navbar', name='navbar')
 

diff --git a/...al/reference/test_screenshot_remove_container_of_text_linux_selenium_chrome.png b/...al/reference/test_screenshot_remove_container_of_text_linux_selenium_chrome.png
diff --git a/...l/reference/test_screenshot_remove_container_of_text_linux_selenium_firefox.png b/...l/reference/test_screenshot_remove_container_of_text_linux_selenium_firefox.png
diff --git a/...al/reference/test_screenshot_remove_second_blocks_row_linux_selenium_chrome.png b/...al/reference/test_screenshot_remove_second_blocks_row_linux_selenium_chrome.png
diff --git a/...l/reference/test_screenshot_remove_second_blocks_row_linux_selenium_firefox.png b/...l/reference/test_screenshot_remove_second_blocks_row_linux_selenium_firefox.png
diff --git a/tests/web_tests/test_assert_screenshot.py b/tests/web_tests/test_assert_screenshot.py
@@ -34,9 +34,14 @@ def test_screenshot_name_with_suffix(base_playground_page, driver_name, platform
     base_playground_page.kube.scroll_into_view().assert_screenshot(filename, name_suffix='second')
 
 
-def test_screenshot_remove(base_playground_page):
-    base_playground_page.text_container.scroll_into_view(sleep=0.5).assert_screenshot(
-            remove=[base_playground_page.inner_text_1, base_playground_page.inner_text_2])
+def test_screenshot_remove(colored_blocks_page):
+    row2_card = colored_blocks_page.row2.card
+    cards = row2_card.wait_elements_count(8).all_elements
+    colored_blocks_page.row2.assert_screenshot(
+        scroll=True,
+        remove=[cards[5], cards[3]],
+        delay=0.5
+    )
 
 
 @pytest.fixture