ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2025-01-31 08:25:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

test_system: factor out find_shares/replace_shares to a common class, so they can be used by other tests

Browse Source
This commit is contained in:
Brian Warner 2008-08-05 18:49:58 -07:00
parent 027408b9d2
commit 3ad0168fbf
2 changed files with 56 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
src/allmydata/test/common.py
View File

@ -745,3 +745,48 @@ N8L+bvLd4BU9g6hRS8b59lQ6GNjryx2bUnCVtLcey4Jd
# To disable the pre-computed tub certs, uncomment this line.
#SYSTEM_TEST_CERTS = []
class ShareManglingMixin(SystemTestMixin):
def find_shares(self, unused=None):
"""Locate shares on disk. Returns a dict that maps
(clientnum,sharenum) to a string that contains the share container
(copied directly from the disk, containing leases etc). You can
modify this dict and then call replace_shares() to modify the shares.
"""
shares = {} # k: (i, sharenum), v: data
for i, c in enumerate(self.clients):
sharedir = c.getServiceNamed("storage").sharedir
for (dirp, dirns, fns) in os.walk(sharedir):
for fn in fns:
try:
sharenum = int(fn)
except TypeError:
# Whoops, I guess that's not a share file then.
pass
else:
data = open(os.path.join(sharedir, dirp, fn), "r").read()
shares[(i, sharenum)] = data
return shares
def replace_shares(self, newshares):
"""Replace shares on disk. Takes a dictionary in the same form
as find_shares() returns."""
for i, c in enumerate(self.clients):
sharedir = c.getServiceNamed("storage").sharedir
for (dirp, dirns, fns) in os.walk(sharedir):
for fn in fns:
try:
sharenum = int(fn)
except TypeError:
# Whoops, I guess that's not a share file then.
pass
else:
pathtosharefile = os.path.join(sharedir, dirp, fn)
os.unlink(pathtosharefile)
newdata = newshares.get((i, sharenum))
if newdata is not None:
open(pathtosharefile, "w").write(newdata)

57
src/allmydata/test/test_system.py
View File

@ -19,7 +19,7 @@ from twisted.python.failure import Failure
from twisted.web.client import getPage
from twisted.web.error import Error
from allmydata.test.common import SystemTestMixin
from allmydata.test.common import SystemTestMixin, ShareManglingMixin
LARGE_DATA = """
This is some data to publish to the virtual drive, which needs to be large
@ -1693,7 +1693,7 @@ class SystemTest(SystemTestMixin, unittest.TestCase):
return d
class Checker(SystemTestMixin, unittest.TestCase):
class Checker(ShareManglingMixin, unittest.TestCase):
def setUp(self):
# Set self.basedir to a temp dir which has the name of the current test method in its
# name.
@ -1714,46 +1714,11 @@ class Checker(SystemTestMixin, unittest.TestCase):
d.addCallback(_stash_it)
return d
def _find_shares(self, unused=None):
shares = {} # k: (i, sharenum), v: data
for i, c in enumerate(self.clients):
sharedir = c.getServiceNamed("storage").sharedir
for (dirp, dirns, fns) in os.walk(sharedir):
for fn in fns:
try:
sharenum = int(fn)
except TypeError:
# Whoops, I guess that's not a share file then.
pass
else:
data = open(os.path.join(sharedir, dirp, fn), "r").read()
shares[(i, sharenum)] = data
return shares
def _replace_shares(self, newshares):
for i, c in enumerate(self.clients):
sharedir = c.getServiceNamed("storage").sharedir
for (dirp, dirns, fns) in os.walk(sharedir):
for fn in fns:
try:
sharenum = int(fn)
except TypeError:
# Whoops, I guess that's not a share file then.
pass
else:
pathtosharefile = os.path.join(sharedir, dirp, fn)
os.unlink(pathtosharefile)
newdata = newshares.get((i, sharenum))
if newdata is not None:
open(pathtosharefile, "w").write(newdata)
def _corrupt_a_share(self, unused=None):
""" Exactly one bit of exactly one share on disk will be flipped (randomly selected from
among the bits of the 'share data' -- the verifiable bits)."""
shares = self._find_shares()
shares = self.find_shares()
ks = shares.keys()
k = random.choice(ks)
data = shares[k]
@ -1765,32 +1730,32 @@ class Checker(SystemTestMixin, unittest.TestCase):
corrupteddata = data[:0xc]+corruptedsharedata+data[0xc+size:]
shares[k] = corrupteddata
self._replace_shares(shares)
self.replace_shares(shares)
def test_test_code(self):
# The following process of stashing the shares, running
# _replace_shares, and asserting that the new set of shares equals the
# replace_shares, and asserting that the new set of shares equals the
# old is more to test this test code than to test the Tahoe code...
d = defer.succeed(None)
d.addCallback(self._find_shares)
d.addCallback(self.find_shares)
stash = [None]
def _stash_it(res):
stash[0] = res
return res
d.addCallback(_stash_it)
d.addCallback(self._replace_shares)
d.addCallback(self.replace_shares)
def _compare(res):
oldshares = stash[0]
self.failUnless(isinstance(oldshares, dict), oldshares)
self.failUnlessEqual(oldshares, res)
d.addCallback(self._find_shares)
d.addCallback(self.find_shares)
d.addCallback(_compare)
d.addCallback(lambda ignore: self._replace_shares({}))
d.addCallback(self._find_shares)
d.addCallback(lambda ignore: self.replace_shares({}))
d.addCallback(self.find_shares)
d.addCallback(lambda x: self.failUnlessEqual(x, {}))
return d
@ -1835,7 +1800,7 @@ class Checker(SystemTestMixin, unittest.TestCase):
d.addCallback(_check2)
return d
d.addCallback(lambda ignore: self._replace_shares({}))
d.addCallback(lambda ignore: self.replace_shares({}))
def _check3(ignored):
before_check_reads = self._count_reads()
d2 = self.filenode.check(verify=False, repair=False)