56e9625a8f
The Commerce API now sets the correct display name for known course modes, but continues to default to the mode slug for unknown course modes.
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
""" Tests for models. """
|
|
import ddt
|
|
from django.test import TestCase
|
|
|
|
from commerce.api.v1.models import Course
|
|
from course_modes.models import CourseMode
|
|
|
|
|
|
@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)
|