test_storage: add (failing) test of the BucketWriter/BucketReader implementation

diff --git a/src/allmydata/test/test_storage.py b/src/allmydata/test/test_storage.py
new file mode 100644 (file)
index 0000000..39d1bd1
--- /dev/null
+++ b/src/allmydata/test/test_storage.py
@@ -0,0 +1,43 @@
+
+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)])
+