test iter_unsigners

Author: davidism

tests/test_itsdangerous/test_jws.py

@@ -22,6 +22,7 @@ def serializer_factory(self):
     test_signer_cls = None
     test_signer_kwargs = None
     test_fallback_signers = None
+    test_iter_unsigners = None
 
     @pytest.mark.parametrize("algorithm_name", ("HS256", "HS384", "HS512", "none"))
     def test_algorithm(self, serializer_factory, algorithm_name):

tests/test_itsdangerous/test_serializer.py

@@ -6,6 +6,7 @@
 
 import pytest
 
+from itsdangerous import Signer
 from itsdangerous.exc import BadPayload
 from itsdangerous.exc import BadSignature
 from itsdangerous.serializer import Serializer
@@ -144,6 +145,25 @@ def test_fallback_signers(self, serializer_factory, value):
 
         assert fallback_serializer.loads(signed) == value
 
+    def test_iter_unsigners(self, serializer, serializer_factory):
+        class Signer256(serializer.signer):
+            default_digest_method = hashlib.sha256
+
+        serializer = serializer_factory(
+            secret_key="secret_key",
+            fallback_signers=[
+                {"digest_method": hashlib.sha256},
+                (Signer, {"digest_method": hashlib.sha256}),
+                Signer256,
+            ],
+        )
+
+        unsigners = serializer.iter_unsigners()
+        assert next(unsigners).digest_method == hashlib.sha1
+
+        for signer in unsigners:
+            assert signer.digest_method == hashlib.sha256
+
 
 def test_digests():
     factory = partial(Serializer, secret_key="dev key", salt="dev salt")