from __future__ import annotations import json import random import time from dataclasses import dataclass from dataclasses import field from typing import Any, Callable, Optional import allure # pyright: ignore[reportMissingImports] from allure_commons.types import AttachmentType # pyright: ignore[reportMissingImports] from worklib import admin_data from worklib.graphql_client import DEFAULT_COMPANY_ID, execute_graphql def _attach_json(name: str, payload: Any) -> None: allure.attach( json.dumps(payload, ensure_ascii=False, indent=2), name=name, attachment_type=AttachmentType.JSON, ) def _exec_or_fail(*, op_name: str, token: str, query: str, variables: dict[str, Any] | None = None, company_id: str) -> dict[str, Any]: try: return execute_graphql( query=query, variables=variables, company_id=company_id, access_token=token, ) except PermissionError as e: allure.attach(str(e), name=f"Forbidden: {op_name}", attachment_type=AttachmentType.TEXT) raise AssertionError(f"Forbidden на операции: {op_name}") from e @dataclass class TicketTestData: """ Хранилище/фабрика тестовых данных для Ticket GraphQL. - Создаёт сущности (place/category/ticket/user/employee/...) по мере необходимости - Кеширует полученные id в полях - Регистрирует cleanup в behave context (если передан) """ company_id: str = DEFAULT_COMPANY_ID parent_place_id: str = "6915dc03462d5aea0adc8cbd" default_user_first_name: str = "kvstest1" default_user_last_name: str = "kvstest2" # Employee that must exist on the stand and be present in multiple groups fixed_user_id: str = "e47362a9-5354-4b42-97cc-c00dfe1c54f1" fixed_employee_id: str = "69cbe1d59547f08c1cf556ff" # cached ids access_token: Optional[str] = None place_id: Optional[str] = None category_id: Optional[str] = None ticket_id: Optional[str] = None account_id: Optional[str] = None employee_id: Optional[str] = None category_group_id: Optional[str] = None username: Optional[str] = None # title -> category_id category_ids: dict[str, str] = field(default_factory=dict) # role -> title (последние 3 категории для теста смены категории) last_category_titles_by_role: dict[str, str] = field(default_factory=dict) _cleanup_fns: Optional[list[Callable[[], None]]] = None @classmethod def from_behave_context(cls, context: Any, *, company_id: str = DEFAULT_COMPANY_ID) -> "TicketTestData": td: TicketTestData | None = getattr(context, "ticket_test_data", None) if isinstance(td, cls): if not td.access_token and getattr(context, "access_token", None): td.access_token = context.access_token if not td._cleanup_fns: td._cleanup_fns = getattr(context, "_cleanup_fns", None) return td td = cls(company_id=company_id) td.access_token = getattr(context, "access_token", None) or None td._cleanup_fns = getattr(context, "_cleanup_fns", None) setattr(context, "ticket_test_data", td) return td def _register_cleanup(self, fn: Callable[[], None]) -> None: if self._cleanup_fns is not None: self._cleanup_fns.append(fn) def ensure_token(self) -> str: if self.access_token: return self.access_token token = admin_data.get_or_create_user("tester").access_token assert token, "Нет access_token (ни в context, ни в admin_data.get_or_create_user('tester'))." self.access_token = token return token def ensure_place(self) -> str: if self.place_id: return self.place_id token = self.ensure_token() mutation = """ mutation ($place_type: PlaceType!, $names: [String!]!, $parent_id: String, $address_id: String) { createPlaceMultiple( dto: {place_type: $place_type, names: $names, parent_id: $parent_id, address_id: $address_id} ) { id __typename } } """.strip() variables = {"names": ["test"], "parent_id": self.parent_place_id, "place_type": "flat"} with allure.step("GraphQL: createPlaceMultiple"): resp = _exec_or_fail(op_name="createPlaceMultiple(mutation)", token=token, query=mutation, variables=variables, company_id=self.company_id) _attach_json("createPlaceMultiple response", resp) created = resp.get("data", {}).get("createPlaceMultiple") if isinstance(created, list): assert created, f"createPlaceMultiple вернул пустой список. Ответ: {resp}" created0 = created[0] else: created0 = created assert isinstance(created0, dict), f"createPlaceMultiple вернул не объект: {created0!r}. Ответ: {resp}" place_id = created0.get("id") assert place_id, f"createPlaceMultiple не вернул id. Ответ: {resp}" self.place_id = place_id def _cleanup_delete_place() -> None: delete_mutation = """mutation deleteplace($id: String!) { deletePlace(id: $id) }""".strip() _exec_or_fail(op_name="deletePlace(mutation)", token=token, query=delete_mutation, variables={"id": place_id}, company_id=self.company_id) self._register_cleanup(_cleanup_delete_place) return place_id def ensure_ticket_category(self) -> str: if self.category_id: return self.category_id token = self.ensure_token() place_id = self.ensure_place() mutation = """ mutation ($input: CreateTicketCategoryInput!) { createTicketCategory(dto: $input) { id title place_ids company_id } } """.strip() variables = {"input": {"title": "tester1", "place_ids": [place_id], "company_id": self.company_id}} with allure.step("GraphQL: createTicketCategory"): resp = _exec_or_fail(op_name="createTicketCategory(mutation)", token=token, query=mutation, variables=variables, company_id=self.company_id) _attach_json("createTicketCategory response", resp) category = resp.get("data", {}).get("createTicketCategory") assert isinstance(category, dict) and category.get("id"), f"createTicketCategory не вернул id. Ответ: {resp}" category_id = category["id"] self.category_id = category_id def _cleanup_delete_category() -> None: delete_mutation = """mutation deleteTicketcategory($id: String!) { deleteTicketCategory(id: $id) }""".strip() _exec_or_fail(op_name="deleteTicketCategory(mutation)", token=token, query=delete_mutation, variables={"id": category_id}, company_id=self.company_id) self._register_cleanup(_cleanup_delete_category) return category_id def create_ticket_category(self, *, title: str, place_id: Optional[str] = None) -> str: """ Создаёт ticket category (не перетирая self.category_id), сохраняет в self.category_ids[title]. """ if title in self.category_ids: return self.category_ids[title] token = self.ensure_token() pid = place_id or self.ensure_place() mutation = """ mutation ($input: CreateTicketCategoryInput!) { createTicketCategory(dto: $input) { id title place_ids company_id } } """.strip() variables = {"input": {"title": title, "place_ids": [pid], "company_id": self.company_id}} with allure.step(f"GraphQL: createTicketCategory ({title})"): resp = _exec_or_fail(op_name="createTicketCategory(mutation)", token=token, query=mutation, variables=variables, company_id=self.company_id) _attach_json("createTicketCategory response", resp) category = resp.get("data", {}).get("createTicketCategory") assert isinstance(category, dict) and category.get("id"), f"createTicketCategory не вернул id. Ответ: {resp}" category_id = category["id"] self.category_ids[title] = category_id def _cleanup_delete_category() -> None: delete_mutation = """mutation deleteTicketcategory($id: String!) { deleteTicketCategory(id: $id) }""".strip() _exec_or_fail(op_name="deleteTicketCategory(mutation)", token=token, query=delete_mutation, variables={"id": category_id}, company_id=self.company_id) self._register_cleanup(_cleanup_delete_category) return category_id def ensure_three_ticket_categories(self) -> dict[str, str]: """ Создаёт 3 категории с разными названиями под текущий place. Возвращает dict с ключами: old/in_group/out_group. """ pid = self.ensure_place() suffix = str(int(time.time())) titles = { "old": f"cat-old-{suffix}", "in_group": f"cat-in-group-{suffix}", "out_group": f"cat-out-group-{suffix}", } self.last_category_titles_by_role = titles ids = {k: self.create_ticket_category(title=v, place_id=pid) for k, v in titles.items()} return ids def create_ticket_with_category(self, *, category_id: str, place_id: Optional[str] = None) -> str: """ Создаёт ticket с указанными place_id и category_id. ВАЖНО: если на стенде нет прав на createTicket, тест должен падать (как и требуется). """ token = self.ensure_token() pid = place_id or self.ensure_place() mutation = """ mutation createticket($category_id: String!, $place_id: String!) { createTicket(dto: {subject:"my first ticket", body:"try a ticket", category_id:$category_id, place_id:$place_id }) { id } } """.strip() variables = {"category_id": category_id, "place_id": pid} with allure.step("GraphQL: createTicket"): resp = _exec_or_fail(op_name="createTicket(mutation)", token=token, query=mutation, variables=variables, company_id=self.company_id) _attach_json("createTicket response", resp) ticket_id = resp.get("data", {}).get("createTicket", {}).get("id") assert ticket_id, f"createTicket не вернул id. Ответ: {resp}" self.ticket_id = ticket_id def _cleanup_delete_ticket() -> None: delete_mutation = """mutation deleteTicket($id: String!) { deleteTicket(id: $id) }""".strip() _exec_or_fail(op_name="deleteTicket(mutation)", token=token, query=delete_mutation, variables={"id": ticket_id}, company_id=self.company_id) self._register_cleanup(_cleanup_delete_ticket) return ticket_id def ensure_ticket(self) -> str: if self.ticket_id: return self.ticket_id token = self.ensure_token() place_id = self.ensure_place() category_id = self.ensure_ticket_category() mutation = """ mutation createticket($category_id: String!, $place_id: String!) { createTicket(dto: {subject:"my first ticket", body:"try a ticket", category_id:$category_id, place_id:$place_id }) { id } } """.strip() variables = {"category_id": category_id, "place_id": place_id} with allure.step("GraphQL: createTicket"): resp = _exec_or_fail(op_name="createTicket(mutation)", token=token, query=mutation, variables=variables, company_id=self.company_id) _attach_json("createTicket response", resp) ticket_id = resp.get("data", {}).get("createTicket", {}).get("id") assert ticket_id, f"createTicket не вернул id. Ответ: {resp}" self.ticket_id = ticket_id def _cleanup_delete_ticket() -> None: delete_mutation = """mutation deleteTicket($id: String!) { deleteTicket(id: $id) }""".strip() _exec_or_fail(op_name="deleteTicket(mutation)", token=token, query=delete_mutation, variables={"id": ticket_id}, company_id=self.company_id) # Удалять ticket первым (до категории/плейса). self._register_cleanup(_cleanup_delete_ticket) return ticket_id def create_user(self) -> str: if self.account_id: return self.account_id token = self.ensure_token() username = f"+7999{random.randint(1000000, 9999999)}" mutation = """ mutation createUser($input: CreateAccountDTO!) { createUser(dto: $input) } """.strip() variables = { "input": { "username": username, "first_name": self.default_user_first_name, "last_name": self.default_user_last_name, "is_demo": True, "password": "", } } with allure.step("GraphQL: createUser"): resp = _exec_or_fail(op_name="createUser(mutation)", token=token, query=mutation, variables=variables, company_id=self.company_id) _attach_json("createUser response", resp) created = resp.get("data", {}).get("createUser") if isinstance(created, dict): account_id = created.get("id") returned_username = created.get("username") else: account_id = created returned_username = None assert isinstance(account_id, str) and account_id, f"createUser не вернул id. Ответ: {resp}" self.account_id = account_id self.username = returned_username or username def _cleanup_delete_user() -> None: delete_mutation = """ mutation deleteUser($account_id: String!) { deleteUser(account_id: $account_id) } """.strip() _exec_or_fail( op_name="deleteUser(mutation)", token=token, query=delete_mutation, variables={"account_id": account_id}, company_id=self.company_id, ) self._register_cleanup(_cleanup_delete_user) return account_id def create_employee(self) -> str: if self.employee_id: return self.employee_id token = self.ensure_token() account_id = self.create_user() mutation_with_status = """ mutation createEmployee($user_id: String!) { addEmployee(dto: {user_id: $user_id, attributes: [], role: admin, company_id: "%s"}) { id status } } """.strip() % self.company_id mutation_without_status = """ mutation createEmployee($user_id: String!) { addEmployee(dto: {user_id: $user_id, attributes: [], role: admin, company_id: "%s"}) { id } } """.strip() % self.company_id with allure.step("GraphQL: addEmployee"): try: resp = _exec_or_fail( op_name="addEmployee(mutation, with status)", token=token, query=mutation_with_status, variables={"user_id": account_id}, company_id=self.company_id, ) except RuntimeError as e: # На стенде поле EmployeeObject.status иногда ломает GraphQL (500: non-nullable + null). if "EmployeeObject.status" not in str(e): raise allure.attach( str(e), name="Skipping employee.status check (API bug)", attachment_type=AttachmentType.TEXT, ) resp = _exec_or_fail( op_name="addEmployee(mutation, without status)", token=token, query=mutation_without_status, variables={"user_id": account_id}, company_id=self.company_id, ) _attach_json("addEmployee response", resp) add_employee = resp.get("data", {}).get("addEmployee", {}) employee_id = add_employee.get("id") if isinstance(add_employee, dict) else None assert employee_id, f"addEmployee не вернул employee_id. Ответ: {resp}" status = add_employee.get("status") if isinstance(add_employee, dict) else None if status is not None: assert status == "active", f"status при создании employee должен быть active, получено: {status!r}. Ответ: {resp}" self.employee_id = employee_id return employee_id def create_category_group(self) -> str: category_id = self.ensure_ticket_category() return self.create_category_group_for_categories([category_id]) def create_category_group_for_categories( self, category_ids: list[str], *, members: Optional[list[dict[str, str]]] = None, cache: bool = True, ) -> str: """ Создаёт CategoryGroup для указанных категорий. - members: список участников вида {"user_id": "...", "employee_id": "..."} Если не передан, создаётся один новый user/employee и добавляется в группу. - cache: если True, сохраняет group_id в self.category_group_id (старое поведение). Если False, позволяет создавать несколько групп в рамках одного теста. """ if cache and self.category_group_id: return self.category_group_id assert category_ids and all(isinstance(x, str) and x for x in category_ids), f"category_ids пустой/некорректный: {category_ids!r}" token = self.ensure_token() if members is None: account_id = self.create_user() employee_id = self.create_employee() members = [{"user_id": account_id, "employee_id": employee_id}] assert members and all(isinstance(m, dict) and m.get("user_id") and m.get("employee_id") for m in members), f"members некорректный: {members!r}" mutation = """ mutation createcategoryGroup($name: String!, $category_ids: [String!]!, $employees: [GroupMemberDto!]!) { createCategoryGroup(dto: { name: $name, employees: $employees, company_id: "%s", category_ids: $category_ids }) { id } } """.strip() % self.company_id variables = { "name": f"tester-{int(time.time())}", "category_ids": category_ids, "employees": members, } with allure.step("GraphQL: createCategoryGroup"): resp = _exec_or_fail(op_name="createCategoryGroup(mutation)", token=token, query=mutation, variables=variables, company_id=self.company_id) _attach_json("createCategoryGroup response", resp) group_id = resp.get("data", {}).get("createCategoryGroup", {}).get("id") assert group_id, f"createCategoryGroup не вернул id. Ответ: {resp}" if cache: self.category_group_id = group_id def _cleanup_delete_group() -> None: delete_mutation = """ mutation deletecategoryGroup($id: String!) { deleteCategoryGroup(id: $id) } """.strip() _exec_or_fail(op_name="deleteCategoryGroup(mutation)", token=token, query=delete_mutation, variables={"id": group_id}, company_id=self.company_id) self._register_cleanup(_cleanup_delete_group) return group_id def connect_employee_to_category_group(self) -> None: token = self.ensure_token() group_id = self.create_category_group() employee_id = self.create_employee() account_id = self.account_id _attach_json("connectEmployee inputs", {"group_id": group_id, "employee_id": employee_id, "account_id": account_id}) def _looks_like_uuid(value: str) -> bool: return isinstance(value, str) and value.count("-") == 4 and len(value) >= 32 mutation = """ mutation connectEmployee($input: AddEmployeesToCategoryGroupDTO!) { addEmployeesToCategoryGroup(dto: $input) } """.strip() employee_ids = [employee_id] if _looks_like_uuid(employee_id) else ([account_id] if account_id else [employee_id]) variables = {"input": {"id": group_id, "employee_ids": employee_ids}} with allure.step("GraphQL: addEmployeesToCategoryGroup (connectEmployee)"): resp = _exec_or_fail(op_name="addEmployeesToCategoryGroup(mutation)", token=token, query=mutation, variables=variables, company_id=self.company_id) _attach_json("addEmployeesToCategoryGroup response", resp)