def test_init(self): StorageClass = RandomFilenameMetaStorage(storage_class=StubStorage) with patch(settings, RANDOM_FILENAME_LENGTH=NotImplemented): storage = StorageClass() self.assertFilename(storage.get_available_name(''), '') storage = StorageClass(randomfilename_length=10) self.assertFilename(storage.get_available_name(''), '', length=10) with patch(settings, RANDOM_FILENAME_LENGTH=5): storage = StorageClass() self.assertFilename(storage.get_available_name(''), '', length=5) storage = StorageClass(randomfilename_length=20) self.assertFilename(storage.get_available_name(''), '', length=20)
def test_get_available_name(self): with media_root(): storage = RandomFilenameFileSystemStorage( randomfilename_length=DEFAULT_LENGTH) self.assertFilename(storage.get_available_name(''), '') self.assertFilename(storage.get_available_name('foo'), 'foo') self.assertFilename(storage.get_available_name('foo.txt'), 'foo.txt') self.assertFilename(storage.get_available_name('foo/bar'), 'foo/bar') self.assertFilename(storage.get_available_name('foo/bar.txt'), 'foo/bar.txt')
def test_get_available_name(self): with media_root(): storage = RandomFilenameFileSystemStorage( randomfilename_length=DEFAULT_LENGTH ) self.assertFilename(storage.get_available_name(''), '') self.assertFilename(storage.get_available_name('foo'), 'foo') self.assertFilename(storage.get_available_name('foo.txt'), 'foo.txt') self.assertFilename(storage.get_available_name('foo/bar'), 'foo/bar') self.assertFilename(storage.get_available_name('foo/bar.txt'), 'foo/bar.txt')
def test_get_available_name_retry(self): # With retries StorageClass = RandomFilenameMetaStorage(storage_class=StubStorage) storage = StorageClass(tries=2) stub_random_string.count = 0 with patch(django_randomfilenamestorage.storage, random_string=stub_random_string): self.assertEqual(storage.get_available_name('name.txt'), '2.txt') # Without retries StorageClass = RandomFilenameMetaStorage(storage_class=StubStorage) storage = StorageClass(tries=2) stub_random_string.count = 0 with patch(django_randomfilenamestorage.storage, random_string=stub_random_string): self.assertEqual( storage.get_available_name('name.txt', retry=False), '1.txt')
def test_get_available_name_retry(self): # With retries StorageClass = RandomFilenameMetaStorage(storage_class=StubStorage) storage = StorageClass(tries=2) stub_random_string.count = 0 with patch(django_randomfilenamestorage.storage, random_string=stub_random_string): self.assertEqual(storage.get_available_name('name.txt'), '2.txt') # Without retries StorageClass = RandomFilenameMetaStorage(storage_class=StubStorage) storage = StorageClass(tries=2) stub_random_string.count = 0 with patch(django_randomfilenamestorage.storage, random_string=stub_random_string): self.assertEqual(storage.get_available_name('name.txt', retry=False), '1.txt')