tahoe-lafs/tahoe-lafs
tahoe-lafs
/
tahoe-lafs
2
0
Fork 0
Code Issues Pull Requests 23 Packages Projects Releases 1 Wiki Activity

test/common.py: update FakeMutableFileNode to new contents= callable scheme

This commit is contained in:
Brian Warner 2009-10-12 22:21:54 -07:00
parent 480e1d318d
commit e63f59f50c
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -157,13 +157,22 @@ class FakeMutableFileNode:
def __init__(self, storage_broker, secret_holder, def __init__(self, storage_broker, secret_holder,
default_encoding_parameters, history): default_encoding_parameters, history):
self.init_from_uri(make_mutable_file_uri()) self.init_from_uri(make_mutable_file_uri())
def create(self, initial_contents, key_generator=None, keysize=None): def create(self, contents, key_generator=None, keysize=None):
initial_contents = self._get_initial_contents(contents)
if len(initial_contents) > self.MUTABLE_SIZELIMIT: if len(initial_contents) > self.MUTABLE_SIZELIMIT:
raise FileTooLargeError("SDMF is limited to one segment, and " raise FileTooLargeError("SDMF is limited to one segment, and "
"%d > %d" % (len(initial_contents), "%d > %d" % (len(initial_contents),
self.MUTABLE_SIZELIMIT)) self.MUTABLE_SIZELIMIT))
self.all_contents[self.storage_index] = initial_contents self.all_contents[self.storage_index] = initial_contents
return defer.succeed(self) return defer.succeed(self)
def _get_initial_contents(self, contents):
if isinstance(contents, str):
return contents
if contents is None:
return ""
assert callable(contents), "%s should be callable, not %s" % \
(contents, type(contents))
return contents(self)
def init_from_uri(self, filecap): def init_from_uri(self, filecap):
assert isinstance(filecap, str) assert isinstance(filecap, str)
if filecap.startswith("URI:SSK:"): if filecap.startswith("URI:SSK:"):