Tests for contents.py

This commit is contained in:
Alexis Metaireau 2011-01-13 00:32:37 +01:00
commit c13c707a62
3 changed files with 58 additions and 58 deletions

View file

@ -1,4 +1,5 @@
from pelican.utils import slugify, truncate_html_words
from pelican.settings import DEFAULT_CONFIG
class Page(object):
@ -22,16 +23,17 @@ class Page(object):
if 'AUTHOR' in settings:
self.author = settings['AUTHOR']
default_lang = settings.get('DEFAULT_LANG').lower()
default_lang = settings.get('DEFAULT_LANG',
DEFAULT_CONFIG['DEFAULT_LANG']).lower()
if not hasattr(self, 'lang'):
self.lang = default_lang
self.in_default_lang = (self.lang == default_lang)
if not hasattr(self, 'slug'):
if not hasattr(self, 'slug') and hasattr(self, 'title'):
self.slug = slugify(self.title)
if not hasattr(self, 'save_as'):
if not hasattr(self, 'save_as') and hasattr(self, 'slug'):
if self.in_default_lang:
self.save_as = '%s.html' % self.slug
clean_url = '%s/' % self.slug
@ -41,7 +43,7 @@ class Page(object):
if settings.get('CLEAN_URLS', False):
self.url = clean_url
else:
elif hasattr(self, 'save_as'):
self.url = self.save_as
if filename:

View file

@ -1,54 +0,0 @@
from unittest2 import TestCase
from pelican.contents import Page
from pelican.settings import DEFAULT_CONFIG
class TestPage(TestCase):
def test_use_args(self):
# creating a page with arguments passed to the connstructor should use
# them to initialise object's attributes
metadata = {'foo': 'bar', 'foobar': 'baz'}
page = Page('content', metadatas=metadata)
for key, value in metadata.items():
self.assertTrue(hasattr(page, key))
self.assertEqual(value, getattr(page, key))
self.assertEqual(page.content, "content")
def test_mandatory_properties(self):
# if the title is not set, must throw an exception
page = Page('content')
with self.assertRaises(NameError) as cm:
page.check_properties()
page = Page('content', metadatas={'title': 'foobar'})
page.check_properties()
def test_slug(self):
# if a title is given, it should be used to generate the slug
page = Page('content', {'title': 'foobar is foo'})
self.assertEqual(page.slug, 'foobar-is-foo')
def test_defaultlang(self):
# if no lang is given, default to the default one
page = Page('content')
self.assertEqual(page.lang, DEFAULT_CONFIG['DEFAULT_LANG'])
# it is possible to specify the lang in the metadata infos
page = Page('content', {'lang': 'fr'})
self.assertEqual(page.lang, 'fr')
def test_save_as(self):
# if a lang is not the default lang, save_as should be set accordingly
page = Page('content', {'title': 'foobar', 'lang': 'fr'}) #default lang is en
self.assertEqual(page.save_as, "foobar-fr.html")
# otherwise, if a title is defined, save_as should be set
page = Page('content', {'title': 'foobar'})
page.save_as = 'foobar.html'
# if no title is given, there is no save_as
page = Page('content')
self.assertFalse(hasattr(page, 'save_as'))