def test_filehandler_get_key_hash(self):
from pewtils.io import FileHandler
h = FileHandler("tests/files", use_s3=False)
self.assertEqual(
h.get_key_hash("temp"),
"c51bf90ccb22befa316b7a561fe9d5fd9650180b14421fc6d71bcd57",
)
self.assertEqual(
h.get_key_hash({"key": "value"}),
"37e13e1116c86a6e9f3f8926375c7cb977ca74d2d598572ced03cd09",
)
def test_clear_file(self):
from pewtils.io import FileHandler
h = FileHandler("tests/files/temp", use_s3=False)
with closing(open("tests/files/temp/temp.txt", "wb")) as output:
output.write(b"test")
h.clear_file("temp", format="txt")
files = []
for file in h.iterate_path():
files.append(file)
self.assertNotIn("temp.txt", files)
self.assertEqual(len(files), 0)
os.rmdir("tests/files/temp")
h = FileHandler("tests/files/temp", use_s3=False)
key = h.get_key_hash("temp")
with closing(open("tests/files/temp/{}.txt".format(key),
"wb")) as output:
output.write(b"test")
h.clear_file("temp", format="txt", hash_key=True)
files = []
for file in h.iterate_path():
files.append(file)
self.assertNotIn("{}.txt".format(key), files)
self.assertEqual(len(files), 0)
os.rmdir("tests/files/temp")