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")