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