63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
""" Tests for setting and displaying the site status message. """
|
|
from __future__ import absolute_import
|
|
|
|
import unittest
|
|
|
|
import ddt
|
|
from django.conf import settings
|
|
from django.core.cache import cache
|
|
from django.test import TestCase
|
|
from opaque_keys.edx.locations import CourseLocator
|
|
|
|
# Status is not in CMS' INSTALLED_APPS so these imports will error during test collection
|
|
if settings.ROOT_URLCONF == 'lms.urls':
|
|
from .models import CourseMessage, GlobalStatusMessage
|
|
from .status import get_site_status_msg
|
|
|
|
|
|
@ddt.ddt
|
|
class TestStatus(TestCase):
|
|
"""Test that the get_site_status_msg function does the right thing"""
|
|
|
|
def setUp(self):
|
|
super(TestStatus, self).setUp()
|
|
# Clear the cache between test runs.
|
|
cache.clear()
|
|
self.course_key = CourseLocator(org='TestOrg', course='TestCourse', run='TestRun')
|
|
|
|
@unittest.skipUnless(settings.ROOT_URLCONF == 'lms.urls', 'Test only valid in lms')
|
|
@ddt.data(
|
|
("Test global message", "Test course message"),
|
|
(u" Ŧɇsŧ sŧȺŧᵾs", u"Ṫëṡẗ ċöüṛṡë ṡẗäẗüṡ "),
|
|
(u"", u"Ṫëṡẗ ċöüṛṡë ṡẗäẗüṡ "),
|
|
(u" Ŧɇsŧ sŧȺŧᵾs", u""),
|
|
)
|
|
@ddt.unpack
|
|
def test_get_site_status_msg(self, test_global_message, test_course_message):
|
|
"""Test status messages in a variety of situations."""
|
|
|
|
# When we don't have any data set.
|
|
self.assertEqual(get_site_status_msg(None), None)
|
|
self.assertEqual(get_site_status_msg(self.course_key), None)
|
|
|
|
msg = GlobalStatusMessage.objects.create(message=test_global_message, enabled=True)
|
|
msg.save()
|
|
|
|
self.assertEqual(get_site_status_msg(None), test_global_message)
|
|
|
|
course_msg = CourseMessage.objects.create(
|
|
global_message=msg, message=test_course_message, course_key=self.course_key
|
|
)
|
|
course_msg.save()
|
|
self.assertEqual(
|
|
get_site_status_msg(self.course_key),
|
|
u"{} <br /> {}".format(test_global_message, test_course_message)
|
|
)
|
|
|
|
msg = GlobalStatusMessage.objects.create(message="", enabled=False)
|
|
msg.save()
|
|
|
|
self.assertEqual(get_site_status_msg(None), None)
|
|
self.assertEqual(get_site_status_msg(self.course_key), None)
|