class MultiFolderStorageTest(unittest.TestCase): def setUp(self): self.temp_dir = tempfile.mktemp() os.makedirs(self.temp_dir) self.storage = MultiFolderStorage(location=self.temp_dir) def tearDown(self): shutil.rmtree(self.temp_dir) def test_storage(self): for size in (0, 1, 1000, 5000, 100000): # write file text = 'X' * size content = ContentFile(text) path = self.storage.save('data.txt', content) # open and test file f = self.storage.open(path) self.assertEqual(f.read(), text) # verify that file are saved under 3 sub dirs dir, file = os.path.split(path) num_dir = len(dir.split('/')) self.assertEqual(num_dir, 3)
def setUp(self): self.temp_dir = tempfile.mktemp() os.makedirs(self.temp_dir) self.storage = MultiFolderStorage(location=self.temp_dir)