def test_get_course_enrollment(self, course_modes, enrollment_mode):
self._create_course_modes(course_modes)
# Try to get an enrollment before it exists.
result = data.get_course_enrollment(self.user.username,
six.text_type(self.course.id))
assert result is None
# Create the original enrollment.
enrollment = data.create_course_enrollment(
self.user.username, six.text_type(self.course.id), enrollment_mode,
True)
# Get the enrollment and compare it to the original.
result = data.get_course_enrollment(self.user.username,
six.text_type(self.course.id))
assert self.user.username == result['user']
assert enrollment == result
def test_get_course_enrollment(self, course_modes, enrollment_mode):
self._create_course_modes(course_modes)
# Try to get an enrollment before it exists.
result = data.get_course_enrollment(self.user.username, six.text_type(self.course.id))
self.assertIsNone(result)
# Create the original enrollment.
enrollment = data.create_course_enrollment(
self.user.username,
six.text_type(self.course.id),
enrollment_mode,
True
)
# Get the enrollment and compare it to the original.
result = data.get_course_enrollment(self.user.username, six.text_type(self.course.id))
self.assertEqual(self.user.username, result['user'])
self.assertEqual(enrollment, result)
