From 7c5f131417049120c968fc047ef63cb25d245d2d Mon Sep 17 00:00:00 2001 From: Paul Kehrer Date: Thu, 1 May 2014 08:27:28 -0500 Subject: move use after finalize to its own test --- tests/hazmat/primitives/test_dsa.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/tests/hazmat/primitives/test_dsa.py b/tests/hazmat/primitives/test_dsa.py index 3f65fd5a..67f90295 100644 --- a/tests/hazmat/primitives/test_dsa.py +++ b/tests/hazmat/primitives/test_dsa.py @@ -761,10 +761,19 @@ class TestDSAVerification(object): verifier.verify() else: verifier.verify() - with pytest.raises(AlreadyFinalized): - verifier.verify() - with pytest.raises(AlreadyFinalized): - verifier.update(b"more data") + + def test_use_after_finalize(self, backend): + parameters = dsa.DSAParameters.generate(1024, backend) + private_key = dsa.DSAPrivateKey.generate(parameters, backend) + public_key = private_key.public_key() + verifier = public_key.verifier(b'fakesig', hashes.SHA1(), backend) + verifier.update(b'irrelevant') + with pytest.raises(InvalidSignature): + verifier.verify() + with pytest.raises(AlreadyFinalized): + verifier.verify() + with pytest.raises(AlreadyFinalized): + verifier.update(b"more data") def test_dsa_verifier_invalid_backend(self, backend): pretend_backend = object() -- cgit v1.2.3