Example #1
0
def test_client_cookies():
    client = AsyncClient()
    client.cookies = {"a": "b"}
    assert isinstance(client.cookies, Cookies)
    mycookies = list(client.cookies.jar)
    assert len(mycookies) == 1
    assert mycookies[0].name == "a" and mycookies[0].value == "b"
Example #2
0
async def test_setting_client_cookies_to_cookiejar() -> None:
    """
    Send a request including a cookie, using a `CookieJar` instance.
    """

    url = "http://example.org/echo_cookies"
    cookies = CookieJar()
    cookie = Cookie(
        version=0,
        name="example-name",
        value="example-value",
        port=None,
        port_specified=False,
        domain="",
        domain_specified=False,
        domain_initial_dot=False,
        path="/",
        path_specified=True,
        secure=False,
        expires=None,
        discard=True,
        comment=None,
        comment_url=None,
        rest={"HttpOnly": ""},
        rfc2109=False,
    )
    cookies.set_cookie(cookie)

    client = AsyncClient(transport=MockTransport())
    client.cookies = cookies  # type: ignore
    response = await client.get(url)

    assert response.status_code == 200
    assert response.json() == {"cookies": "example-name=example-value"}
Example #3
0
async def logged_session(client: AsyncClient, user_id: int = None) -> None:
    cookies = {}
    headers = {}
    if user_id:
        session_id = await create_session(dict(id=user_id))
        csrf_token = create_csrf(session_id)
        cookies = {'session_id': session_id}
        headers = {'x-csrf-token': csrf_token}
    client.cookies = cookies
    client.headers = headers
    return