37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from django.test import TestCase
|
|
from byostorage.multi import MultiStorage
|
|
from django.core.files.storage import FileSystemStorage
|
|
|
|
|
|
class MultiStorageTestCase(TestCase):
|
|
|
|
def test_storage_selection(self):
|
|
ms = MultiStorage({'one': {'location': 'storage/one'}, 'two': {}})
|
|
self.assertEqual(str(ms), "<MultiStorage: 2 storages>")
|
|
|
|
# use the default if nothing set
|
|
default = ms.get_storage('foo')
|
|
self.assertIsInstance(default, FileSystemStorage)
|
|
self.assertEqual(default.base_location, 'media')
|
|
|
|
# get first instance
|
|
one = ms.get_storage('one')
|
|
self.assertEqual(one.base_location, 'storage/one')
|
|
|
|
def test_storage_proxies(self):
|
|
ms = MultiStorage({
|
|
'one': {'location': 'storage/one', 'base_url': 'http://one'},
|
|
'two': {'location': 'storage/two', 'base_url': 'http://two'}
|
|
})
|
|
|
|
|
|
self.assertEqual(ms.url("one:foo.txt"), 'http://one/foo.txt')
|
|
self.assertEqual(ms.url("two:foo.txt"), 'http://two/foo.txt')
|
|
|
|
def test_with_http(self):
|
|
ms = MultiStorage({
|
|
'https': {'storage': 'byostorage.http.HTTPStorage', 'protocol': 'https'}
|
|
})
|
|
|
|
self.assertEqual(ms.url('https://google.com'), 'https://google.com')
|