Commit 6e543c14 by Jason Bau

pep8 + pylint

parent 4c664240
...@@ -8,6 +8,7 @@ navigation. We want to do this in the middleware to ...@@ -8,6 +8,7 @@ navigation. We want to do this in the middleware to
from django.conf import settings from django.conf import settings
import shoppingcart import shoppingcart
class UserHasCartMiddleware(object): class UserHasCartMiddleware(object):
""" """
Detects whether request.user has a cart and sets it as part of the request Detects whether request.user has a cart and sets it as part of the request
...@@ -23,9 +24,11 @@ class UserHasCartMiddleware(object): ...@@ -23,9 +24,11 @@ class UserHasCartMiddleware(object):
be displayed. Anonymous users don't. be displayed. Anonymous users don't.
""" """
request.display_shopping_cart = False request.display_shopping_cart = False
if (request.user.is_authenticated() and # user exists if (
settings.MITX_FEATURES.get('ENABLE_PAID_COURSE_REGISTRATION') and # settings are set request.user.is_authenticated() and # user exists
settings.MITX_FEATURES.get('ENABLE_SHOPPING_CART') and settings.MITX_FEATURES.get('ENABLE_PAID_COURSE_REGISTRATION') and # settings is set
shoppingcart.models.Order.user_cart_has_items(request.user)): # user's cart is non-empty settings.MITX_FEATURES.get('ENABLE_SHOPPING_CART') and # setting is set
shoppingcart.models.Order.user_cart_has_items(request.user) # user's cart is non-empty
):
request.display_shopping_cart = True request.display_shopping_cart = True
return None return None
\ No newline at end of file
...@@ -14,14 +14,21 @@ from course_modes.models import CourseMode ...@@ -14,14 +14,21 @@ from course_modes.models import CourseMode
from shoppingcart.models import Order, PaidCourseRegistration from shoppingcart.models import Order, PaidCourseRegistration
from shoppingcart.middleware import UserHasCartMiddleware from shoppingcart.middleware import UserHasCartMiddleware
@override_settings(MODULESTORE=TEST_DATA_MONGO_MODULESTORE) @override_settings(MODULESTORE=TEST_DATA_MONGO_MODULESTORE)
class UserCartMiddlewareUnitTest(ModuleStoreTestCase): class UserCartMiddlewareUnitTest(ModuleStoreTestCase):
"""
Unit test for shoppingcart middleware UserHasCartMiddleware
"""
def setUp(self): def setUp(self):
self.user = UserFactory.create() self.user = UserFactory.create()
self.request = Mock() self.request = Mock()
self.mw = UserHasCartMiddleware() self.middleware = UserHasCartMiddleware()
def add_to_cart(self): def add_to_cart(self):
"""
Adds content to self.user's cart
"""
course = CourseFactory.create(org='MITx', number='999', display_name='Robot Super Course') course = CourseFactory.create(org='MITx', number='999', display_name='Robot Super Course')
course_mode = CourseMode(course_id=course.id, course_mode = CourseMode(course_id=course.id,
mode_slug="honor", mode_slug="honor",
...@@ -38,7 +45,7 @@ class UserCartMiddlewareUnitTest(ModuleStoreTestCase): ...@@ -38,7 +45,7 @@ class UserCartMiddlewareUnitTest(ModuleStoreTestCase):
""" """
self.add_to_cart() self.add_to_cart()
self.request.user = self.user self.request.user = self.user
self.mw.process_request(self.request) self.middleware.process_request(self.request)
self.assertFalse(self.request.display_shopping_cart) self.assertFalse(self.request.display_shopping_cart)
@patch.dict(settings.MITX_FEATURES, {'ENABLE_SHOPPING_CART': True, 'ENABLE_PAID_COURSE_REGISTRATION': False}) @patch.dict(settings.MITX_FEATURES, {'ENABLE_SHOPPING_CART': True, 'ENABLE_PAID_COURSE_REGISTRATION': False})
...@@ -48,7 +55,7 @@ class UserCartMiddlewareUnitTest(ModuleStoreTestCase): ...@@ -48,7 +55,7 @@ class UserCartMiddlewareUnitTest(ModuleStoreTestCase):
""" """
self.add_to_cart() self.add_to_cart()
self.request.user = self.user self.request.user = self.user
self.mw.process_request(self.request) self.middleware.process_request(self.request)
self.assertFalse(self.request.display_shopping_cart) self.assertFalse(self.request.display_shopping_cart)
@patch.dict(settings.MITX_FEATURES, {'ENABLE_SHOPPING_CART': True, 'ENABLE_PAID_COURSE_REGISTRATION': True}) @patch.dict(settings.MITX_FEATURES, {'ENABLE_SHOPPING_CART': True, 'ENABLE_PAID_COURSE_REGISTRATION': True})
...@@ -57,7 +64,7 @@ class UserCartMiddlewareUnitTest(ModuleStoreTestCase): ...@@ -57,7 +64,7 @@ class UserCartMiddlewareUnitTest(ModuleStoreTestCase):
Tests when request.user is anonymous Tests when request.user is anonymous
""" """
self.request.user = AnonymousUser() self.request.user = AnonymousUser()
self.mw.process_request(self.request) self.middleware.process_request(self.request)
self.assertFalse(self.request.display_shopping_cart) self.assertFalse(self.request.display_shopping_cart)
@patch.dict(settings.MITX_FEATURES, {'ENABLE_SHOPPING_CART': True, 'ENABLE_PAID_COURSE_REGISTRATION': True}) @patch.dict(settings.MITX_FEATURES, {'ENABLE_SHOPPING_CART': True, 'ENABLE_PAID_COURSE_REGISTRATION': True})
...@@ -66,7 +73,7 @@ class UserCartMiddlewareUnitTest(ModuleStoreTestCase): ...@@ -66,7 +73,7 @@ class UserCartMiddlewareUnitTest(ModuleStoreTestCase):
Tests when request.user doesn't have a cart with items Tests when request.user doesn't have a cart with items
""" """
self.request.user = self.user self.request.user = self.user
self.mw.process_request(self.request) self.middleware.process_request(self.request)
self.assertFalse(self.request.display_shopping_cart) self.assertFalse(self.request.display_shopping_cart)
@patch.dict(settings.MITX_FEATURES, {'ENABLE_SHOPPING_CART': True, 'ENABLE_PAID_COURSE_REGISTRATION': True}) @patch.dict(settings.MITX_FEATURES, {'ENABLE_SHOPPING_CART': True, 'ENABLE_PAID_COURSE_REGISTRATION': True})
...@@ -76,5 +83,5 @@ class UserCartMiddlewareUnitTest(ModuleStoreTestCase): ...@@ -76,5 +83,5 @@ class UserCartMiddlewareUnitTest(ModuleStoreTestCase):
""" """
self.add_to_cart() self.add_to_cart()
self.request.user = self.user self.request.user = self.user
self.mw.process_request(self.request) self.middleware.process_request(self.request)
self.assertTrue(self.request.display_shopping_cart) self.assertTrue(self.request.display_shopping_cart)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment