Доступ ко всем файлам cookie в ответе на тест Flask - PullRequest
Новая
       13

Доступ ко всем файлам cookie в ответе на тест Flask

1 голос
/ 04 апреля 2019

После того, как я сделаю запрос с помощью тестового клиента 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 и их свойствам для тестирования?

1 Ответ

0 голосов
/ 04 апреля 2019

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