|
8 | 8 | from rest_framework import status |
9 | 9 | from rest_framework.test import APIClient |
10 | 10 |
|
11 | | -from common.djangoapps.student.tests.factories import InstructorFactory, UserFactory |
| 11 | +from common.djangoapps.student.tests.factories import CourseEnrollmentFactory, InstructorFactory, UserFactory |
12 | 12 | from lms.djangoapps.courseware.models import StudentModule |
13 | 13 | from lms.djangoapps.instructor_task.models import InstructorTask |
14 | 14 | from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase |
@@ -46,12 +46,30 @@ def test_get_learner_by_username(self): |
46 | 46 | 'student_id': self.student.id, |
47 | 47 | }) |
48 | 48 |
|
49 | | - self.assertEqual(response.status_code, status.HTTP_200_OK) # noqa: PT009 |
| 49 | + assert response.status_code == status.HTTP_200_OK |
50 | 50 | data = response.json() |
51 | | - self.assertEqual(data['username'], 'john_harvard') # noqa: PT009 |
52 | | - self. assertEqual( data[ 'email'], '[email protected]') # noqa: PT009 |
53 | | - self.assertEqual(data['full_name'], 'John Harvard') # noqa: PT009 |
54 | | - self.assertEqual(data['progress_url'], expected_progress_url) # noqa: PT009 |
| 51 | + assert data['username'] == 'john_harvard' |
| 52 | + assert data[ 'email'] == '[email protected]' |
| 53 | + assert data['full_name'] == 'John Harvard' |
| 54 | + assert data['progress_url'] == expected_progress_url |
| 55 | + assert not data['is_enrolled'] |
| 56 | + |
| 57 | + def test_get_learner_by_username_enrolled(self): |
| 58 | + """Test that is_enrolled is true for users enrolled in the course""" |
| 59 | + CourseEnrollmentFactory( |
| 60 | + is_active=True, |
| 61 | + course_id=self.course.id, |
| 62 | + user=self.student |
| 63 | + ) |
| 64 | + url = reverse('instructor_api_v2:learner_detail', kwargs={ |
| 65 | + 'course_id': str(self.course.id), |
| 66 | + 'email_or_username': self.student.username |
| 67 | + }) |
| 68 | + response = self.client.get(url) |
| 69 | + |
| 70 | + assert response.status_code == status.HTTP_200_OK |
| 71 | + data = response.json() |
| 72 | + assert data['is_enrolled'] |
55 | 73 |
|
56 | 74 | def test_get_learner_by_email(self): |
57 | 75 | """Test retrieving learner info by email""" |
|
0 commit comments