diff --git a/src/_zkapauthorizer/tests/storage_common.py b/src/_zkapauthorizer/tests/storage_common.py index 6538b3bc4afee1cab06b5f9b57c601eb79dbefd8..39d85d25c6132ab0b8a14cddc267787b1e160619 100644 --- a/src/_zkapauthorizer/tests/storage_common.py +++ b/src/_zkapauthorizer/tests/storage_common.py @@ -36,13 +36,17 @@ from .strategies import bytes_for_share # Not really a strategy... LEASE_INTERVAL = 60 * 60 * 24 * 31 -def cleanup_storage_server(storage_server): +def reset_storage_server(storage_server): """ - Delete all of the shares held by the given storage server. + Restore a storage server to a default state. This includes + deleting all of the shares it holds. :param allmydata.storage.server.StorageServer storage_server: The storage server with some on-disk shares to delete. """ + # A storage server is read-write by default. + storage_server.readonly_storage = False + starts = [ FilePath(storage_server.sharedir), FilePath(storage_server.corruption_advisory_dir), diff --git a/src/_zkapauthorizer/tests/test_storage_protocol.py b/src/_zkapauthorizer/tests/test_storage_protocol.py index ed54432b9d36907a4baa826d91331b089306e8db..10e30c0fa51da66f27f975a44bc16dd6689cd65f 100644 --- a/src/_zkapauthorizer/tests/test_storage_protocol.py +++ b/src/_zkapauthorizer/tests/test_storage_protocol.py @@ -61,10 +61,10 @@ from .foolscap import LocalRemote from .matchers import matches_spent_passes, matches_version_dictionary from .storage_common import ( LEASE_INTERVAL, - cleanup_storage_server, get_passes, pass_factory, privacypass_passes, + reset_storage_server, whitebox_write_sparse_share, write_toy_shares, ) @@ -191,7 +191,7 @@ class ShareTests(TestCase): self.spending_recorder.reset() # And clean out any shares that might confuse things. - cleanup_storage_server(self.anonymous_storage_server) + reset_storage_server(self.anonymous_storage_server) def test_get_version(self): """ diff --git a/src/_zkapauthorizer/tests/test_storage_server.py b/src/_zkapauthorizer/tests/test_storage_server.py index a085bc5f8003f218250383ba8474917c3668ef8f..a0237bb7b82cf5c6069d7ae6f243bb37479f22d1 100644 --- a/src/_zkapauthorizer/tests/test_storage_server.py +++ b/src/_zkapauthorizer/tests/test_storage_server.py @@ -45,7 +45,7 @@ from ..storage_common import ( from .common import skipIf from .fixtures import AnonymousStorageServer from .matchers import matches_spent_passes, raises -from .storage_common import cleanup_storage_server, get_passes, write_toy_shares +from .storage_common import get_passes, reset_storage_server, write_toy_shares from .strategies import ( lease_cancel_secrets, lease_renew_secrets, @@ -222,10 +222,7 @@ class PassValidationTests(TestCase): # Hypothesis and testtools fixtures don't play nicely together in a # way that allows us to just move everything from `setUp` into this # method. - cleanup_storage_server(self.anonymous_storage_server) - # One of the tests makes the server read-only partway through. Make - # sure that doesn't leak into other examples. - self.anonymous_storage_server.readonly_storage = False + reset_storage_server(self.anonymous_storage_server) self.spending_recorder.reset()