85 lines
3.0 KiB
Python
85 lines
3.0 KiB
Python
"""Базовый класс страницы для работы с Playwright."""
|
||
|
||
from playwright.sync_api import Page, Response, APIRequestContext, expect
|
||
from typing import Any, Dict, List, Optional
|
||
from data.environment import host
|
||
from tools.logger import get_logger
|
||
import json
|
||
|
||
logger = get_logger("BASE_PAGE")
|
||
|
||
|
||
class BasePage:
|
||
"""Базовый класс для работы со страницами через Playwright."""
|
||
|
||
def __init__(self, page: Page) -> None:
|
||
self.page = page
|
||
|
||
# Действия:
|
||
def current_url(self) -> str:
|
||
return self.page.url
|
||
|
||
def open(self, uri: str) -> Optional[Response]:
|
||
return self.page.goto(f"{host.get_base_url()}{uri}", wait_until='domcontentloaded')
|
||
|
||
def page_reload(self) -> None:
|
||
self.page.reload()
|
||
|
||
def wait_for_timeout(self, timeout: int) -> None:
|
||
self.page.wait_for_timeout(timeout)
|
||
|
||
def get_api_request_context(self) -> APIRequestContext:
|
||
return self.page.context.request
|
||
|
||
def send_get_api_request(self, uri: str) -> Response:
|
||
api_request_context = self.get_api_request_context()
|
||
token = host.get_access_token()
|
||
headers = {"Accept": "application/json", "Authorization": f"Bearer {token}"}
|
||
response = api_request_context.get(
|
||
f"{host.get_request_url()}{uri}",
|
||
headers=headers
|
||
)
|
||
return response
|
||
|
||
def send_post_api_request(self, uri: str, payload: Dict[str, Any]) -> Response:
|
||
api_request_context = self.get_api_request_context()
|
||
token = host.get_access_token()
|
||
headers = {"Accept": "application/json", "Authorization": f"Bearer {token}"}
|
||
response = api_request_context.post(
|
||
f"{host.get_request_url()}{uri}",
|
||
headers=headers,
|
||
data=payload
|
||
)
|
||
return response
|
||
|
||
def get_response_body(self, response: Response) -> Optional[Dict[str, Any]]:
|
||
try:
|
||
response_body = response.json()
|
||
except json.JSONDecodeError:
|
||
logger.error("Failed to decode JSON response")
|
||
return None
|
||
return response_body
|
||
|
||
# Проверки:
|
||
def check_URL(self, uri: str, msg: str) -> None:
|
||
expect(self.page).to_have_url(
|
||
f"{host.get_base_url()}{uri}",
|
||
timeout=60000
|
||
), msg
|
||
|
||
def check_equals(self, actual: Any, expected: Any, msg: str) -> None:
|
||
assert actual == expected, msg
|
||
|
||
def check_lists_equals(self, actual: List[Any], expected: List[Any], msg: str) -> None:
|
||
def compare_lists(list1: List[Any], list2: List[Any]) -> bool:
|
||
if len(list1) != len(list2):
|
||
return False
|
||
for item1, item2 in zip(list1, list2):
|
||
if isinstance(item1, list) and isinstance(item2, list):
|
||
if not compare_lists(item1, item2):
|
||
return False
|
||
elif item1 != item2:
|
||
return False
|
||
return True
|
||
|
||
assert compare_lists(actual, expected), msg |