Блог

После того, как я сделаю запрос с помощью тестового клиента Flask, я хочу получить доступ к файлам cookie, установленным сервером. Если я повторяю response.headers , я вижу несколько Set-Cookie заголовков, но если я это делаю response.headers["Set-Cookie"] , я получаю только одно значение. Кроме того, заголовки представляют собой непрочитанные строки, которые сложно протестировать.

 response = client.get("/")
print(response.headers['Set-Cookie'])
'mycookie=value; Expires=Thu, 27-Jun-2019 13:42:19 GMT; Max-Age=1800; Path=/'

for item in response.headers:
    print(item)

('Content-Type', 'application/javascript')
('Content-Length', '215')
('Set-Cookie', 'mycookie=value; Expires=Thu, 27-Jun-2019 13:42:19 GMT; Max-Age=1800; Path=/')
('Set-Cookie', 'mycookie2=another; Domain=.client.com; Expires=Sun, 04-Apr-2021 13:42:19 GMT; Max-Age=62208000; Path=/')
('Set-Cookie', 'mycookie3=something; Domain=.client.com; Expires=Thu, 04-Apr-2019 14:12:19 GMT; Max-Age=1800; Path=/')
  

Почему доступ к Set-Cookie заголовку дает мне только один заголовок? Как я могу получить доступ к файлам cookie и их свойствам для тестирования?

response.headers является MultiDict , который предоставляет getlist метод для получения всех значений для данного ключа.

 response.headers.getlist('Set-Cookie')
  

Возможно, было бы более полезно изучить файлы cookie, имеющиеся у клиента, а не конкретные необработанные Set-Cookie заголовки, возвращаемые ответом. client.cookie_jar является CookieJar экземпляром, итерация по нему дает Cookie экземпляры. Например, чтобы получить значение файла cookie с именем «user_id»:

 client.post("/login")
cookie = next(
    (cookie for cookie in client.cookie_jar if cookie.name == "user_id"),
    None
)
assert cookie is not None
assert cookie.value == "4"
  

Предыдущий ответ привел меня к немного альтернативной версии в зависимости от того, что вы хотите сделать с файлом cookie.

Я пытался использовать client.cookie_jar , но я тестировал несколько «нестандартных» атрибутов, таких как HttpOnly и SameSite . Cookie, возвращенный из client.cookie_jar , не возвращает их, поэтому я вместо этого проверяю Set-Cookie заголовок:

 from werkzeug.http import parse_cookie

cookies = response.headers.getlist('Set-Cookie')
cookie = next(
    (cookie for cookie in cookies if expected_cookie_name in cookie),
    None
)

assert cookie is not None
cookie_attrs = parse_cookie(cookie)

assert cookie_attrs[expected_cookie_name] == expected_cookie_value
assert 'Secure' in cookie_attrs
assert 'HttpOnly' in cookie_attrs
assert cookie_attrs['SameSite'] == 'Lax'