From d8341bf02444e58ff62afe04f15d9679a1b5b8f8 Mon Sep 17 00:00:00 2001 From: Tim Graham Date: Tue, 10 Feb 2015 07:51:02 -0500 Subject: Moved contrib.sites tests out of contrib. --- django/contrib/sites/tests.py | 195 ------------------------------------------ tests/sites_tests/__init__.py | 0 tests/sites_tests/tests.py | 194 +++++++++++++++++++++++++++++++++++++++++ 3 files changed, 194 insertions(+), 195 deletions(-) delete mode 100644 django/contrib/sites/tests.py create mode 100644 tests/sites_tests/__init__.py create mode 100644 tests/sites_tests/tests.py diff --git a/django/contrib/sites/tests.py b/django/contrib/sites/tests.py deleted file mode 100644 index f58c807622..0000000000 --- a/django/contrib/sites/tests.py +++ /dev/null @@ -1,195 +0,0 @@ -from __future__ import unicode_literals - -from django.apps import apps -from django.conf import settings -from django.core.exceptions import ObjectDoesNotExist, ValidationError -from django.db.models.signals import post_migrate -from django.http import HttpRequest -from django.test import TestCase, modify_settings, override_settings -from django.test.utils import captured_stdout - -from . import models -from .management import create_default_site -from .middleware import CurrentSiteMiddleware -from .models import Site, clear_site_cache -from .requests import RequestSite -from .shortcuts import get_current_site - - -@modify_settings(INSTALLED_APPS={'append': 'django.contrib.sites'}) -class SitesFrameworkTests(TestCase): - - def setUp(self): - self.site = Site( - id=settings.SITE_ID, - domain="example.com", - name="example.com", - ) - self.site.save() - - def test_site_manager(self): - # Make sure that get_current() does not return a deleted Site object. - s = Site.objects.get_current() - self.assertIsInstance(s, Site) - s.delete() - self.assertRaises(ObjectDoesNotExist, Site.objects.get_current) - - def test_site_cache(self): - # After updating a Site object (e.g. via the admin), we shouldn't return a - # bogus value from the SITE_CACHE. - site = Site.objects.get_current() - self.assertEqual("example.com", site.name) - s2 = Site.objects.get(id=settings.SITE_ID) - s2.name = "Example site" - s2.save() - site = Site.objects.get_current() - self.assertEqual("Example site", site.name) - - def test_delete_all_sites_clears_cache(self): - # When all site objects are deleted the cache should also - # be cleared and get_current() should raise a DoesNotExist. - self.assertIsInstance(Site.objects.get_current(), Site) - Site.objects.all().delete() - self.assertRaises(Site.DoesNotExist, Site.objects.get_current) - - @override_settings(ALLOWED_HOSTS=['example.com']) - def test_get_current_site(self): - # Test that the correct Site object is returned - request = HttpRequest() - request.META = { - "SERVER_NAME": "example.com", - "SERVER_PORT": "80", - } - site = get_current_site(request) - self.assertIsInstance(site, Site) - self.assertEqual(site.id, settings.SITE_ID) - - # Test that an exception is raised if the sites framework is installed - # but there is no matching Site - site.delete() - self.assertRaises(ObjectDoesNotExist, get_current_site, request) - - # A RequestSite is returned if the sites framework is not installed - with self.modify_settings(INSTALLED_APPS={'remove': 'django.contrib.sites'}): - site = get_current_site(request) - self.assertIsInstance(site, RequestSite) - self.assertEqual(site.name, "example.com") - - @override_settings(SITE_ID='', ALLOWED_HOSTS=['example.com']) - def test_get_current_site_no_site_id(self): - request = HttpRequest() - request.META = { - "SERVER_NAME": "example.com", - "SERVER_PORT": "80", - } - del settings.SITE_ID - site = get_current_site(request) - self.assertEqual(site.name, "example.com") - - def test_domain_name_with_whitespaces(self): - # Regression for #17320 - # Domain names are not allowed contain whitespace characters - site = Site(name="test name", domain="test test") - self.assertRaises(ValidationError, site.full_clean) - site.domain = "test\ttest" - self.assertRaises(ValidationError, site.full_clean) - site.domain = "test\ntest" - self.assertRaises(ValidationError, site.full_clean) - - def test_clear_site_cache(self): - request = HttpRequest() - request.META = { - "SERVER_NAME": "example.com", - "SERVER_PORT": "80", - } - self.assertEqual(models.SITE_CACHE, {}) - get_current_site(request) - expected_cache = {self.site.id: self.site} - self.assertEqual(models.SITE_CACHE, expected_cache) - - with self.settings(SITE_ID=''): - get_current_site(request) - - expected_cache.update({self.site.domain: self.site}) - self.assertEqual(models.SITE_CACHE, expected_cache) - - clear_site_cache(Site, instance=self.site) - self.assertEqual(models.SITE_CACHE, {}) - - -class JustOtherRouter(object): - def allow_migrate(self, db, model): - return db == 'other' - - -@modify_settings(INSTALLED_APPS={'append': 'django.contrib.sites'}) -class CreateDefaultSiteTests(TestCase): - multi_db = True - - def setUp(self): - self.app_config = apps.get_app_config('sites') - # Delete the site created as part of the default migration process. - Site.objects.all().delete() - - def test_basic(self): - """ - #15346, #15573 - create_default_site() creates an example site only if - none exist. - """ - with captured_stdout() as stdout: - create_default_site(self.app_config) - self.assertEqual(Site.objects.count(), 1) - self.assertIn("Creating example.com", stdout.getvalue()) - - with captured_stdout() as stdout: - create_default_site(self.app_config) - self.assertEqual(Site.objects.count(), 1) - self.assertEqual("", stdout.getvalue()) - - @override_settings(DATABASE_ROUTERS=[JustOtherRouter()]) - def test_multi_db(self): - """ - #16353, #16828 - The default site creation should respect db routing. - """ - create_default_site(self.app_config, using='default', verbosity=0) - create_default_site(self.app_config, using='other', verbosity=0) - self.assertFalse(Site.objects.using('default').exists()) - self.assertTrue(Site.objects.using('other').exists()) - - def test_save_another(self): - """ - #17415 - Another site can be created right after the default one. - - On some backends the sequence needs to be reset after saving with an - explicit ID. Test that there isn't a sequence collisions by saving - another site. This test is only meaningful with databases that use - sequences for automatic primary keys such as PostgreSQL and Oracle. - """ - create_default_site(self.app_config, verbosity=0) - Site(domain='example2.com', name='example2.com').save() - - def test_signal(self): - """ - #23641 - Sending the ``post_migrate`` signal triggers creation of the - default site. - """ - post_migrate.send(sender=self.app_config, app_config=self.app_config, verbosity=0) - self.assertTrue(Site.objects.exists()) - - @override_settings(SITE_ID=35696) - def test_custom_site_id(self): - """ - #23945 - The configured ``SITE_ID`` should be respected. - """ - create_default_site(self.app_config, verbosity=0) - self.assertEqual(Site.objects.get().pk, 35696) - - -class MiddlewareTest(TestCase): - - def test_request(self): - """ Makes sure that the request has correct `site` attribute. """ - middleware = CurrentSiteMiddleware() - request = HttpRequest() - middleware.process_request(request) - self.assertEqual(request.site.id, settings.SITE_ID) diff --git a/tests/sites_tests/__init__.py b/tests/sites_tests/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/tests/sites_tests/tests.py b/tests/sites_tests/tests.py new file mode 100644 index 0000000000..eeb6a44646 --- /dev/null +++ b/tests/sites_tests/tests.py @@ -0,0 +1,194 @@ +from __future__ import unicode_literals + +from django.apps import apps +from django.conf import settings +from django.contrib.sites import models +from django.contrib.sites.management import create_default_site +from django.contrib.sites.middleware import CurrentSiteMiddleware +from django.contrib.sites.models import Site, clear_site_cache +from django.contrib.sites.requests import RequestSite +from django.contrib.sites.shortcuts import get_current_site +from django.core.exceptions import ObjectDoesNotExist, ValidationError +from django.db.models.signals import post_migrate +from django.http import HttpRequest +from django.test import TestCase, modify_settings, override_settings +from django.test.utils import captured_stdout + + +@modify_settings(INSTALLED_APPS={'append': 'django.contrib.sites'}) +class SitesFrameworkTests(TestCase): + + def setUp(self): + self.site = Site( + id=settings.SITE_ID, + domain="example.com", + name="example.com", + ) + self.site.save() + + def test_site_manager(self): + # Make sure that get_current() does not return a deleted Site object. + s = Site.objects.get_current() + self.assertIsInstance(s, Site) + s.delete() + self.assertRaises(ObjectDoesNotExist, Site.objects.get_current) + + def test_site_cache(self): + # After updating a Site object (e.g. via the admin), we shouldn't return a + # bogus value from the SITE_CACHE. + site = Site.objects.get_current() + self.assertEqual("example.com", site.name) + s2 = Site.objects.get(id=settings.SITE_ID) + s2.name = "Example site" + s2.save() + site = Site.objects.get_current() + self.assertEqual("Example site", site.name) + + def test_delete_all_sites_clears_cache(self): + # When all site objects are deleted the cache should also + # be cleared and get_current() should raise a DoesNotExist. + self.assertIsInstance(Site.objects.get_current(), Site) + Site.objects.all().delete() + self.assertRaises(Site.DoesNotExist, Site.objects.get_current) + + @override_settings(ALLOWED_HOSTS=['example.com']) + def test_get_current_site(self): + # Test that the correct Site object is returned + request = HttpRequest() + request.META = { + "SERVER_NAME": "example.com", + "SERVER_PORT": "80", + } + site = get_current_site(request) + self.assertIsInstance(site, Site) + self.assertEqual(site.id, settings.SITE_ID) + + # Test that an exception is raised if the sites framework is installed + # but there is no matching Site + site.delete() + self.assertRaises(ObjectDoesNotExist, get_current_site, request) + + # A RequestSite is returned if the sites framework is not installed + with self.modify_settings(INSTALLED_APPS={'remove': 'django.contrib.sites'}): + site = get_current_site(request) + self.assertIsInstance(site, RequestSite) + self.assertEqual(site.name, "example.com") + + @override_settings(SITE_ID='', ALLOWED_HOSTS=['example.com']) + def test_get_current_site_no_site_id(self): + request = HttpRequest() + request.META = { + "SERVER_NAME": "example.com", + "SERVER_PORT": "80", + } + del settings.SITE_ID + site = get_current_site(request) + self.assertEqual(site.name, "example.com") + + def test_domain_name_with_whitespaces(self): + # Regression for #17320 + # Domain names are not allowed contain whitespace characters + site = Site(name="test name", domain="test test") + self.assertRaises(ValidationError, site.full_clean) + site.domain = "test\ttest" + self.assertRaises(ValidationError, site.full_clean) + site.domain = "test\ntest" + self.assertRaises(ValidationError, site.full_clean) + + def test_clear_site_cache(self): + request = HttpRequest() + request.META = { + "SERVER_NAME": "example.com", + "SERVER_PORT": "80", + } + self.assertEqual(models.SITE_CACHE, {}) + get_current_site(request) + expected_cache = {self.site.id: self.site} + self.assertEqual(models.SITE_CACHE, expected_cache) + + with self.settings(SITE_ID=''): + get_current_site(request) + + expected_cache.update({self.site.domain: self.site}) + self.assertEqual(models.SITE_CACHE, expected_cache) + + clear_site_cache(Site, instance=self.site) + self.assertEqual(models.SITE_CACHE, {}) + + +class JustOtherRouter(object): + def allow_migrate(self, db, model): + return db == 'other' + + +@modify_settings(INSTALLED_APPS={'append': 'django.contrib.sites'}) +class CreateDefaultSiteTests(TestCase): + multi_db = True + + def setUp(self): + self.app_config = apps.get_app_config('sites') + # Delete the site created as part of the default migration process. + Site.objects.all().delete() + + def test_basic(self): + """ + #15346, #15573 - create_default_site() creates an example site only if + none exist. + """ + with captured_stdout() as stdout: + create_default_site(self.app_config) + self.assertEqual(Site.objects.count(), 1) + self.assertIn("Creating example.com", stdout.getvalue()) + + with captured_stdout() as stdout: + create_default_site(self.app_config) + self.assertEqual(Site.objects.count(), 1) + self.assertEqual("", stdout.getvalue()) + + @override_settings(DATABASE_ROUTERS=[JustOtherRouter()]) + def test_multi_db(self): + """ + #16353, #16828 - The default site creation should respect db routing. + """ + create_default_site(self.app_config, using='default', verbosity=0) + create_default_site(self.app_config, using='other', verbosity=0) + self.assertFalse(Site.objects.using('default').exists()) + self.assertTrue(Site.objects.using('other').exists()) + + def test_save_another(self): + """ + #17415 - Another site can be created right after the default one. + + On some backends the sequence needs to be reset after saving with an + explicit ID. Test that there isn't a sequence collisions by saving + another site. This test is only meaningful with databases that use + sequences for automatic primary keys such as PostgreSQL and Oracle. + """ + create_default_site(self.app_config, verbosity=0) + Site(domain='example2.com', name='example2.com').save() + + def test_signal(self): + """ + #23641 - Sending the ``post_migrate`` signal triggers creation of the + default site. + """ + post_migrate.send(sender=self.app_config, app_config=self.app_config, verbosity=0) + self.assertTrue(Site.objects.exists()) + + @override_settings(SITE_ID=35696) + def test_custom_site_id(self): + """ + #23945 - The configured ``SITE_ID`` should be respected. + """ + create_default_site(self.app_config, verbosity=0) + self.assertEqual(Site.objects.get().pk, 35696) + + +class MiddlewareTest(TestCase): + + def test_request(self): + """ Makes sure that the request has correct `site` attribute. """ + middleware = CurrentSiteMiddleware() + request = HttpRequest() + middleware.process_request(request) + self.assertEqual(request.site.id, settings.SITE_ID) -- cgit v1.3