def test_put_node_ok(self):
# write it
node_id = "dinw78cdync8"
ds = DiskStorage(self.tmpdir)
data = 'test content to write'
consumer = ds.put(node_id)
consumer.write(data)
consumer.unregisterProducer()
# check the file
path = ds._get_treepath(node_id)
with open(os.path.join(path, node_id), 'rb') as fh:
written = fh.read()
self.assertEqual(written, data)
def test_put_node_resumed(self):
# write some
node_id = "dinw78cdync8"
ds = DiskStorage(self.tmpdir)
data1 = b'test content to write part 1'
consumer = ds.put(node_id)
consumer.write(data1)
consumer.unregisterProducer()
# write more
data2 = b' and part 2'
consumer = ds.put(node_id, len(data1))
consumer.write(data2)
consumer.unregisterProducer()
# check the file
path = ds._get_treepath(node_id)
with open(os.path.join(path, node_id), 'rb') as fh:
written = fh.read()
self.assertEqual(written, data1 + data2)
