From b301627b88f1535d63e2bccbd7845161f7c619b8 Mon Sep 17 00:00:00 2001 From: Alex Gaynor Date: Wed, 13 Jan 2016 09:01:42 -0500 Subject: Added hypothesis tests for padding --- tests/hypothesis/test_padding.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 tests/hypothesis/test_padding.py (limited to 'tests/hypothesis/test_padding.py') diff --git a/tests/hypothesis/test_padding.py b/tests/hypothesis/test_padding.py new file mode 100644 index 00000000..d2b77756 --- /dev/null +++ b/tests/hypothesis/test_padding.py @@ -0,0 +1,20 @@ +# This file is dual licensed under the terms of the Apache License, Version +# 2.0, and the BSD License. See the LICENSE file in the root of this repository +# for complete details. + +from hypothesis import given +from hypothesis.strategies import binary + +from cryptography.hazmat.primitives.padding import PKCS7 + + +@given(binary()) +def test_pkcs7(data): + # TODO: add additional tests with arbitrary block sizes + p = PKCS7(block_size=128) + padder = p.padder() + unpadder = p.unpadder() + + padded = padder.update(data) + padder.finalize() + + assert unpadder.update(padded) + unpadder.finalize() == data -- cgit v1.2.3 From 3066bf4150635a8c6be5ddb8dc4da20721d5daed Mon Sep 17 00:00:00 2001 From: Alex Gaynor Date: Wed, 13 Jan 2016 22:22:18 -0500 Subject: resolve the TODO, random block sizes --- tests/hypothesis/test_padding.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'tests/hypothesis/test_padding.py') diff --git a/tests/hypothesis/test_padding.py b/tests/hypothesis/test_padding.py index d2b77756..21c9a234 100644 --- a/tests/hypothesis/test_padding.py +++ b/tests/hypothesis/test_padding.py @@ -3,15 +3,16 @@ # for complete details. from hypothesis import given -from hypothesis.strategies import binary +from hypothesis.strategies import binary, integers from cryptography.hazmat.primitives.padding import PKCS7 -@given(binary()) -def test_pkcs7(data): - # TODO: add additional tests with arbitrary block sizes - p = PKCS7(block_size=128) +@given(integers(min_value=1, max_value=31), binary()) +def test_pkcs7(block_size, data): + # Generate in [1, 31] so we can easily get block_size in bits by + # multiplying by 8. + p = PKCS7(block_size=block_size * 8) padder = p.padder() unpadder = p.unpadder() -- cgit v1.2.3