add css locator for identifing all buttons at once

add a test counting the buttons using this method

Author: amielnoy

Ultimateqa/pages/SubPages/sub_page_buttons.py

@@ -5,6 +5,7 @@ class SubPageButtons:
     def __init__(self, page: Page):
         self.page = page
         self.some_element = self.page.locator("#subpage1_element")
+        self.all_buttons= self.page.locator("[class^='et_pb_button_']")
 
     def locate_button(self, index: int):
         return self.page.locator(f'.et_pb_button_{index}')
@@ -22,4 +23,6 @@ def count_buttons(self):
         while self.does_button_exist(count):
             count += 1
         return count
+    def count_buttons_by_class(self):
+        return self.all_buttons.count()
 

Ultimateqa/ui_tests/test_fe.py

@@ -1,16 +1,6 @@
-from dotenv import load_dotenv
-import os
 import pytest
-
-import sys
-
-#from conftest import base_url
-
-sys.path.append('/Users/amielpeled/PycharmProjects/test-force-point/ForcePoint/Ultimateqa')
-
 from playwright.sync_api import Page
 from Ultimateqa.pages.SubPages.sub_page_buttons import SubPageButtons
-from Ultimateqa.pages.SubPages.sub_page_buttons import SubPageButtons
 from Ultimateqa.pages.SubPages.sub_page_social_media import SubPageSocialMedia
 from Ultimateqa.pages.SubPages.sub_page_forms import SubPageForms
 import Ultimateqa.ui_tests.consts as consts
@@ -21,9 +11,22 @@ def test_count_buttons(page: Page, base_url_fe):
     sub_page_buttons = SubPageButtons(page)
 
     count_of_buttons = sub_page_buttons.count_buttons()
+
     assert count_of_buttons == consts.NUN_OF_BUTTONS_TO_FIND, \
         f"Found: {count_of_buttons} buttons, expected: {consts.NUN_OF_BUTTONS_TO_FIND}"
 
+    count_all_buttons_using_css = sub_page_buttons.all_buttons.count()
+    assert count_all_buttons_using_css == consts.NUN_OF_BUTTONS_TO_FIND, \
+        f"Found: {count_of_buttons} buttons, expected: {consts.NUN_OF_BUTTONS_TO_FIND}"
+
+def test_count_buttons_using_css(page: Page, base_url_fe):
+    page.goto(base_url_fe)
+    sub_page_buttons = SubPageButtons(page)
+
+    count_all_buttons_using_css = sub_page_buttons.all_buttons.count()
+    assert count_all_buttons_using_css == consts.NUN_OF_BUTTONS_TO_FIND, \
+        f"Found: {count_all_buttons_using_css} buttons, expected: {consts.NUN_OF_BUTTONS_TO_FIND}"
+
 
 def test_verify_href_link(page: Page, base_url_fe):
     # load_dotenv()