diff --git a/src/_zkapauthorizer/_storage_client.py b/src/_zkapauthorizer/_storage_client.py index 7615bbaf1b2fb08a8f9509e596bacedffc9d67dc..7c7c2a8158090e77dfaac9d9b73169a14844431b 100644 --- a/src/_zkapauthorizer/_storage_client.py +++ b/src/_zkapauthorizer/_storage_client.py @@ -59,6 +59,13 @@ class ZKAPAuthorizerStorageClient(object): def _rref(self): return self._get_rref() + def _get_encoded_tokens(self): + return list( + t.text.encode("ascii") + for t + in self._get_tokens() + ) + def get_version(self): return self._rref.callRemote( "get_version", @@ -75,7 +82,7 @@ class ZKAPAuthorizerStorageClient(object): ): return self._rref.callRemote( "allocate_buckets", - self._get_tokens(), + self._get_encoded_tokens(), storage_index, renew_secret, cancel_secret, @@ -101,7 +108,7 @@ class ZKAPAuthorizerStorageClient(object): ): return self._rref.callRemote( "add_lease", - self._get_tokens(), + self._get_encoded_tokens(), storage_index, renew_secret, cancel_secret, @@ -114,7 +121,7 @@ class ZKAPAuthorizerStorageClient(object): ): return self._rref.callRemote( "renew_lease", - self._get_tokens(), + self._get_encoded_tokens(), storage_index, renew_secret, ) @@ -143,7 +150,7 @@ class ZKAPAuthorizerStorageClient(object): ): return self._rref.callRemote( "slot_testv_and_readv_and_writev", - self._get_tokens(), + self._get_encoded_tokens(), storage_index, secrets, tw_vectors, diff --git a/src/_zkapauthorizer/tests/test_storage_protocol.py b/src/_zkapauthorizer/tests/test_storage_protocol.py index 202d87e52d056bf9cabe58e481dfda4bdef0a54c..421d6e7e342f68a7dc2ac39e85805367fb017056 100644 --- a/src/_zkapauthorizer/tests/test_storage_protocol.py +++ b/src/_zkapauthorizer/tests/test_storage_protocol.py @@ -84,9 +84,12 @@ from ..api import ( ZKAPAuthorizerStorageServer, ZKAPAuthorizerStorageClient, ) -from .._storage_server import ( +from ..foolscap import ( TOKEN_LENGTH, ) +from ..model import ( + Pass, +) class AnonymousStorageServer(Fixture): """ @@ -157,7 +160,7 @@ class ShareTests(TestCase): self.anonymous_storage_server = self.useFixture(AnonymousStorageServer()).storage_server def get_tokens(): - return [b"x" * TOKEN_LENGTH] + return [Pass(u"x" * TOKEN_LENGTH)] self.server = ZKAPAuthorizerStorageServer( self.anonymous_storage_server,