90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
from django.conf import settings
|
|
from django.test import TestCase
|
|
import os
|
|
from django.test.utils import override_settings
|
|
from tempfile import NamedTemporaryFile
|
|
|
|
from .status import get_site_status_msg
|
|
|
|
# Get a name where we can put test files
|
|
TMP_FILE = NamedTemporaryFile(delete=False)
|
|
TMP_NAME = TMP_FILE.name
|
|
# Close it--we just want the path.
|
|
TMP_FILE.close()
|
|
|
|
|
|
@override_settings(STATUS_MESSAGE_PATH=TMP_NAME)
|
|
class TestStatus(TestCase):
|
|
"""Test that the get_site_status_msg function does the right thing"""
|
|
|
|
no_file = None
|
|
|
|
invalid_json = """{
|
|
"global" : "Hello, Globe",
|
|
}"""
|
|
|
|
global_only = """{
|
|
"global" : "Hello, Globe"
|
|
}"""
|
|
|
|
toy_only = """{
|
|
"edX/toy/2012_Fall" : "A toy story"
|
|
}"""
|
|
|
|
global_and_toy = """{
|
|
"global" : "Hello, Globe",
|
|
"edX/toy/2012_Fall" : "A toy story"
|
|
}"""
|
|
|
|
# json to use, expected results for course=None (e.g. homepage),
|
|
# for toy course, for full course. Note that get_site_status_msg
|
|
# is supposed to return global message even if course=None. The
|
|
# template just happens to not display it outside the courseware
|
|
# at the moment...
|
|
checks = [
|
|
(no_file, None, None, None),
|
|
(invalid_json, None, None, None),
|
|
(global_only, "Hello, Globe", "Hello, Globe", "Hello, Globe"),
|
|
(toy_only, None, "A toy story", None),
|
|
(global_and_toy, "Hello, Globe", "Hello, Globe<br>A toy story", "Hello, Globe"),
|
|
]
|
|
|
|
def setUp(self):
|
|
"""
|
|
Fake course ids, since we don't have to have full django
|
|
settings (common tests run without the lms settings imported)
|
|
"""
|
|
self.full_id = 'edX/full/2012_Fall'
|
|
self.toy_id = 'edX/toy/2012_Fall'
|
|
|
|
def create_status_file(self, contents):
|
|
"""
|
|
Write contents to settings.STATUS_MESSAGE_PATH.
|
|
"""
|
|
with open(settings.STATUS_MESSAGE_PATH, 'w') as f:
|
|
f.write(contents)
|
|
|
|
def remove_status_file(self):
|
|
"""Delete the status file if it exists"""
|
|
if os.path.exists(settings.STATUS_MESSAGE_PATH):
|
|
os.remove(settings.STATUS_MESSAGE_PATH)
|
|
|
|
def tearDown(self):
|
|
self.remove_status_file()
|
|
|
|
def test_get_site_status_msg(self):
|
|
"""run the tests"""
|
|
for (json_str, exp_none, exp_toy, exp_full) in self.checks:
|
|
|
|
self.remove_status_file()
|
|
if json_str:
|
|
self.create_status_file(json_str)
|
|
|
|
print "checking results for {0}".format(json_str)
|
|
print "course=None:"
|
|
self.assertEqual(get_site_status_msg(None), exp_none)
|
|
print "course=toy:"
|
|
self.assertEqual(get_site_status_msg(self.toy_id), exp_toy)
|
|
print "course=full:"
|
|
self.assertEqual(get_site_status_msg(self.full_id), exp_full)
|