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)