def test_list_links(self, client, org_name, status_code, expected_count): org = Organization.objects.get(name=org_name) course = CourseOverviewFactory.create() OrganizationCourse.objects.create(course_id=str(course.id), organization=org) GroupCourseFactory.create_batch(3, group__organization=org, course=course) response = client.get(self.url) assert response.status_code == HTTP_200_OK, response.content results = response.json()['results'] assert len(results) == expected_count
def test_list_courses(self, client, org_name, expected_count): org = Organization.objects.get(name=org_name) private_courses = CourseOverviewFactory.create_batch(2) GroupCourseFactory.create_batch(2, course=private_courses[0], group__organization=org) public_courses = [p.course for p in PublicCourseFactory.create_batch(3)] OrganizationCourseFactory.create_for( org, courses=private_courses + public_courses, ) response = client.get(self.url) results = response.json()['results'] assert response.status_code == HTTP_200_OK, response.content assert len(results) == expected_count