Test showing BytestringSplitter taking arbitrary bytes.

pull/116/head
jMyles 2017-11-11 13:21:24 -08:00
parent 4f803ba2df
commit 1ed6e6d30b
1 changed files with 11 additions and 1 deletions

View File

@ -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)