from fastapi import FastAPI, Response

app = FastAPI()


@app.post("/cookie-and-object/")
def create_cookie(response: Response):
    response.set_cookie(key="fakesession", value="fake-cookie-session-value")

    client = TestClient(app, cookies={"session": "fakesession"})
    with pytest.raises(WebSocketDisconnect):
        with client.websocket_connect("/items/foo/ws") as websocket:
            message = "Message one"
            websocket.send_text(message)
            data = websocket.receive_text()
            assert data == "Session cookie or query token value is: fakesession"
            data = websocket.receive_text()

    response = client.post("/cookie/")
    assert response.status_code == 200, response.text
    assert response.json() == {"message": "Come to the dark side, we have cookies"}

    response = client.post("/cookie-and-object/")
    assert response.status_code == 200, response.text
    assert response.json() == {"message": "Come to the dark side, we have cookies"}

app = FastAPI()


@app.post("/cookie/")
def create_cookie():
    content = {"message": "Come to the dark side, we have cookies"}
    response = JSONResponse(content=content)
    response.set_cookie(key="fakesession", value="fake-cookie-session-value")

            }

            @Override
            public UrlQueue<?> poll(String sessionId) {
                return null;
            }

            @Override
            public void saveSession(String sessionId) {
            }

            @Override
            public boolean visited(UrlQueue<?> urlQueue) {
                return false;
            }

            @Override

        if (logger.isInfoEnabled()) {
            logger.info("Crawler execution completed: sessionId={}, status={}", crawlerContext.sessionId, CrawlerStatus.DONE);
        }

        try {
            urlQueueService.saveSession(crawlerContext.sessionId);
            if (logger.isDebugEnabled()) {
                logger.debug("Saved crawler session: sessionId={}", crawlerContext.sessionId);
            }
        } catch (final Exception e) {

        boolean result = service.visited(newUrl);

        // Verify
        assertFalse(result);
    }

    public void test_saveSession() {
        // Execute - should be a no-op
        service.saveSession("session123");

        // No verification needed as it's a no-op method
    }

    public void test_generateUrlQueues() {
        // Setup
        String previousSessionId = "prevSession";

    }

    /**
     * Saves the session state (currently not implemented).
     *
     * @param sessionId The session ID.
     */
    @Override
    public void saveSession(final String sessionId) {
        // TODO use cache
    }

    /**
     * Checks if a URL has been visited by looking in both the queue and access results.
     *