diff options
Diffstat (limited to 'tests/servers/tests.py')
-rw-r--r-- | tests/servers/tests.py | 170 |
1 files changed, 170 insertions, 0 deletions
diff --git a/tests/servers/tests.py b/tests/servers/tests.py new file mode 100644 index 0000000000..4495596dfa --- /dev/null +++ b/tests/servers/tests.py @@ -0,0 +1,170 @@ +# -*- encoding: utf-8 -*- +""" +Tests for django.core.servers. +""" +from __future__ import unicode_literals + +import os +try: + from urllib.request import urlopen, HTTPError +except ImportError: # Python 2 + from urllib2 import urlopen, HTTPError + +from django.core.exceptions import ImproperlyConfigured +from django.test import LiveServerTestCase +from django.core.servers.basehttp import WSGIServerException +from django.test.utils import override_settings +from django.utils.http import urlencode +from django.utils._os import upath + +from .models import Person + + +TEST_ROOT = os.path.dirname(upath(__file__)) +TEST_SETTINGS = { + 'MEDIA_URL': '/media/', + 'MEDIA_ROOT': os.path.join(TEST_ROOT, 'media'), + 'STATIC_URL': '/static/', + 'STATIC_ROOT': os.path.join(TEST_ROOT, 'static'), +} + + +class LiveServerBase(LiveServerTestCase): + urls = 'regressiontests.servers.urls' + fixtures = ['testdata.json'] + + @classmethod + def setUpClass(cls): + # Override settings + cls.settings_override = override_settings(**TEST_SETTINGS) + cls.settings_override.enable() + super(LiveServerBase, cls).setUpClass() + + @classmethod + def tearDownClass(cls): + # Restore original settings + cls.settings_override.disable() + super(LiveServerBase, cls).tearDownClass() + + def urlopen(self, url): + return urlopen(self.live_server_url + url) + + +class LiveServerAddress(LiveServerBase): + """ + Ensure that the address set in the environment variable is valid. + Refs #2879. + """ + + @classmethod + def setUpClass(cls): + # Backup original environment variable + address_predefined = 'DJANGO_LIVE_TEST_SERVER_ADDRESS' in os.environ + old_address = os.environ.get('DJANGO_LIVE_TEST_SERVER_ADDRESS') + + # Just the host is not accepted + cls.raises_exception('localhost', ImproperlyConfigured) + + # The host must be valid + cls.raises_exception('blahblahblah:8081', WSGIServerException) + + # The list of ports must be in a valid format + cls.raises_exception('localhost:8081,', ImproperlyConfigured) + cls.raises_exception('localhost:8081,blah', ImproperlyConfigured) + cls.raises_exception('localhost:8081-', ImproperlyConfigured) + cls.raises_exception('localhost:8081-blah', ImproperlyConfigured) + cls.raises_exception('localhost:8081-8082-8083', ImproperlyConfigured) + + # If contrib.staticfiles isn't configured properly, the exception + # should bubble up to the main thread. + old_STATIC_URL = TEST_SETTINGS['STATIC_URL'] + TEST_SETTINGS['STATIC_URL'] = None + cls.raises_exception('localhost:8081', ImproperlyConfigured) + TEST_SETTINGS['STATIC_URL'] = old_STATIC_URL + + # Restore original environment variable + if address_predefined: + os.environ['DJANGO_LIVE_TEST_SERVER_ADDRESS'] = old_address + else: + del os.environ['DJANGO_LIVE_TEST_SERVER_ADDRESS'] + + @classmethod + def raises_exception(cls, address, exception): + os.environ['DJANGO_LIVE_TEST_SERVER_ADDRESS'] = address + try: + super(LiveServerAddress, cls).setUpClass() + raise Exception("The line above should have raised an exception") + except exception: + pass + finally: + super(LiveServerAddress, cls).tearDownClass() + + def test_test_test(self): + # Intentionally empty method so that the test is picked up by the + # test runner and the overriden setUpClass() method is executed. + pass + +class LiveServerViews(LiveServerBase): + def test_404(self): + """ + Ensure that the LiveServerTestCase serves 404s. + Refs #2879. + """ + try: + self.urlopen('/') + except HTTPError as err: + self.assertEqual(err.code, 404, 'Expected 404 response') + else: + self.fail('Expected 404 response') + + def test_view(self): + """ + Ensure that the LiveServerTestCase serves views. + Refs #2879. + """ + f = self.urlopen('/example_view/') + self.assertEqual(f.read(), b'example view') + + def test_static_files(self): + """ + Ensure that the LiveServerTestCase serves static files. + Refs #2879. + """ + f = self.urlopen('/static/example_static_file.txt') + self.assertEqual(f.read().rstrip(b'\r\n'), b'example static file') + + def test_media_files(self): + """ + Ensure that the LiveServerTestCase serves media files. + Refs #2879. + """ + f = self.urlopen('/media/example_media_file.txt') + self.assertEqual(f.read().rstrip(b'\r\n'), b'example media file') + + def test_environ(self): + f = self.urlopen('/environ_view/?%s' % urlencode({'q': 'ัะตัั'})) + self.assertIn(b"QUERY_STRING: 'q=%D1%82%D0%B5%D1%81%D1%82'", f.read()) + + +class LiveServerDatabase(LiveServerBase): + + def test_fixtures_loaded(self): + """ + Ensure that fixtures are properly loaded and visible to the + live server thread. + Refs #2879. + """ + f = self.urlopen('/model_view/') + self.assertEqual(f.read().splitlines(), [b'jane', b'robert']) + + def test_database_writes(self): + """ + Ensure that data written to the database by a view can be read. + Refs #2879. + """ + self.urlopen('/create_model_instance/') + self.assertQuerysetEqual( + Person.objects.all().order_by('pk'), + ['jane', 'robert', 'emily'], + lambda b: b.name + ) |