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")

            return urlQueueList.poll();
        }
    }

    /*
     * (non-Javadoc)
     *
     * @see
     * org.codelibs.fess.crawler.service.UrlQueueService#saveSession(java.lang.String)
     */
    @Override
    public void saveSession(final String sessionId) {
        // NOP
    }

    /*
     * (non-Javadoc)
     *
     * @see org.codelibs.fess.crawler.service.UrlQueueService#visited(UrlQueue)

     * @return The URL queue.
     */
    QUEUE poll(String sessionId);

    /**
     * Saves the session.
     *
     * @param sessionId The session ID.
     */
    void saveSession(String sessionId);

    /**
     * Checks if a URL has been visited.
     *
     * @param urlQueue The URL queue.
     * @return true if the URL has been visited, otherwise false.
     */

            }

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

            @Override
            public void saveSession(String sessionId) {
            }

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

            @Override

        // Group archived keys by session
        for (String key : sessionKeys.keySet()) {
            if (key.contains(".v")) {
                String baseSession = key.substring(0, key.indexOf(".v"));
                sessionArchives.computeIfAbsent(baseSession, k -> new java.util.ArrayList<>()).add(key);
            }
        }

        // Remove old versions
        int removed = 0;

        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) {