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)