diff --git a/src/_zkapauthorizer/tests/foolscap.py b/src/_zkapauthorizer/tests/foolscap.py index 52b745a6a0b50d154388b0a96aee3b3c20f59fb8..9cef79e99c4cdc687f55ff8d627a646f5036a508 100644 --- a/src/_zkapauthorizer/tests/foolscap.py +++ b/src/_zkapauthorizer/tests/foolscap.py @@ -43,6 +43,7 @@ class StubStorageServer(object): def set_implicit_slot_lease_renewal(self, enabled): pass + def get_anonymous_storage_server(): return StubStorageServer() diff --git a/src/_zkapauthorizer/tests/test_storage_protocol.py b/src/_zkapauthorizer/tests/test_storage_protocol.py index 74b3cadadc52787233b8565c4f037b4602aaf833..32e7e406380eaff0ce140f22b37db0cf394ae0a3 100644 --- a/src/_zkapauthorizer/tests/test_storage_protocol.py +++ b/src/_zkapauthorizer/tests/test_storage_protocol.py @@ -466,10 +466,15 @@ class ShareTests(TestCase): expected_leases = {} # Chop off the non-integer part of the expected values because share # files only keep integer precision. - expected_leases.update({sharenum: [int(when)] for sharenum in existing_sharenums}) - expected_leases.update({ - sharenum: [int(when + interval)] for sharenum in all_sharenums - existing_sharenums - }) + expected_leases.update( + {sharenum: [int(when)] for sharenum in existing_sharenums} + ) + expected_leases.update( + { + sharenum: [int(when + interval)] + for sharenum in all_sharenums - existing_sharenums + } + ) self.assertThat( dict(get_lease_grant_times(self.anonymous_storage_server, storage_index)),