diff --git a/common/djangoapps/mitxmako/tests.py b/common/djangoapps/mitxmako/tests.py new file mode 100644 index 0000000000..a1a4fa4582 --- /dev/null +++ b/common/djangoapps/mitxmako/tests.py @@ -0,0 +1,26 @@ +from django.test import TestCase +from django.test.utils import override_settings +from django.core.urlresolvers import reverse +from django.conf import settings +from mitxmako.shortcuts import marketing_link +from mock import patch + + +class ShortcutsTests(TestCase): + """ + Test the mitxmako shortcuts file + """ + + @override_settings(MKTG_URLS={'ROOT': 'dummy-root', 'ABOUT': '/about-us'}) + @override_settings(MKTG_URL_LINK_MAP={'ABOUT': 'about_edx'}) + def test_marketing_link(self): + # test marketing site on + with patch.dict('django.conf.settings.MITX_FEATURES', {'ENABLE_MKTG_SITE': True}): + expected_link = 'dummy-root/about-us' + link = marketing_link('ABOUT') + self.assertEquals(link, expected_link) + # test marketing site off + with patch.dict('django.conf.settings.MITX_FEATURES', {'ENABLE_MKTG_SITE': False}): + expected_link = reverse('about_edx') + link = marketing_link('ABOUT') + self.assertEquals(link, expected_link)