From 00387ea1d02a5800ff4480a3a177ecc87b34532f Mon Sep 17 00:00:00 2001 From: Jean-Paul Calderone <exarkun@twistedmatrix.com> Date: Fri, 27 Sep 2019 12:06:31 -0400 Subject: [PATCH] Have the tests validate the message for us --- src/_zkapauthorizer/tests/test_storage_protocol.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/_zkapauthorizer/tests/test_storage_protocol.py b/src/_zkapauthorizer/tests/test_storage_protocol.py index 73f9017..0bd6cb8 100644 --- a/src/_zkapauthorizer/tests/test_storage_protocol.py +++ b/src/_zkapauthorizer/tests/test_storage_protocol.py @@ -161,6 +161,13 @@ class ShareTests(TestCase): self.anonymous_storage_server = self.useFixture(AnonymousStorageServer()).storage_server def get_passes(message, count): + if not isinstance(message, bytes): + raise TypeError("message must be bytes") + try: + message.decode("utf-8") + except UnicodeDecodeError: + raise TypeError("message must be valid utf-8") + return [Pass(u"x" * TOKEN_LENGTH)] * count self.server = ZKAPAuthorizerStorageServer( -- GitLab