django-mediasync/mediasync/tests/tests.py

from django.conf import settings
from django.core.exceptions import ImproperlyConfigured
from mediasync import backends
import mediasync
import os
import sys
import unittest

class BaseTestCase(unittest.TestCase):
    
    def testInvalidBackend(self):
        settings.MEDIASYNC = {
            'BACKEND': 'not.a.backend',
        }
        self.assertRaises(ImproperlyConfigured, backends.client)

class DummyBackendTestCase(unittest.TestCase):
    
    def setUp(self):
        settings.DEBUG = False
        settings.MEDIASYNC = {
            'BACKEND': 'mediasync.backends.dummy',
        }
        self.client = backends.client()
    
    def testPush(self):
        
        def callback(*args):
            pass
        
        self.client.put_callback = callback
        mediasync.sync(self.client)
    
    def testJoinedPush(self):
        pass
    
class S3BackendTestCase(unittest.TestCase):

    def setUp(self):
        settings.DEBUG = False
        settings.MEDIASYNC = {
            'BACKEND': 'mediasync.backends.s3',
            'AWS_BUCKET': 'mediasync_test',
            'AWS_KEY': os.environ['AWS_KEY'],
            'AWS_SECRET': os.environ['AWS_SECRET'],
        }
        self.client = backends.client()
        
    def testMediaURL(self):
        
        try:
            del settings.MEDIASYNC['SERVE_REMOTE']
        except KeyError:
            pass
            
        settings.DEBUG = True
        self.assertEqual(backends.client().media_url(), '/media')
        
        settings.DEBUG = False
        self.assertEqual(backends.client().media_url(), 'http://mediasync_test.s3.amazonaws.com')
    
    def testServeRemote(self):
        
        settings.DEBUG = False
        settings.MEDIASYNC['SERVE_REMOTE'] = False
        self.assertEqual(backends.client().media_url(), '/media')
        
        settings.DEBUG = True
        settings.MEDIASYNC['SERVE_REMOTE'] = True
        self.assertEqual(backends.client().media_url(), 'http://mediasync_test.s3.amazonaws.com')
    
    def testMissingBucket(self):
        del settings.MEDIASYNC['AWS_BUCKET']
        self.assertRaises(AssertionError, backends.client)

class ProcessorTestCase(unittest.TestCase):
    
    def setUp(self):
        settings.MEDIASYNC['PROCESSORS'] = (
            'mediasync.processors.js_minifier',
            lambda fd, ct, rp, r: fd.upper(),
        )
        self.client = backends.client()
    
    def testProcessor(self):
        content = """var foo = function() {
            alert(1);
        };"""
        ct = 'text/javascript'
        procd = self.client.process(content, ct, 'test.js')
        self.assertEqual(procd, "VAR FOO = FUNCTION(){ALERT(1)};")