requested changes for quality and performance
This commit is contained in:
@@ -72,13 +72,13 @@ class MobileAPITestCase(ModuleStoreTestCase, APITestCase):
|
||||
self.login()
|
||||
self.enroll(course_id)
|
||||
|
||||
def api_response(self, reverse_args=None, expected_response_code=200, qargs={}, **kwargs):
|
||||
def api_response(self, reverse_args=None, expected_response_code=200, data=None, **kwargs):
|
||||
"""
|
||||
Helper method for calling endpoint, verifying and returning response.
|
||||
If expected_response_code is None, doesn't verify the response' status_code.
|
||||
"""
|
||||
url = self.reverse_url(reverse_args, **kwargs)
|
||||
response = self.url_method(url, qargs=qargs, **kwargs)
|
||||
response = self.url_method(url, data=data, **kwargs)
|
||||
if expected_response_code is not None:
|
||||
self.assertEqual(response.status_code, expected_response_code)
|
||||
return response
|
||||
@@ -92,9 +92,9 @@ class MobileAPITestCase(ModuleStoreTestCase, APITestCase):
|
||||
reverse_args.update({'username': kwargs.get('username', self.user.username)})
|
||||
return reverse(self.REVERSE_INFO['name'], kwargs=reverse_args)
|
||||
|
||||
def url_method(self, url, qargs={}, **kwargs): # pylint: disable=unused-argument
|
||||
def url_method(self, url, data=None, **kwargs): # pylint: disable=unused-argument
|
||||
"""Base implementation that returns response from the GET method of the URL."""
|
||||
return self.client.get(url, qargs)
|
||||
return self.client.get(url, data=data)
|
||||
|
||||
|
||||
class MobileAuthTestMixin(object):
|
||||
|
||||
@@ -292,16 +292,11 @@ class TestUserEnrollmentApi(UrlResetMixin, MobileAPITestCase, MobileAuthUserTest
|
||||
for course in courses:
|
||||
self.enroll(course.id)
|
||||
|
||||
response = self.api_response(qargs={'org':'edX'})
|
||||
response = self.api_response(data={'org':'edX'})
|
||||
|
||||
# Verify only edX courses are returned
|
||||
self.assertEqual(len(response.data), 3)
|
||||
for course_index in range(3):
|
||||
result = response.data[course_index]['course']
|
||||
self.assertEqual(result['org'], 'edX')
|
||||
|
||||
# Verify most recently enrolled course is staff only but still returned
|
||||
self.assertFalse(response.data[0]['course']['courseware_access']['has_access'])
|
||||
for entry in response.data:
|
||||
self.assertEqual(entry['course']['org'], 'edX')
|
||||
|
||||
|
||||
@attr(shard=2)
|
||||
|
||||
@@ -271,7 +271,7 @@ class UserCourseEnrollmentsList(generics.ListAPIView):
|
||||
pagination_class = None
|
||||
|
||||
def is_org(self, check_org, course_org):
|
||||
return check_org == None or (check_org.lower() == course_org.lower())
|
||||
return check_org is None or (check_org.lower() == course_org.lower())
|
||||
|
||||
def get_queryset(self):
|
||||
enrollments = self.queryset.filter(
|
||||
|
||||
Reference in New Issue
Block a user