def test_get_node_ok(self):
# write a fake file
node_id = "dinw78cdync8"
path = DiskStorage(self.tmpdir)._get_treepath(node_id)
os.makedirs(path)
data = 'test content'
with open(os.path.join(path, node_id), 'wb') as fh:
fh.write(data)
# get it
ds = DiskStorage(self.tmpdir)
producer = ds.get(node_id)
consumer = StringIO.StringIO()
yield producer.startProducing(consumer)
self.assertEqual(consumer.getvalue(), data)
