def setUp(self):
self.o = MDOFS(storage_dir="pt_deleteme")
class TestMDOFS(unittest.TestCase):
def setUp(self):
self.o = MDOFS(storage_dir="pt_deleteme")
def tearDown(self):
shutil.rmtree("pt_deleteme")
def test_empty(self):
pass
def test_claim_bucket(self):
a = self.o.claim_bucket()
self.assertTrue(self.o.exists(a))
def test_store_bytes_no_params(self):
a = self.o.claim_bucket()
label = "foo.txt"
b = self.o.put_stream(a, label, "Some bytes to store")
self.assertEquals(b['_label'], "foo.txt")
self.assertEquals(b['_content_length'], 19)
self.assertEquals(b['_checksum'],
'md5:eee89bbbcf416f658c7bc18cd8f2b61d')
def test_store_bytes_with_params(self):
a = self.o.claim_bucket()
label = "foo.txt"
b = self.o.put_stream(a, label, "Some bytes to store", {
"a": "1",
"b": [1, 2, 3, 4, 5]
})
self.assertEquals(b['a'], "1")
self.assertEquals(b['b'], [1, 2, 3, 4, 5])
self.assertEquals(b['_label'], "foo.txt")
self.assertEquals(b['_content_length'], 19)
self.assertEquals(b['_checksum'],
'md5:eee89bbbcf416f658c7bc18cd8f2b61d')
def test_store_params_after_bytes(self):
a = self.o.claim_bucket()
label = "foo.txt"
self.o.put_stream(a, label, "Some bytes to store")
b = self.o.update_metadata(a, label, {"a": "1", "b": [1, 2, 3, 4, 5]})
self.assertEquals(b['a'], "1")
self.assertEquals(b['b'], [1, 2, 3, 4, 5])
def test_params_persistence(self):
a = self.o.claim_bucket()
label = "foo.txt"
self.o.put_stream(a, label, "Some bytes to store", {
"a": "1",
"b": [1, 2, 3, 4, 5]
})
b = self.o.get_metadata(a, label)
self.assertEquals(b['a'], "1")
self.assertEquals(b['b'], [1, 2, 3, 4, 5])
def test_params_deletion(self):
a = self.o.claim_bucket()
label = "foo.txt"
self.o.put_stream(a, label, "Some bytes to store", {
"a": "1",
"b": [1, 2, 3, 4, 5]
})
self.o.del_metadata_keys(a, label, ['b'])
b = self.o.get_metadata(a, label)
self.assertEquals(b['a'], "1")
self.assertFalse(b.has_key('b'))
