Test explicit pickle serialization

Doist · Sep 3, 2024 · d0ed44d · d0ed44d
1 parent 61b7abe
commit d0ed44d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_cookie.py b/tests/test_cookie.py
@@ -1,5 +1,6 @@
 import datetime
 import json
+import pickle
 
 from werkzeug.http import parse_cookie
 from werkzeug.wrappers import Request
@@ -86,3 +87,12 @@ class JSONSecureCookie(SecureCookie):
     secure = JSONSecureCookie({"foo": "bar"}, "secret").serialize()
     data = JSONSecureCookie.unserialize(secure, "secret")
     assert data == {"foo": "bar"}
+
+
+def test_pickle():
+    class PickleSecureCookie(SecureCookie):
+        serialization_method = pickle
+
+    secure = PickleSecureCookie({"foo": "bar"}, "secret").serialize()
+    data = PickleSecureCookie.unserialize(secure, "secret")
+    assert data == {"foo": "bar"}