Merge pull request #4325 from ydb-platform/mergelibs-240506-1255

Library import 240506-1255

1 files changed, 99 insertions, 0 deletions

diff --git a/contrib/python/itsdangerous/py2/tests/test_itsdangerous/test_signer.py b/contrib/python/itsdangerous/py2/tests/test_itsdangerous/test_signer.py
new file mode 100644
index 0000000000..5f7fe8e159
--- /dev/null
+++ b/contrib/python/itsdangerous/py2/tests/test_itsdangerous/test_signer.py
@@ -0,0 +1,99 @@
+import hashlib
+from functools import partial
+
+import pytest
+
+from itsdangerous.exc import BadSignature
+from itsdangerous.signer import HMACAlgorithm
+from itsdangerous.signer import NoneAlgorithm
+from itsdangerous.signer import Signer
+from itsdangerous.signer import SigningAlgorithm
+
+
+class _ReverseAlgorithm(SigningAlgorithm):
+    def get_signature(self, key, value):
+        return (key + value)[::-1]
+
+
+class TestSigner(object):
+    @pytest.fixture()
+    def signer_factory(self):
+        return partial(Signer, secret_key="secret-key")
+
+    @pytest.fixture()
+    def signer(self, signer_factory):
+        return signer_factory()
+
+    def test_signer(self, signer):
+        signed = signer.sign("my string")
+        assert isinstance(signed, bytes)
+        assert signer.validate(signed)
+        out = signer.unsign(signed)
+        assert out == b"my string"
+
+    def test_no_separator(self, signer):
+        signed = signer.sign("my string")
+        signed = signed.replace(signer.sep, b"*", 1)
+        assert not signer.validate(signed)
+
+        with pytest.raises(BadSignature):
+            signer.unsign(signed)
+
+    def test_broken_signature(self, signer):
+        signed = signer.sign("b")
+        bad_signed = signed[:-1]
+        bad_sig = bad_signed.rsplit(b".", 1)[1]
+        assert not signer.verify_signature(b"b", bad_sig)
+
+        with pytest.raises(BadSignature) as exc_info:
+            signer.unsign(bad_signed)
+
+        assert exc_info.value.payload == b"b"
+
+    def test_changed_value(self, signer):
+        signed = signer.sign("my string")
+        signed = signed.replace(b"my", b"other", 1)
+        assert not signer.validate(signed)
+
+        with pytest.raises(BadSignature):
+            signer.unsign(signed)
+
+    def test_invalid_separator(self, signer_factory):
+        with pytest.raises(ValueError) as exc_info:
+            signer_factory(sep="-")
+
+        assert "separator cannot be used" in str(exc_info.value)
+
+    @pytest.mark.parametrize(
+        "key_derivation", ("concat", "django-concat", "hmac", "none")
+    )
+    def test_key_derivation(self, signer_factory, key_derivation):
+        signer = signer_factory(key_derivation=key_derivation)
+        assert signer.unsign(signer.sign("value")) == b"value"
+
+    def test_invalid_key_derivation(self, signer_factory):
+        signer = signer_factory(key_derivation="invalid")
+
+        with pytest.raises(TypeError):
+            signer.derive_key()
+
+    def test_digest_method(self, signer_factory):
+        signer = signer_factory(digest_method=hashlib.md5)
+        assert signer.unsign(signer.sign("value")) == b"value"
+
+    @pytest.mark.parametrize(
+        "algorithm", (None, NoneAlgorithm(), HMACAlgorithm(), _ReverseAlgorithm())
+    )
+    def test_algorithm(self, signer_factory, algorithm):
+        signer = signer_factory(algorithm=algorithm)
+        assert signer.unsign(signer.sign("value")) == b"value"
+
+        if algorithm is None:
+            assert signer.algorithm.digest_method == signer.digest_method
+
+
+def test_abstract_algorithm():
+    alg = SigningAlgorithm()
+
+    with pytest.raises(NotImplementedError):
+        alg.get_signature("a", "b")