from robot.api.deco import keyword, library
from robot.libraries.BuiltIn import BuiltIn


@library(scope="GLOBAL")
class QuipApiLibrary:
    """Reusable keywords for Quip session and stream checks."""

    def __init__(self, base_url: str):
        self.base_url = base_url.rstrip("/")
        self._http = BuiltIn().get_library_instance("RequestsLibrary")

    @keyword("Session Should Be Active")
    def session_should_be_active(self, session_id: str) -> None:
        response = self._http.get_on_session(
            "quip",
            f"/api/sessions/{session_id}",
        )
        assert response.status_code == 200
        assert response.json()["status"] == "active"

    @keyword("Open Quip Stream View")
    def open_quip_stream_view(self, session_id: str) -> None:
        selenium = BuiltIn().get_library_instance("SeleniumLibrary")
        selenium.go_to(f"{self.base_url}/streams/{session_id}")
        selenium.wait_until_element_is_visible(
            "css:[data-testid=stream-canvas]"
        )

*** Settings ***
Documentation     Smoke checks for Quip streaming sessions.
Library           SeleniumLibrary
Library           RequestsLibrary
Library           ../libraries/QuipApiLibrary.py    ${BASE_URL}
Suite Setup       Create Session    quip    ${BASE_URL}
Test Tags         smoke

*** Variables ***
${BASE_URL}       http://quip.local
${SESSION_ID}     demo-or-stream

*** Test Cases ***
Session Should Be Active
    [Documentation]    Verify REST session state before opening the UI.
    Session Should Be Active    ${SESSION_ID}

Open Quip Stream View
    [Documentation]    Chromium canvas is ready for the active session.
    Open Quip Stream View    ${SESSION_ID}
    Page Should Contain Element    css:[data-testid=stream-canvas]

Network Path Remains Reachable
    [Documentation]    Ports and routes respond on the Quip device.
    ${response}=    GET On Session    quip    /health
    Should Be Equal As Integers    ${response.status_code}    200

$ robot --include smoke tests/quip_smoke.robot
==============================================================================
Quip Smoke :: Smoke checks for Quip streaming sessions.
==============================================================================
Session Should Be Active :: Verify REST session state                 | PASS |
Open Quip Stream View :: Chromium canvas is ready                     | PASS |
Network Path Remains Reachable :: Ports and routes respond            | PASS |
==============================================================================
3 tests, 3 passed, 0 failed