From 1ed6e6d30b799fa58c2264bbefab26e448fe4a2d Mon Sep 17 00:00:00 2001 From: jMyles Date: Sat, 11 Nov 2017 13:21:24 -0800 Subject: [PATCH] Test showing BytestringSplitter taking arbitrary bytes. --- tests/crypto/test_bytestring_types.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/tests/crypto/test_bytestring_types.py b/tests/crypto/test_bytestring_types.py index 240eab6a1..df211d134 100644 --- a/tests/crypto/test_bytestring_types.py +++ b/tests/crypto/test_bytestring_types.py @@ -8,6 +8,16 @@ def test_split_two_signatures(): We make two random Signatures and concat them. Then split them and show that we got the proper result. """ sig1, sig2 = Signature(secure_random(65)), Signature(secure_random(65)) + sigs_concatted = sig1 + sig2 two_signature_splitter = BytestringSplitter(Signature, Signature) - rebuilt_sig1, rebuilt_sig2 = two_signature_splitter(sig1 + sig2) + rebuilt_sig1, rebuilt_sig2 = two_signature_splitter(sigs_concatted) assert (sig1, sig2) == (rebuilt_sig1, rebuilt_sig2) + + +def test_split_signature_from_arbitrary_bytes(): + how_many_bytes = 10 + signature = Signature(secure_random(65)) + some_bytes = secure_random(how_many_bytes) + splitter = BytestringSplitter(Signature, how_many_bytes) + rebuilt_signature, rebuilt_bytes = splitter(signature + some_bytes) +