Moved tests to package

byostorage/tests/test_byo_storage.py

@@ -0,0 +1,110 @@
+from django.test import TestCase
+from django.core.files.storage import FileSystemStorage
+from django.core.exceptions import ValidationError
+
+from byostorage.models import UserStorage
+from byostorage.user import BYOStorage
+
+from example.models import LinkedStorageTestModel
+
+import tempfile
+import os.path
+
+#import logging
+#logging.basicConfig(level=logging.DEBUG)
+
+class UserStorageTestCase(TestCase):
+
+    def test_construction(self):
+        us = UserStorage.objects.create(name='one', storage='django.core.files.storage.FileSystemStorage',
+                settings_data='{"location": "storage/one"}')
+
+        self.assertEqual(str(us), 'one')
+
+        storage = us.instance()
+        self.assertIsInstance(storage, FileSystemStorage)
+        self.assertEqual(storage.base_location, 'storage/one')
+
+    def test_bad_json(self):
+        with self.assertRaisesMessage(ValidationError, "Unterminated string"):
+            UserStorage.objects.create(name='one', storage='django.core.files.storage.FileSystemStorage',
+            settings_data='{"location": "foo}')
+
+    def test_connection(self):
+        instance = UserStorage.objects.create(name='one', storage='storages.backends.s3boto3.S3Boto3Storage',
+                settings_data='''
+{
+  "access_key": "minioadmin",
+  "secret_key": "minioadmin",
+  "endpoint_url": "http://localhost:9000",
+  "bucket_name": "byostorage_test"
+}
+''')
+
+    
+    def test_bad_connection(self):
+        try:
+            UserStorage.objects.create(name='one', storage='storages.backends.s3boto3.S3Boto3Storage',
+                   settings_data='''
+{
+  "access_key": "minioadmin",
+  "secret_key": "the_wrong_secret",
+  "endpoint_url": "http://localhost:9000",
+  "bucket_name": "byostorage_test"
+}
+''')
+            self.fail("Should have raised an exception")
+        except Exception as e:
+            pass
+
+
+class BYOStorageTestCase(TestCase):
+
+    def setUp(self):
+        UserStorage.objects.create(name='one', storage='django.core.files.storage.FileSystemStorage',
+                settings_data='{"location": "storage/one"}')
+    
+    def test_pickup_changes(self):
+        storage = BYOStorage()
+        one = storage.get_storage('one')
+        self.assertEqual(one.base_location, 'storage/one')
+
+        # if not already in cache then no issues
+        UserStorage.objects.create(name='two', storage='django.core.files.storage.FileSystemStorage',
+                settings_data='{"location": "storage/two"}')
+        two = storage.get_storage('two')
+        self.assertEqual(two.base_location, 'storage/two')
+
+        # modify the settings
+        o = UserStorage.objects.get(name='one')
+        o.settings_data = '{"location": "other/one"}'
+        o.save()
+
+        # signal should have fired and invalidated 'one'
+        one = storage.get_storage('one')
+        self.assertEqual(one.base_location, 'other/one')
+
+class BYOStorageFieldTestCase(TestCase):
+
+    def setUp(self):
+        self.test_dir = tempfile.TemporaryDirectory()
+        self.one = UserStorage.objects.create(name='one', storage='django.core.files.storage.FileSystemStorage',
+            settings_data = '{"location": "%s/one"}' % self.test_dir.name)
+
+    def tearDown(self):
+        self.test_dir.cleanup()
+
+    def test_linked_storage(self):
+        
+        from django.core.files.uploadedfile import UploadedFile
+
+        with open(__file__, 'rb') as f:
+            result = LinkedStorageTestModel.objects.create(name="first", my_storage_id=self.one.pk, my_file=UploadedFile(f))
+
+        expected = os.path.basename(__file__)
+
+        self.assertEqual(result.my_file.name, expected)
+        self.assertGreater(result.my_file.size, 0)
+
+        self.assertEqual(os.listdir(os.path.join(self.test_dir.name, 'one')), [expected])
+

byostorage/tests/test_http_storage.py

@@ -0,0 +1,28 @@
+from django.test import TestCase
+
+from byostorage.http import HTTPStorage
+
+class HTTPStorageTestCase(TestCase):
+
+    def test_url(self):
+        s = HTTPStorage()
+
+        self.assertEqual(s.url('//google.com'), 'https://google.com')
+
+    def test_exists(self):
+        s = HTTPStorage()
+
+        self.assertTrue(s.exists('//gitea.tfconsulting.com.au/tris'))
+        self.assertFalse(s.exists('//gitea.tfconsulting.com.au/foo.txt'))
+
+    def test_save(self):
+        s = HTTPStorage()
+        with self.assertRaisesMessage(NotImplementedError, "Unable to save to web locations"):
+            s.save('//gitea.tfconsulting.com.au/foo', 'Some content')
+
+    def test_open(self):
+        s = HTTPStorage()
+        with s.open('//gitea.tfconsulting.com.au', 'rb') as f:
+            data = f.read()
+
+        self.assertTrue(len(data) > 4000)

byostorage/tests/test_multi_storage.py

@@ -0,0 +1,36 @@
+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')