Adds delete_all_shares

This commit is contained in:
Mark Berger 2013-06-24 18:28:05 -04:00 committed by Brian Warner
parent 1467eb53b9
commit 2ab6172302

View File

@ -13,7 +13,7 @@
# Tubs, so it is not useful for tests that involve a Helper, a KeyGenerator, # Tubs, so it is not useful for tests that involve a Helper, a KeyGenerator,
# or the control.furl . # or the control.furl .
import os.path import os
from zope.interface import implements from zope.interface import implements
from twisted.application import service from twisted.application import service
from twisted.internet import defer, reactor from twisted.internet import defer, reactor
@ -401,6 +401,12 @@ class GridTestMixin:
if i_shnum in shnums: if i_shnum in shnums:
os.unlink(i_sharefile) os.unlink(i_sharefile)
def delete_all_shares(self, serverdir):
sharedir = os.path.join(serverdir, "shares")
for prefixdir in os.listdir(sharedir):
if prefixdir != 'incoming':
fileutil.rm_dir(os.path.join(sharedir, prefixdir))
def corrupt_share(self, (shnum, serverid, sharefile), corruptor_function): def corrupt_share(self, (shnum, serverid, sharefile), corruptor_function):
sharedata = open(sharefile, "rb").read() sharedata = open(sharefile, "rb").read()
corruptdata = corruptor_function(sharedata) corruptdata = corruptor_function(sharedata)