class TestAddTemplate(unittest.TestCase):
def setUp(self):
self.templater = Templater()
def test_add_new_template(self):
template = {'test': {'a': 'z'}}
self.templater.add_templates(template)
self.assertEqual(template, self.templater.get_templates())
def test_add_multiple_templates(self):
template1, template2 = {
'test': {
'a': 'z'
}
}, {
'another': {
'123': '456'
}
}
end_template = {'test': {'a': 'z'}, 'another': {'123': '456'}}
self.templater.add_templates(template1, template2)
self.assertEqual(end_template, self.templater.get_templates())
def test_insert_partial_template(self):
initial_template = {'test': {'a': 'z'}, 'another': {'123': '456'}}
new_template = {'test': {'b': 'y'}}
end_template = {
'test': {
'a': 'z',
'b': 'y'
},
'another': {
'123': '456'
}
}
self.templater.add_templates(initial_template)
self.templater.add_templates(new_template)
self.assertEqual(end_template, self.templater.get_templates())
class TestBuildPages(unittest.TestCase):
md_parser = MarkdownParser()
def setUp(self):
self.templater = Templater()
def test_assemble_page(self):
header = '<header>This is the header throughout the site</header>'
body = self.md_parser.parse(
dedent('''\
# Some stuff
yeah wow this is markdown
*Thing*'''))
page_structure = dedent('''\
{{ modules.header }}
{{ page }}''')
self.templater.add_templates({
'structures': {
'page': page_structure
},
'modules': {
'header': header
},
'page': {
'_html': body
}
})
assembled_page = self.templater.fill_structure('page')
expected_page = dedent('''\
<header>This is the header throughout the site</header>
<h1>Some stuff</h1>
<p>yeah wow this is markdown</p>
<p><em>Thing</em></p>''')
self.assertEqual(expected_page, assembled_page)
def test_assemble_nested_page(self):
head = dedent('''\
<head>
<title>{{ page.title }}</title
<meta name="description" content="{{page.description}}">
</head>''')
page = dedent('''\
---
title: This is the inserted page title!
description: Some stuff about the page.
---
# {{ page.title }}
Contents of the post
*Something else*''')
footer = dedent('''\
<footer>
<strong>Copyright 2021 Some Guys</strong>
</footer>''')
page_structure = dedent('''\
<!DOCTYPE html>
<html lang="en">
{{ modules.head }}
<body>
{{ page }}
{{modules.footer}}
</body>
</html>''')
# Get front matter and markdown
page_template, markdown = split_fm_md.split_page(page)
# Parse markdown
parsed_markdown = self.md_parser.parse(markdown)
# Put front matter and markdown into templates
page_template['_html'] = parsed_markdown
self.templater.add_templates({
'structures': {
'page': page_structure
},
'modules': {
'head': head,
'footer': footer
},
'page': page_template
})
assembled_page = self.templater.fill_structure('page')
# Does not maintain the indents above
expected_page = dedent('''\
<!DOCTYPE html>
<html lang="en">
<head>
<title>This is the inserted page title!</title
<meta name="description" content="Some stuff about the page.">
</head>
<body>
<h1>This is the inserted page title!</h1>
<p>Contents of the post</p>
<p><em>Something else</em></p>
<footer>
<strong>Copyright 2021 Some Guys</strong>
</footer>
</body>
</html>''')
self.assertEqual(expected_page, assembled_page)
