root/trunk/common/tests.py

# vim: encoding=utf-8 :

from urlparse import urlparse

from lxml import etree

from django.contrib.sites.models import Site
from django.conf import settings
from django.test import TestCase

from blog.models import Entry
from common.models import FlatPage

CURRENT_SITE = Site.objects.get_current()


class SimpleTest(TestCase):
    """
    common's test.
    """
    def setUp(self):
        settings.SUB_CONTENTS = ()
        settings.EXTRA_CONTENTS = ()

    def test_sitemap(self):
        """
        Test sitemap.
        """
        #   send request.
        response = self.client.get('/sitemap.xml')
        #   check status code.
        self.assertEqual(response.status_code, 200)
        #   check template name(s).
        self.assertTemplateUsed(response, 'sitemap.xml')
        #   check content type.
        self.assertTrue(response.has_header('Content-type'))
        self.assertEqual(response['Content-type'], u'text/xml; charset=utf-8')
        #   check objects.
        self.assertTrue(response.context.has_key('entries'))
        self.assertTrue(response.context.has_key('flatpages'))
        entries = response.context['entries']
        flatpages = response.context['flatpages']
        checked_entries = Entry.objects.filter(is_published__exact=True)\
                                       .filter(is_active__exact=True)\
                                       .order_by('-updated_at')
        checked_flatpages = FlatPage.objects.filter(is_published__exact=True)\
                                            .filter(is_active__exact=True)\
                                            .order_by('-updated_at')
        self.assertEqual([e.id for e in entries], [c.id for c in checked_entries])
        self.assertEqual([e.id for e in flatpages], [c.id for c in checked_flatpages])
        #   check content.
        namespaces = {'n':'http://www.google.com/schemas/sitemap/0.84'}
        xml = etree.fromstring(response.content, parser=etree.XMLParser())
        checked_length = checked_entries.count() + checked_flatpages.count() + 1
        self.assertEqual(len(xml.xpath('/n:urlset/n:url', namespaces=namespaces)),
                         checked_length)


    def test_flatpages(self):
        """
        Test flatpages.
        """
        #   send request (not exist).
        response = self.client.get('/notexist/')
        #   check status code.
        self.assertEqual(response.status_code, 404)
        #   send request (exist).
        response = self.client.get('/about/')
        #   check status code.
        self.assertEqual(response.status_code, 200)
        #   check template name(s).
        for template_name in (u'flatpages/default.html', u'base.html'):
            self.assertTemplateUsed(response, template_name)
        checked = FlatPage.objects.filter(url__exact='/about/')\
                                  .filter(is_published__exact=True)\
                                  .get(is_active__exact=True)
        #   parse conetnt.
        html = etree.fromstring(response.content, parser=etree.HTMLParser())
        #   check title.
        self.assertEqual(html.xpath('//head/title/text()')[0],
                         u'%s - %s' % (checked.title, CURRENT_SITE.name))

        #   check context
        context = response.context[-1]
        self.assertTrue(context.has_key('obj'))
        self.assertEqual(context['obj'], checked)
        #   send request (not published)
        response = self.client.get('/notpublished/')
        #   check status code.
        self.assertEqual(response.status_code, 404)
        #   log in
        self.assertTrue(self.client.login(username='tester', password='tester'))
        #   send request (not published)
        response = self.client.get('/notpublished/')
        #   check status code.
        self.assertEqual(response.status_code, 200)
        #   check template name(s).
        for template_name in (u'flatpages/default.html', u'base.html'):
            self.assertTemplateUsed(response, template_name)
        checked = FlatPage.objects.filter(url__exact='/notpublished/')\
                                  .filter(is_published__exact=False)\
                                  .get(is_active__exact=True)
        #   parse conetnt.
        html = etree.fromstring(response.content, parser=etree.HTMLParser())
        #   check title.
        self.assertEqual(html.xpath('//head/title/text()')[0],
                         u'%s - %s' % (checked.title, CURRENT_SITE.name))
        #   check context
        context = response.context[-1]
        self.assertTrue(context.has_key('obj'))
        self.assertEqual(context['obj'], checked)