From ec2c31a95a20aa1c63b60d76a541adb083e66ed1 Mon Sep 17 00:00:00 2001
From: Jean-Paul Calderone <exarkun@twistedmatrix.com>
Date: Mon, 27 Sep 2021 09:56:25 -0400
Subject: [PATCH] Pass the test Clock in to our client and server at
 initialization

This is better than stuffing it in later.
---
 src/_zkapauthorizer/tests/test_storage_protocol.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/_zkapauthorizer/tests/test_storage_protocol.py b/src/_zkapauthorizer/tests/test_storage_protocol.py
index f8d57a4..b8e2e7a 100644
--- a/src/_zkapauthorizer/tests/test_storage_protocol.py
+++ b/src/_zkapauthorizer/tests/test_storage_protocol.py
@@ -224,12 +224,14 @@ class ShareTests(TestCase):
             self.anonymous_storage_server,
             self.pass_value,
             self.signing_key,
+            clock=self.clock,
         )
         self.local_remote_server = LocalRemote(self.server)
         self.client = ZKAPAuthorizerStorageClient(
             self.pass_value,
             get_rref=lambda: self.local_remote_server,
             get_passes=self.pass_factory.get,
+            clock=self.clock,
         )
 
     def test_get_version(self):
@@ -1012,10 +1014,6 @@ class ShareTests(TestCase):
         # up between iterations.
         cleanup_storage_server(self.anonymous_storage_server)
 
-        # Make the client and server use our clock.
-        self.server._clock = clock
-        self.client._clock = clock
-
         secrets = (write_enabler, renew_secret, cancel_secret)
 
         def write():
-- 
GitLab