/tests/regressiontests/sites_framework/tests.py

https://code.google.com/p/mango-py/ · Python · 34 lines · 27 code · 7 blank · 0 comment · 0 complexity · 54da205fd26e49acb0bf2ad515da19e8 MD5 · raw file

  1. from django.conf import settings
  2. from django.contrib.sites.models import Site
  3. from django.test import TestCase
  4. from models import SyndicatedArticle, ExclusiveArticle, CustomArticle, InvalidArticle, ConfusedArticle
  5. class SitesFrameworkTestCase(TestCase):
  6. def setUp(self):
  7. Site.objects.get_or_create(id=settings.SITE_ID, domain="example.com", name="example.com")
  8. Site.objects.create(id=settings.SITE_ID+1, domain="example2.com", name="example2.com")
  9. def test_site_fk(self):
  10. article = ExclusiveArticle.objects.create(title="Breaking News!", site_id=settings.SITE_ID)
  11. self.assertEqual(ExclusiveArticle.on_site.all().get(), article)
  12. def test_sites_m2m(self):
  13. article = SyndicatedArticle.objects.create(title="Fresh News!")
  14. article.sites.add(Site.objects.get(id=settings.SITE_ID))
  15. article.sites.add(Site.objects.get(id=settings.SITE_ID+1))
  16. article2 = SyndicatedArticle.objects.create(title="More News!")
  17. article2.sites.add(Site.objects.get(id=settings.SITE_ID+1))
  18. self.assertEqual(SyndicatedArticle.on_site.all().get(), article)
  19. def test_custom_named_field(self):
  20. article = CustomArticle.objects.create(title="Tantalizing News!", places_this_article_should_appear_id=settings.SITE_ID)
  21. self.assertEqual(CustomArticle.on_site.all().get(), article)
  22. def test_invalid_name(self):
  23. article = InvalidArticle.objects.create(title="Bad News!", site_id=settings.SITE_ID)
  24. self.assertRaises(ValueError, InvalidArticle.on_site.all)
  25. def test_invalid_field_type(self):
  26. article = ConfusedArticle.objects.create(title="More Bad News!", site=settings.SITE_ID)
  27. self.assertRaises(TypeError, ConfusedArticle.on_site.all)