Add POM and main page tests

.gitignore

@@ -1 +1,8 @@
-venv
+venv
+
+.idea/
+__pycache__/
+data/__pycache__/
+locators/__pycache__/
+
+database_connection.py

.pylintrc

@@ -4,4 +4,5 @@ disable=
     C0116,
     C0115,
     E0401,
-    W0621
+    W0621,
+    R0903

data/urls.py

@@ -0,0 +1,7 @@
+
+class Urls:
+    MAIN_PAGE = "https://www.it-roast.com/#"
+    MAIN_PAGE_TEST = "https://front-test-a2ykv.ondigitalocean.app"
+    LINKEDIN_PAGE = "https://www.linkedin.com/company/train-lab-interns/"
+    REG_PAGE = "https://www.it-roast.com/auth"
+    GITHUB_PAGE = "https://github.com/Train-lab-intern"

locators/main_page_locators.py

@@ -0,0 +1,11 @@
+from selenium.webdriver.common.by import By
+
+
+class MainPageLocators:
+    LOGO = (By.CSS_SELECTOR, 'img[alt="Logo"]')
+    LINKEDIN_LINK = (By.CSS_SELECTOR, "[alt='linkedin']")
+    GITHUB_LINK = (By.CSS_SELECTOR, "[alt='github']")
+    ABOUT_US_BUTTON = (By.CSS_SELECTOR, '[class="Navigation_link__Jlf0e active"]:first-of-type')
+    TASKS_BUTTON = (By.CSS_SELECTOR, '[class="Navigation_link__Jlf0e active"]:nth-of-type(2)')
+    REG_BUTTON = (By.CSS_SELECTOR, "[href='/auth']")
+    HEADER_BUTTONS = [ABOUT_US_BUTTON, TASKS_BUTTON, REG_BUTTON]

pages/base_page.py

@@ -0,0 +1,33 @@
+
+
+
+class BasePage:
+
+    def __init__(self, driver, link=None):
+        self.driver = driver
+        self.link = link
+
+    def open(self):
+        self.driver.get(self.link)
+
+    def find_element(self, locator):
+        return self.driver.find_element(*locator)
+
+    def check_element_visibility(self, locator):
+        element = self.find_element(locator)
+        assert element.is_displayed()
+
+    def check_element_clickability(self, locator):
+        element = self.find_element(locator)
+        assert element.is_enabled()
+
+    def find_element_and_click(self, locator):
+        self.find_element(locator).click()
+
+    def switch_to_new_window(self):
+        self.driver.switch_to.window(self.driver.window_handles[-1])
+
+
+    def assert_current_url(self, expected_url):
+        current_url = self.driver.current_url
+        assert current_url == expected_url, f"URL does not match. Actual page is: {current_url}"

pages/main_page.py

@@ -0,0 +1,22 @@
+from pages.base_page import BasePage
+from locators.main_page_locators import MainPageLocators
+# from data.urls import LINKEDIN_PAGE, GITHUB_PAGE, MAIN_PAGE, MAIN_PAGE_TEST
+from data.urls import Urls
+
+
+class MainPage(BasePage):
+
+    def __init__(self, driver):
+        super().__init__(driver, Urls.MAIN_PAGE_TEST)
+
+    def logo_visibility(self):
+        self.check_element_visibility(MainPageLocators.LOGO)
+
+    def check_linkedin_redirection(self):
+        self.find_element_and_click(MainPageLocators.LINKEDIN_LINK)
+        self.switch_to_new_window()
+        self.assert_current_url(Urls.LINKEDIN_PAGE)
+
+    def check_github_redirection(self):
+        self.find_element_and_click(MainPageLocators.GITHUB_LINK)
+        self.assert_current_url(Urls.GITHUB_PAGE)

requirements.txt

@@ -28,7 +28,7 @@ PyNaCl==1.5.0
 PySocks==1.7.1
 pytest==7.4.0
 requests==2.31.0
-selenium==4.10.0
+selenium==4.12.0
 sniffio==1.3.0
 sortedcontainers==2.4.0
 sshtunnel==0.4.0

test_main.py

@@ -0,0 +1,30 @@
+import pytest
+from locators.main_page_locators import MainPageLocators
+from pages.main_page import MainPage
+
+
+# Header tests
+def test_check_logo_visibility(driver):
+    page = MainPage(driver)
+    page.open()
+    page.logo_visibility()
+
+
+@pytest.mark.parametrize('button', MainPageLocators.HEADER_BUTTONS)
+def test_check_header_buttons_clickability(driver, button):
+    page = MainPage(driver)
+    page.open()
+    page.check_element_clickability(button)
+
+# Footer tests
+@pytest.mark.xfail
+def test_linkedin_redirection(driver):
+    page = MainPage(driver)
+    page.open()
+    page.check_linkedin_redirection()
+
+
+def test_github_redirection(driver):
+    page = MainPage(driver)
+    page.open()
+    page.check_github_redirection()