You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
blog/test/test_content.py

105 lines
3.4 KiB
Python

5 years ago
import unittest
import shutil
import os
import makesite
from test import path
class ContentTest(unittest.TestCase):
def setUp(self):
self.blog_path = path.temppath('blog')
self.undated_path = os.path.join(self.blog_path, 'foo.txt')
self.dated_path = os.path.join(self.blog_path, '2018-01-01-foo.txt')
self.normal_post_path = os.path.join(self.blog_path, 'baz.txt')
self.md_post_path = os.path.join(self.blog_path, 'qux.md')
self.no_md_post_path = os.path.join(self.blog_path, 'qux.txt')
os.makedirs(self.blog_path)
with open(self.undated_path, 'w') as f:
f.write('hello world')
with open(self.dated_path, 'w') as f:
f.write('hello world')
with open(self.normal_post_path, 'w') as f:
f.write('<!-- a: 1 -->\n<!-- b: 2 -->\nFoo')
with open(self.md_post_path, 'w') as f:
f.write('*Foo*')
with open(self.no_md_post_path, 'w') as f:
f.write('*Foo*')
def tearDown(self):
shutil.rmtree(self.blog_path)
# Rudimentary mock because unittest.mock is unavailable in Python 2.7.
def mock(self, *args):
self.mock_args = args
def test_content_content(self):
content = makesite.read_content(self.undated_path)
self.assertEqual(content['content'], 'hello world')
def test_content_date(self):
content = makesite.read_content(self.dated_path)
self.assertEqual(content['date'], '2018-01-01')
def test_content_date_missing(self):
content = makesite.read_content(self.undated_path)
self.assertEqual(content['date'], '1970-01-01')
def test_content_slug_dated(self):
content = makesite.read_content(self.dated_path)
self.assertEqual(content['slug'], 'foo')
def test_content_slug_undated(self):
content = makesite.read_content(self.undated_path)
self.assertEqual(content['slug'], 'foo')
def test_content_headers(self):
content = makesite.read_content(self.normal_post_path)
self.assertEqual(content['a'], '1')
self.assertEqual(content['b'], '2')
self.assertEqual(content['content'], 'Foo')
def test_markdown_rendering(self):
content = makesite.read_content(self.md_post_path)
self.assertEqual(content['content'], '<p><em>Foo</em></p>\n')
def test_markdown_import_error(self):
makesite._test = 'ImportError'
original_log = makesite.log
makesite.log = self.mock
self.mock_args = None
content = makesite.read_content(self.md_post_path)
makesite._test = None
makesite.log = original_log
self.assertEqual(content['content'], '*Foo*')
self.assertEqual(self.mock_args,
('WARNING: Cannot render Markdown in {}: {}',
self.md_post_path, 'Error forced by test'))
def test_no_markdown_rendering(self):
content = makesite.read_content(self.no_md_post_path)
self.assertEqual(content['content'], '*Foo*')
def test_no_markdown_import_error(self):
makesite._test = 'ImportError'
original_log = makesite.log
makesite.log = self.mock
self.mock_args = None
content = makesite.read_content(self.no_md_post_path)
makesite._test = None
makesite.log = original_log
self.assertEqual(content['content'], '*Foo*')
self.assertIsNone(self.mock_args)