from tests import BaseTestCase from md_pdf.build.context import get_context, EXTRA_CONTEXT from md_pdf import consts from datetime import datetime from md_pdf import __version__ import os class ExtraContextTestCase(BaseTestCase): def test_directories(self): self.assertEqual(EXTRA_CONTEXT['templates_dir'], consts.TEMPLATES_DIR) self.assertEqual(EXTRA_CONTEXT['static_dir'], consts.STATIC_DIR) self.assertEqual(EXTRA_CONTEXT['internal_templates_dir'], consts.INTERNAL_TEMPLATES_DIR) self.assertEqual(EXTRA_CONTEXT['internal_static_dir'], consts.INTERNAL_STATIC_DIR) def test_datetimes(self): now = datetime.now() self.assertEqual(EXTRA_CONTEXT['date'], now.strftime(consts.DATE_FORMAT)) self.assertEqual(EXTRA_CONTEXT['time'], now.strftime(consts.TIME_FORMAT)) self.assertEqual(EXTRA_CONTEXT['datetime'], now.strftime(consts.DATETIME_FORMAT)) def test_version(self): self.assertEqual(EXTRA_CONTEXT['mdp_version'], __version__) class ContextTestCase(BaseTestCase): def test_context_contains_extra(self): context = get_context(self.BASE_VALID_CONFIG, 'test') for key in EXTRA_CONTEXT.keys(): self.assertIn(key, context) def test_context_contains_context(self): config = dict(self.BASE_VALID_CONFIG, **{ 'context': { '1': '2' } }) context = get_context(config, 'test') self.assertEqual(context['1'], '2') self.assertNotIn('context', context) def test_context_contains_config(self): context = get_context(self.BASE_VALID_CONFIG, 'test') for key in self.BASE_VALID_CONFIG.keys(): self.assertIn(key, context) def test_has_output_dir(self): context = get_context(self.BASE_VALID_CONFIG, 'test') self.assertEqual(context['output_dir'], os.path.abspath(self.BASE_VALID_CONFIG['output_dir'])) def test_word_count(self): config = dict(self.BASE_VALID_CONFIG, **{ 'show_word_count': True }) context = get_context(config, 'testy test test') self.assertEqual(context['word_count'], 3) def test_transparent_datetime_for_submission_date(self): for value in [ datetime.now().date(), datetime.now().time(), datetime.now() ]: config = dict(self.BASE_VALID_CONFIG, **{ 'submission_date': value }) context = get_context(config, 'test') self.assertEqual(context['submission_date'], value.strftime(consts.DATE_FORMAT)) def test_date_format(self): config = dict(self.BASE_VALID_CONFIG, **{ 'submission_date': '2017-01-01' }) context = get_context(config, 'test') self.assertEqual(context['submission_date'], '01 January 2017')