import typing

from werkzeug.datastructures import MultiDict
from werkzeug.http import parse_cookie
from werkzeug.test import TestResponse


def get_cookie(
    resp: TestResponse, name: str
) -> typing.Optional["MultiDict[str, str]"]:
    cookies = resp.headers.getlist("Set-Cookie")
    cookie = next((cookie for cookie in cookies if name in cookie), None)
    if cookie is None:
        return

    cookie_attrs = parse_cookie(cookie)
    return cookie_attrs