feat: Error if products in basket are already purchased

ecommerce/extensions/iap/api/v1/tests/test_utils.py

@@ -0,0 +1,39 @@
+import mock
+
+from ecommerce.courses.tests.factories import CourseFactory
+from ecommerce.extensions.iap.api.v1.utils import products_in_basket_already_purchased
+from ecommerce.extensions.order.utils import UserAlreadyPlacedOrder
+from ecommerce.extensions.test.factories import create_basket, create_order
+from ecommerce.tests.testcases import TestCase
+
+
+class TestProductsInBasketPurchased(TestCase):
+    """ Tests for products_in_basket_already_purchased method. """
+
+    def setUp(self):
+        super(TestProductsInBasketPurchased, self).setUp()
+        self.user = self.create_user()
+        self.client.login(username=self.user.username, password=self.password)
+
+        self.course = CourseFactory(partner=self.partner)
+        product = self.course.create_or_update_seat('verified', False, 50)
+        self.basket = create_basket(
+            owner=self.user, site=self.site, price='50.0', product_class=product.product_class
+        )
+        create_order(site=self.site, user=self.user, basket=self.basket)
+
+    def test_already_purchased(self):
+        """
+        Test products in basket already purchased by user
+        """
+        with mock.patch.object(UserAlreadyPlacedOrder, 'user_already_placed_order', return_value=True):
+            return_value = products_in_basket_already_purchased(self.user, self.basket, self.site)
+            self.assertTrue(return_value)
+
+    def test_not_purchased_yet(self):
+        """
+        Test products in basket not yet purchased by user
+        """
+        with mock.patch.object(UserAlreadyPlacedOrder, 'user_already_placed_order', return_value=False):
+            return_value = products_in_basket_already_purchased(self.user, self.basket, self.site)
+            self.assertFalse(return_value)

ecommerce/extensions/iap/api/v1/utils.py

@@ -0,0 +1,19 @@
+
+
+from oscar.core.loading import get_model
+
+from ecommerce.extensions.order.utils import UserAlreadyPlacedOrder
+
+Product = get_model('catalogue', 'Product')
+
+
+def products_in_basket_already_purchased(user, basket, site):
+    """
+    Check if products in a basket are already purchased by a user.
+    """
+    products = Product.objects.filter(line__order__basket=basket)
+    for product in products:
+        if not product.is_enrollment_code_product and \
+                UserAlreadyPlacedOrder.user_already_placed_order(user=user, product=product, site=site):
+            return True
+    return False

ecommerce/extensions/iap/api/v1/views.py

@@ -56,6 +56,7 @@
 )
 from ecommerce.extensions.iap.api.v1.exceptions import RefundCompletionException
 from ecommerce.extensions.iap.api.v1.serializers import MobileOrderSerializer
+from ecommerce.extensions.iap.api.v1.utils import products_in_basket_already_purchased
 from ecommerce.extensions.iap.models import IAPProcessorConfiguration
 from ecommerce.extensions.iap.processors.android_iap import AndroidIAP
 from ecommerce.extensions.iap.processors.ios_iap import IOSIAP
@@ -166,6 +167,9 @@ def _get_basket(self, request, basket_id):
         Applicator().apply(basket, basket.owner, self.request)
         basket_add_organization_attribute(basket, self.request.GET)
 
+        if products_in_basket_already_purchased(request.user, basket, request.site):
+            raise AlreadyPlacedOrderException
+
         return basket
 
     # Disable atomicity for the view. Otherwise, we'd be unable to commit to the database
@@ -191,6 +195,8 @@ def post(self, request):
         except ObjectDoesNotExist:
             logger.exception(LOGGER_BASKET_NOT_FOUND, basket_id)
             return JsonResponse({'error': ERROR_BASKET_NOT_FOUND.format(basket_id)}, status=400)
+        except AlreadyPlacedOrderException:
+            return JsonResponse({'error': _(ERROR_ALREADY_PURCHASED)}, status=406)
         except:  # pylint: disable=bare-except
             error_message = ERROR_WHILE_OBTAINING_BASKET_FOR_USER.format(request.user.email)
             logger.exception(error_message)
@@ -228,6 +234,14 @@ class MobileCheckoutView(APIView):
     permission_classes = (IsAuthenticated,)
 
     def post(self, request):
+        basket_id = request.data.get('basket_id')
+        try:
+            basket = request.user.baskets.get(id=basket_id)
+        except ObjectDoesNotExist:
+            return JsonResponse({'error': ERROR_BASKET_NOT_FOUND.format(basket_id)}, status=400)
+        if products_in_basket_already_purchased(request.user, basket, request.site):
+            return JsonResponse({'error': _(ERROR_ALREADY_PURCHASED)}, status=406)
+
         response = CheckoutView.as_view()(request._request)  # pylint: disable=W0212
         if response.status_code != 200:
             return JsonResponse({'error': response.content.decode()}, status=response.status_code)