def test_get_template_no_custom(): archivist = S3archivist(bucket=testbucket, siteconfig=siteconfig, jinja=mock.Mock()) template = scribe.get_template(archivist, {}) tlist = ["test_template.j2", mock.ANY] archivist.jinja.select_template.assert_called_with(tlist) assert template
def test_get_template_default_is_list(): archivist = S3archivist(bucket=testbucket, siteconfig={"default_template": ["t1.j2", "t2.j2"]}, jinja=mock.Mock()) template = scribe.get_template(archivist, {"template": "custom_template.j2"}) tlist = ["custom_template.j2", "t1.j2", "t2.j2", mock.ANY] archivist.jinja.select_template.assert_called_with(tlist) assert template