def test_put(self):
storage_bucket = tempfile.mkdtemp()
client = FSFileClient(storage_bucket)
client.put("tfile", "file")
fpath = os.path.join(storage_bucket, "tfile")
self.assert_(os.path.isfile(fpath))
self.assertEqual(open(fpath, "r").read(), "file")
def test_get(self):
storage_bucket = tempfile.mkdtemp()
fd, fpath = tempfile.mkstemp(prefix="tfile",
dir=storage_bucket,
text=True)
os.write(fd, "file")
os.close(fd)
client = FSFileClient(storage_bucket)
self.assertIsNone(client.get("notexists"))
self.assertEqual(client.get(os.path.split(fpath)[-1]), "file")
def test_copy_source(self):
storage_bucket = tempfile.mkdtemp()
fd, fpath = tempfile.mkstemp(prefix="tfile",
dir=storage_bucket,
text=True)
client = FSFileClient(storage_bucket)
client.copy_source(os.path.split(fpath)[-1], "new_tfile")
self.assert_(os.path.isfile(fpath))
self.assert_(os.path.isfile(os.path.join(storage_bucket, "new_tfile")))
# clean up
client.delete("new_tfile")
self.assert_(not os.path.isfile(os.path.join(storage_bucket,
"new_tfile")))
def test_delete(self):
storage_bucket = tempfile.mkdtemp()
fd, fpath = tempfile.mkstemp(prefix="tfile",
dir=storage_bucket,
text=True)
os.write(fd, "file")
os.close(fd)
client = FSFileClient(storage_bucket)
client.delete("notexisted")
self.assert_(os.path.isfile(fpath))
client.delete(os.path.split(fpath)[-1])
self.assert_(not os.path.isfile(fpath))
