68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
import pytz
|
|
|
|
from django.contrib.auth.models import Group
|
|
|
|
from registrasion.models import conditions
|
|
from registrasion.controllers.product import ProductController
|
|
|
|
from registrasion.tests.test_cart import RegistrationCartTestCase
|
|
|
|
UTC = pytz.timezone('UTC')
|
|
|
|
|
|
class GroupMemberTestCase(RegistrationCartTestCase):
|
|
|
|
@classmethod
|
|
def _create_group_and_flag(cls):
|
|
''' Creates cls.GROUP_1, and restricts cls.PROD_1 only to users who are
|
|
members of the group. Likewise GROUP_2 and PROD_2 '''
|
|
|
|
groups = []
|
|
products = [cls.PROD_1, cls.PROD_2]
|
|
for i, product in enumerate(products):
|
|
group = Group.objects.create(name="TEST GROUP" + str(i))
|
|
flag = conditions.GroupMemberFlag.objects.create(
|
|
description="Group member flag " + str(i),
|
|
condition=conditions.FlagBase.ENABLE_IF_TRUE,
|
|
)
|
|
flag.group.add(group)
|
|
flag.products.add(product)
|
|
|
|
groups.append(group)
|
|
|
|
cls.GROUP_1 = groups[0]
|
|
cls.GROUP_2 = groups[1]
|
|
|
|
def test_product_not_enabled_until_user_joins_group(self):
|
|
''' Tests that GroupMemberFlag disables a product for a user until
|
|
they are a member of a specific group. '''
|
|
|
|
self._create_group_and_flag()
|
|
|
|
groups = [self.GROUP_1, self.GROUP_2]
|
|
products = [self.PROD_1, self.PROD_2]
|
|
|
|
for group, product in zip(groups, products):
|
|
|
|
# USER_1 cannot see PROD_1 until they're in GROUP.
|
|
available = ProductController.available_products(
|
|
self.USER_1,
|
|
products=[product],
|
|
)
|
|
self.assertNotIn(product, available)
|
|
|
|
self.USER_1.groups.add(group)
|
|
|
|
# USER_1 cannot see PROD_1 until they're in GROUP.
|
|
available = ProductController.available_products(
|
|
self.USER_1,
|
|
products=[product],
|
|
)
|
|
self.assertIn(product, available)
|
|
|
|
# USER_2 is still locked out
|
|
available = ProductController.available_products(
|
|
self.USER_2,
|
|
products=[product],
|
|
)
|
|
self.assertNotIn(product, available)
|