151bd13666
* Generate common/djangoapps import shims for LMS * Generate common/djangoapps import shims for Studio * Stop appending project root to sys.path * Stop appending common/djangoapps to sys.path * Import from common.djangoapps.course_action_state instead of course_action_state * Import from common.djangoapps.course_modes instead of course_modes * Import from common.djangoapps.database_fixups instead of database_fixups * Import from common.djangoapps.edxmako instead of edxmako * Import from common.djangoapps.entitlements instead of entitlements * Import from common.djangoapps.pipline_mako instead of pipeline_mako * Import from common.djangoapps.static_replace instead of static_replace * Import from common.djangoapps.student instead of student * Import from common.djangoapps.terrain instead of terrain * Import from common.djangoapps.third_party_auth instead of third_party_auth * Import from common.djangoapps.track instead of track * Import from common.djangoapps.util instead of util * Import from common.djangoapps.xblock_django instead of xblock_django * Add empty common/djangoapps/__init__.py to fix pytest collection * Fix pylint formatting violations * Exclude import_shims/ directory tree from linting
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
""" Tests for models. """
|
|
|
|
|
|
import ddt
|
|
from django.test import TestCase
|
|
|
|
from common.djangoapps.course_modes.models import CourseMode
|
|
|
|
from ..models import Course
|
|
|
|
|
|
@ddt.ddt
|
|
class CourseTests(TestCase):
|
|
""" Tests for Course model. """
|
|
|
|
def setUp(self):
|
|
super(CourseTests, self).setUp()
|
|
self.course = Course('a/b/c', [])
|
|
|
|
@ddt.unpack
|
|
@ddt.data(
|
|
('credit', 'Credit'),
|
|
('professional', 'Professional Education'),
|
|
('no-id-professional', 'Professional Education'),
|
|
('verified', 'Verified Certificate'),
|
|
('honor', 'Honor Certificate'),
|
|
('audit', 'Audit'),
|
|
)
|
|
def test_get_mode_display_name(self, slug, expected_display_name):
|
|
""" Verify the method properly maps mode slugs to display names. """
|
|
mode = CourseMode(mode_slug=slug)
|
|
self.assertEqual(self.course.get_mode_display_name(mode), expected_display_name)
|
|
|
|
def test_get_mode_display_name_unknown_slug(self):
|
|
""" Verify the method returns the slug if it has no known mapping. """
|
|
mode = CourseMode(mode_slug='Blah!')
|
|
self.assertEqual(self.course.get_mode_display_name(mode), mode.mode_slug)
|