340 lines
12 KiB
Python
340 lines
12 KiB
Python
import pytz
|
|
|
|
from django.core.exceptions import ValidationError
|
|
|
|
from registrasion import models as rego
|
|
from registrasion.controllers.category import CategoryController
|
|
from cart_controller_helper import TestingCartController
|
|
from registrasion.controllers.product import ProductController
|
|
|
|
from test_cart import RegistrationCartTestCase
|
|
|
|
UTC = pytz.timezone('UTC')
|
|
|
|
|
|
class EnablingConditionTestCases(RegistrationCartTestCase):
|
|
|
|
@classmethod
|
|
def add_product_enabling_condition(cls, mandatory=False):
|
|
''' Adds a product enabling condition: adding PROD_1 to a cart is
|
|
predicated on adding PROD_2 beforehand. '''
|
|
enabling_condition = rego.ProductEnablingCondition.objects.create(
|
|
description="Product condition",
|
|
mandatory=mandatory,
|
|
)
|
|
enabling_condition.save()
|
|
enabling_condition.products.add(cls.PROD_1)
|
|
enabling_condition.enabling_products.add(cls.PROD_2)
|
|
enabling_condition.save()
|
|
|
|
@classmethod
|
|
def add_product_enabling_condition_on_category(cls, mandatory=False):
|
|
''' Adds a product enabling condition that operates on a category:
|
|
adding an item from CAT_1 is predicated on adding PROD_3 beforehand '''
|
|
enabling_condition = rego.ProductEnablingCondition.objects.create(
|
|
description="Product condition",
|
|
mandatory=mandatory,
|
|
)
|
|
enabling_condition.save()
|
|
enabling_condition.categories.add(cls.CAT_1)
|
|
enabling_condition.enabling_products.add(cls.PROD_3)
|
|
enabling_condition.save()
|
|
|
|
def add_category_enabling_condition(cls, mandatory=False):
|
|
''' Adds a category enabling condition: adding PROD_1 to a cart is
|
|
predicated on adding an item from CAT_2 beforehand.'''
|
|
enabling_condition = rego.CategoryEnablingCondition.objects.create(
|
|
description="Category condition",
|
|
mandatory=mandatory,
|
|
enabling_category=cls.CAT_2,
|
|
)
|
|
enabling_condition.save()
|
|
enabling_condition.products.add(cls.PROD_1)
|
|
enabling_condition.save()
|
|
|
|
def test_product_enabling_condition_enables_product(self):
|
|
self.add_product_enabling_condition()
|
|
|
|
# Cannot buy PROD_1 without buying PROD_2
|
|
current_cart = TestingCartController.for_user(self.USER_1)
|
|
with self.assertRaises(ValidationError):
|
|
current_cart.add_to_cart(self.PROD_1, 1)
|
|
|
|
current_cart.add_to_cart(self.PROD_2, 1)
|
|
current_cart.add_to_cart(self.PROD_1, 1)
|
|
|
|
def test_product_enabled_by_product_in_previous_cart(self):
|
|
self.add_product_enabling_condition()
|
|
|
|
current_cart = TestingCartController.for_user(self.USER_1)
|
|
current_cart.add_to_cart(self.PROD_2, 1)
|
|
|
|
current_cart.next_cart()
|
|
|
|
# Create new cart and try to add PROD_1
|
|
current_cart = TestingCartController.for_user(self.USER_1)
|
|
current_cart.add_to_cart(self.PROD_1, 1)
|
|
|
|
def test_product_enabling_condition_enables_category(self):
|
|
self.add_product_enabling_condition_on_category()
|
|
|
|
# Cannot buy PROD_1 without buying item from CAT_2
|
|
current_cart = TestingCartController.for_user(self.USER_1)
|
|
with self.assertRaises(ValidationError):
|
|
current_cart.add_to_cart(self.PROD_1, 1)
|
|
|
|
current_cart.add_to_cart(self.PROD_3, 1)
|
|
current_cart.add_to_cart(self.PROD_1, 1)
|
|
|
|
def test_category_enabling_condition_enables_product(self):
|
|
self.add_category_enabling_condition()
|
|
|
|
# Cannot buy PROD_1 without buying PROD_2
|
|
current_cart = TestingCartController.for_user(self.USER_1)
|
|
with self.assertRaises(ValidationError):
|
|
current_cart.add_to_cart(self.PROD_1, 1)
|
|
|
|
# PROD_3 is in CAT_2
|
|
current_cart.add_to_cart(self.PROD_3, 1)
|
|
current_cart.add_to_cart(self.PROD_1, 1)
|
|
|
|
def test_product_enabled_by_category_in_previous_cart(self):
|
|
self.add_category_enabling_condition()
|
|
|
|
current_cart = TestingCartController.for_user(self.USER_1)
|
|
current_cart.add_to_cart(self.PROD_3, 1)
|
|
|
|
current_cart.next_cart()
|
|
|
|
# Create new cart and try to add PROD_1
|
|
current_cart = TestingCartController.for_user(self.USER_1)
|
|
current_cart.add_to_cart(self.PROD_1, 1)
|
|
|
|
def test_multiple_non_mandatory_conditions(self):
|
|
self.add_product_enabling_condition()
|
|
self.add_category_enabling_condition()
|
|
|
|
# User 1 is testing the product enabling condition
|
|
cart_1 = TestingCartController.for_user(self.USER_1)
|
|
# Cannot add PROD_1 until a condition is met
|
|
with self.assertRaises(ValidationError):
|
|
cart_1.add_to_cart(self.PROD_1, 1)
|
|
cart_1.add_to_cart(self.PROD_2, 1)
|
|
cart_1.add_to_cart(self.PROD_1, 1)
|
|
|
|
# User 2 is testing the category enabling condition
|
|
cart_2 = TestingCartController.for_user(self.USER_2)
|
|
# Cannot add PROD_1 until a condition is met
|
|
with self.assertRaises(ValidationError):
|
|
cart_2.add_to_cart(self.PROD_1, 1)
|
|
cart_2.add_to_cart(self.PROD_3, 1)
|
|
cart_2.add_to_cart(self.PROD_1, 1)
|
|
|
|
def test_multiple_mandatory_conditions(self):
|
|
self.add_product_enabling_condition(mandatory=True)
|
|
self.add_category_enabling_condition(mandatory=True)
|
|
|
|
cart_1 = TestingCartController.for_user(self.USER_1)
|
|
# Cannot add PROD_1 until both conditions are met
|
|
with self.assertRaises(ValidationError):
|
|
cart_1.add_to_cart(self.PROD_1, 1)
|
|
cart_1.add_to_cart(self.PROD_2, 1) # Meets the product condition
|
|
with self.assertRaises(ValidationError):
|
|
cart_1.add_to_cart(self.PROD_1, 1)
|
|
cart_1.add_to_cart(self.PROD_3, 1) # Meets the category condition
|
|
cart_1.add_to_cart(self.PROD_1, 1)
|
|
|
|
def test_mandatory_conditions_are_mandatory(self):
|
|
self.add_product_enabling_condition(mandatory=False)
|
|
self.add_category_enabling_condition(mandatory=True)
|
|
|
|
cart_1 = TestingCartController.for_user(self.USER_1)
|
|
# Cannot add PROD_1 until both conditions are met
|
|
with self.assertRaises(ValidationError):
|
|
cart_1.add_to_cart(self.PROD_1, 1)
|
|
cart_1.add_to_cart(self.PROD_2, 1) # Meets the product condition
|
|
with self.assertRaises(ValidationError):
|
|
cart_1.add_to_cart(self.PROD_1, 1)
|
|
cart_1.add_to_cart(self.PROD_3, 1) # Meets the category condition
|
|
cart_1.add_to_cart(self.PROD_1, 1)
|
|
|
|
def test_available_products_works_with_no_conditions_set(self):
|
|
prods = ProductController.available_products(
|
|
self.USER_1,
|
|
category=self.CAT_1,
|
|
)
|
|
|
|
self.assertTrue(self.PROD_1 in prods)
|
|
self.assertTrue(self.PROD_2 in prods)
|
|
|
|
prods = ProductController.available_products(
|
|
self.USER_1,
|
|
category=self.CAT_2,
|
|
)
|
|
|
|
self.assertTrue(self.PROD_3 in prods)
|
|
self.assertTrue(self.PROD_4 in prods)
|
|
|
|
prods = ProductController.available_products(
|
|
self.USER_1,
|
|
products=[self.PROD_1, self.PROD_2, self.PROD_3, self.PROD_4],
|
|
)
|
|
|
|
self.assertTrue(self.PROD_1 in prods)
|
|
self.assertTrue(self.PROD_2 in prods)
|
|
self.assertTrue(self.PROD_3 in prods)
|
|
self.assertTrue(self.PROD_4 in prods)
|
|
|
|
def test_available_products_on_category_works_when_condition_not_met(self):
|
|
self.add_product_enabling_condition(mandatory=False)
|
|
|
|
prods = ProductController.available_products(
|
|
self.USER_1,
|
|
category=self.CAT_1,
|
|
)
|
|
|
|
self.assertTrue(self.PROD_1 not in prods)
|
|
self.assertTrue(self.PROD_2 in prods)
|
|
|
|
def test_available_products_on_category_works_when_condition_is_met(self):
|
|
self.add_product_enabling_condition(mandatory=False)
|
|
|
|
cart_1 = TestingCartController.for_user(self.USER_1)
|
|
cart_1.add_to_cart(self.PROD_2, 1)
|
|
|
|
prods = ProductController.available_products(
|
|
self.USER_1,
|
|
category=self.CAT_1,
|
|
)
|
|
|
|
self.assertTrue(self.PROD_1 in prods)
|
|
self.assertTrue(self.PROD_2 in prods)
|
|
|
|
def test_available_products_on_products_works_when_condition_not_met(self):
|
|
self.add_product_enabling_condition(mandatory=False)
|
|
|
|
prods = ProductController.available_products(
|
|
self.USER_1,
|
|
products=[self.PROD_1, self.PROD_2],
|
|
)
|
|
|
|
self.assertTrue(self.PROD_1 not in prods)
|
|
self.assertTrue(self.PROD_2 in prods)
|
|
|
|
def test_available_products_on_products_works_when_condition_is_met(self):
|
|
self.add_product_enabling_condition(mandatory=False)
|
|
|
|
cart_1 = TestingCartController.for_user(self.USER_1)
|
|
cart_1.add_to_cart(self.PROD_2, 1)
|
|
|
|
prods = ProductController.available_products(
|
|
self.USER_1,
|
|
products=[self.PROD_1, self.PROD_2],
|
|
)
|
|
|
|
self.assertTrue(self.PROD_1 in prods)
|
|
self.assertTrue(self.PROD_2 in prods)
|
|
|
|
def test_category_enabling_condition_fails_if_cart_refunded(self):
|
|
self.add_category_enabling_condition(mandatory=False)
|
|
|
|
cart = TestingCartController.for_user(self.USER_1)
|
|
cart.add_to_cart(self.PROD_3, 1)
|
|
|
|
|
|
cart.next_cart()
|
|
|
|
cart_2 = TestingCartController.for_user(self.USER_1)
|
|
cart_2.add_to_cart(self.PROD_1, 1)
|
|
cart_2.set_quantity(self.PROD_1, 0)
|
|
|
|
cart.cart.released = True
|
|
cart.next_cart()
|
|
|
|
with self.assertRaises(ValidationError):
|
|
cart_2.set_quantity(self.PROD_1, 1)
|
|
|
|
def test_product_enabling_condition_fails_if_cart_refunded(self):
|
|
self.add_product_enabling_condition(mandatory=False)
|
|
|
|
cart = TestingCartController.for_user(self.USER_1)
|
|
cart.add_to_cart(self.PROD_2, 1)
|
|
|
|
|
|
cart.next_cart()
|
|
|
|
cart_2 = TestingCartController.for_user(self.USER_1)
|
|
cart_2.add_to_cart(self.PROD_1, 1)
|
|
cart_2.set_quantity(self.PROD_1, 0)
|
|
|
|
cart.cart.released = True
|
|
cart.next_cart()
|
|
|
|
with self.assertRaises(ValidationError):
|
|
cart_2.set_quantity(self.PROD_1, 1)
|
|
|
|
def test_available_categories(self):
|
|
self.add_product_enabling_condition_on_category(mandatory=False)
|
|
|
|
cart_1 = TestingCartController.for_user(self.USER_1)
|
|
|
|
cats = CategoryController.available_categories(
|
|
self.USER_1,
|
|
)
|
|
|
|
self.assertFalse(self.CAT_1 in cats)
|
|
self.assertTrue(self.CAT_2 in cats)
|
|
|
|
cart_1.add_to_cart(self.PROD_3, 1)
|
|
|
|
cats = CategoryController.available_categories(
|
|
self.USER_1,
|
|
)
|
|
|
|
self.assertTrue(self.CAT_1 in cats)
|
|
self.assertTrue(self.CAT_2 in cats)
|
|
|
|
def test_validate_cart_when_enabling_conditions_become_unmet(self):
|
|
self.add_product_enabling_condition(mandatory=False)
|
|
|
|
cart = TestingCartController.for_user(self.USER_1)
|
|
cart.add_to_cart(self.PROD_2, 1)
|
|
cart.add_to_cart(self.PROD_1, 1)
|
|
|
|
# Should pass
|
|
cart.validate_cart()
|
|
|
|
cart.set_quantity(self.PROD_2, 0)
|
|
|
|
# Should fail
|
|
with self.assertRaises(ValidationError):
|
|
cart.validate_cart()
|
|
|
|
def test_fix_simple_errors_resolves_unavailable_products(self):
|
|
self.test_validate_cart_when_enabling_conditions_become_unmet()
|
|
cart = TestingCartController.for_user(self.USER_1)
|
|
|
|
# Should just remove all of the unavailable products
|
|
cart.fix_simple_errors()
|
|
# Should now succeed
|
|
cart.validate_cart()
|
|
|
|
# Should keep PROD_2 in the cart
|
|
items = rego.ProductItem.objects.filter(cart=cart.cart)
|
|
self.assertFalse([i for i in items if i.product == self.PROD_1])
|
|
|
|
def test_fix_simple_errors_does_not_remove_limited_items(self):
|
|
cart = TestingCartController.for_user(self.USER_1)
|
|
|
|
cart.add_to_cart(self.PROD_2, 1)
|
|
cart.add_to_cart(self.PROD_1, 10)
|
|
|
|
# Should just remove all of the unavailable products
|
|
cart.fix_simple_errors()
|
|
# Should now succeed
|
|
cart.validate_cart()
|
|
|
|
# Should keep PROD_2 in the cart
|
|
# and also PROD_1, which is now exhausted for user.
|
|
items = rego.ProductItem.objects.filter(cart=cart.cart)
|
|
self.assertTrue([i for i in items if i.product == self.PROD_1])
|