Merge pull request #1705 from pgjones/samesite

Add samesite cookie support to test client

CHANGES.rst

@@ -44,6 +44,8 @@ Unreleased
     such that ``X-Foo`` is the same as ``x-foo``. :pr:`1605`
 -   :meth:`http.dump_cookie` accepts ``'None'`` as a value for
     ``samesite``. :issue:`1549`
+-   :meth:`~test.Client.set_cookie` accepts a ``samesite`` argument.
+    :pr:`1705`
 -   Support the Content Security Policy header through the
     `Response.content_security_policy` data structure. :pr:`1617`
 -   ``LanguageAccept`` will fall back to matching "en" for "en-US" or

src/werkzeug/test.py

@@ -824,6 +824,7 @@ def set_cookie(
         domain=None,
         secure=None,
         httponly=False,
+        samesite=None,
         charset="utf-8",
     ):
         """Sets a cookie in the client's cookie jar.  The server name
@@ -832,7 +833,16 @@ def set_cookie(
         """
         assert self.cookie_jar is not None, "cookies disabled"
         header = dump_cookie(
-            key, value, max_age, expires, path, domain, secure, httponly, charset
+            key,
+            value,
+            max_age,
+            expires,
+            path,
+            domain,
+            secure,
+            httponly,
+            charset,
+            samesite=samesite,
         )
         environ = create_environ(path, base_url="http://" + server_name)
         headers = [("Set-Cookie", header)]