mirror of
https://github.com/tahoe-lafs/tahoe-lafs.git
synced 2024-12-27 16:28:53 +00:00
44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
|
|
||
|
|
from twisted.trial import unittest
|
||
|
|
|
||
|
|
import os.path
|
||
|
|
from allmydata import storageserver
|
||
|
|
from allmydata.util import fileutil
|
||
|
|
|
||
|
|
class Storage(unittest.TestCase):
|
||
|
|
def make_workdir(self, name):
|
||
|
|
tmpdir = os.path.join("test_storage", "Storage", "tmp", name)
|
||
|
|
basedir = os.path.join("test_storage", "Storage", name)
|
||
|
|
fileutil.make_dirs(tmpdir)
|
||
|
|
fileutil.make_dirs(basedir)
|
||
|
|
return tmpdir, basedir
|
||
|
|
|
||
|
|
def test_create(self):
|
||
|
|
tmpdir, basedir = self.make_workdir("test_create")
|
||
|
|
bw = storageserver.BucketWriter(tmpdir, basedir, 25)
|
||
|
|
bw.remote_put_block(0, "a"*25)
|
||
|
|
bw.remote_put_block(1, "b"*25)
|
||
|
|
bw.remote_put_block(2, "c"*7) # last block may be short
|
||
|
|
bw.remote_close()
|
||
|
|
|
||
|
|
def test_readwrite(self):
|
||
|
|
tmpdir, basedir = self.make_workdir("test_readwrite")
|
||
|
|
bw = storageserver.BucketWriter(tmpdir, basedir, 25)
|
||
|
|
bw.remote_put_block(0, "a"*25)
|
||
|
|
bw.remote_put_block(1, "b"*25)
|
||
|
|
bw.remote_put_block(2, "c"*7) # last block may be short
|
||
|
|
bw.remote_put_block_hashes(["1"*32, "2"*32, "3"*32, "4"*32])
|
||
|
|
bw.remote_put_share_hashes([(5, "5"*32), (6, "6"*32)])
|
||
|
|
bw.remote_close()
|
||
|
|
|
||
|
|
# now read from it
|
||
|
|
br = storageserver.BucketReader(basedir)
|
||
|
|
self.failUnlessEqual(br.remote_get_block(0), "a"*25)
|
||
|
|
self.failUnlessEqual(br.remote_get_block(1), "b"*25)
|
||
|
|
self.failUnlessEqual(br.remote_get_block(2), "c"*7)
|
||
|
|
self.failUnlessEqual(br.remote_get_block_hashes(),
|
||
|
|
["1"*32, "2"*32, "3"*32, "4"*32])
|
||
|
|
self.failUnlessEqual(br.remote_get_share_hashes(),
|
||
|
|
[(5, "5"*32), (6, "6"*32)])
|
||
|
|
|