Ticket #14254: storage_tests.diff

File storage_tests.diff, 6.2 KB (added by steph, 14 years ago)

Patch to add more tests.

  • tests/regressiontests/file_storage/tests.py

    diff --git a/tests/regressiontests/file_storage/tests.py b/tests/regressiontests/file_storage/tests.py
    index 2c5f0f4..23b6e67 100644
    a b import unittest  
    88from cStringIO import StringIO
    99from django.conf import settings
    1010from django.core.exceptions import SuspiciousOperation
    11 from django.core.files.base import ContentFile
     11from django.core.files.base import ContentFile, File
    1212from django.core.files.images import get_image_dimensions
    13 from django.core.files.storage import FileSystemStorage
     13from django.core.files.storage import FileSystemStorage, get_storage_class
    1414from django.core.files.uploadedfile import UploadedFile
     15from django.core.exceptions import ImproperlyConfigured
    1516from unittest import TestCase
    1617try:
    1718    import threading
    except ImportError:  
    2930    except ImportError:
    3031        Image = None
    3132
     33class GetStorageClassTests(unittest.TestCase):
     34    def assertRaisesErrorWithMessage(self, error, message, callable,
     35        *args, **kwargs):
     36        self.assertRaises(error, callable, *args, **kwargs)
     37        try:
     38            callable(*args, **kwargs)
     39        except error, e:
     40            self.assertEqual(message, str(e))
     41
     42    def test_get_filesystem_storage(self):
     43        """
     44        get_storage_class returns the class for a storage backend name/path.
     45        """
     46        self.assertEqual(
     47            get_storage_class('django.core.files.storage.FileSystemStorage'),
     48            FileSystemStorage)
     49
     50    def test_get_invalid_storage_module(self):
     51        """
     52        get_storage_class raises an error if the requested import don't exist.
     53        """
     54        self.assertRaisesErrorWithMessage(
     55            ImproperlyConfigured,
     56            "NonExistingStorage isn't a storage module.",
     57            get_storage_class,
     58            'NonExistingStorage')
     59
     60    def test_get_nonexisting_storage_class(self):
     61        """
     62        get_storage_class raises an error if the requested class don't exist.
     63        """
     64        self.assertRaisesErrorWithMessage(
     65            ImproperlyConfigured,
     66            'Storage module "django.core.files.storage" does not define a '\
     67                '"NonExistingStorage" class.',
     68            get_storage_class,
     69            'django.core.files.storage.NonExistingStorage')
     70
     71    def test_get_nonexisting_storage_module(self):
     72        """
     73        get_storage_class raises an error if the requested module don't exist.
     74        """
     75        self.assertRaisesErrorWithMessage(
     76            ImproperlyConfigured,
     77            'Error importing storage module django.core.files.non_existing_'\
     78                'storage: "No module named non_existing_storage"',
     79            get_storage_class,
     80            'django.core.files.non_existing_storage.NonExistingStorage')
     81
    3282class FileStorageTests(unittest.TestCase):
    3383    storage_class = FileSystemStorage
    3484   
    3585    def setUp(self):
    3686        self.temp_dir = tempfile.mktemp()
    3787        os.makedirs(self.temp_dir)
    38         self.storage = self.storage_class(location=self.temp_dir)
     88        self.storage = self.storage_class(location=self.temp_dir,
     89            base_url='/test_media_url/')
    3990   
    4091    def tearDown(self):
    4192        os.rmdir(self.temp_dir)
    class FileStorageTests(unittest.TestCase):  
    57108        self.storage.delete('storage_test')
    58109        self.failIf(self.storage.exists('storage_test'))
    59110
     111    def test_file_save_without_name(self):
     112        """
     113        File storage extracts the filename from the content object if no
     114        name is given explicitly.
     115        """
     116        self.failIf(self.storage.exists('test.file'))
     117
     118        f = ContentFile('custom contents')
     119        f.name = 'test.file'
     120
     121        storage_f_name = self.storage.save(None, f)
     122
     123        self.assertEqual(storage_f_name, f.name)
     124
     125        self.assert_(os.path.exists(os.path.join(self.temp_dir, f.name)))
     126
     127        self.storage.delete(storage_f_name)
     128
     129    def test_file_path(self):
     130        """
     131        File storage returns the full path of a file
     132        """
     133        self.failIf(self.storage.exists('test.file'))
     134
     135        f = ContentFile('custom contents')
     136        f_name = self.storage.save('test.file', f)
     137
     138        self.assertEqual(self.storage.path(f_name),
     139            os.path.join(self.temp_dir, f_name))
     140
     141        self.storage.delete(f_name)
     142
     143    def test_file_url(self):
     144        """
     145        File storage returns a url to access a given file from the web.
     146        """
     147        self.assertEqual(self.storage.url('test.file'),
     148            '%s%s' % (self.storage.base_url, 'test.file'))
     149
     150        self.storage.base_url = None
     151        self.assertRaises(ValueError, self.storage.url, 'test.file')
     152
     153    def test_file_with_mixin(self):
     154        """
     155        File storage can get a mixin to extend the functionality of the
     156        returned file.
     157        """
     158        self.failIf(self.storage.exists('test.file'))
     159
     160        class TestFileMixin(object):
     161            mixed_in = True
     162
     163        f = ContentFile('custom contents')
     164        f_name = self.storage.save('test.file', f)
     165
     166        self.assert_(isinstance(
     167            self.storage.open('test.file', mixin=TestFileMixin),
     168            TestFileMixin
     169        ))
     170
     171        self.storage.delete('test.file')
     172
     173    def test_listdir(self):
     174        """
     175        File storage returns a tuple containing directories and files.
     176        """
     177        self.failIf(self.storage.exists('storage_test_1'))
     178        self.failIf(self.storage.exists('storage_test_2'))
     179        self.failIf(self.storage.exists('storage_dir_1'))
     180
     181        f = self.storage.save('storage_test_1', ContentFile('custom content'))
     182        f = self.storage.save('storage_test_2', ContentFile('custom content'))
     183        os.mkdir(os.path.join(self.temp_dir, 'storage_dir_1'))
     184
     185        self.assertEqual(
     186            self.storage.listdir(''),
     187            ([u'storage_dir_1'], [u'storage_test_1', u'storage_test_2']))
     188
     189        self.storage.delete('storage_test_1')
     190        self.storage.delete('storage_test_2')
     191        os.rmdir(os.path.join(self.temp_dir, 'storage_dir_1'))
     192
    60193    def test_file_storage_prevents_directory_traversal(self):
    61194        """
    62195        File storage prevents directory traversal (files can only be accessed if
Back to Top