Conservancy/symposion_app
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Adds test for GroupMemberCondition

Browse source
This commit is contained in:
Christopher Neugebauer 2016-09-15 15:33:43 +10:00
parent 31d057c750
commit 2c8ed9a51a
1 changed files with 39 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
registrasion/tests/test_group_member.py
View file

@ -19,21 +19,24 @@ class GroupMemberTestCase(RegistrationCartTestCase):
@classmethod @classmethod
def _create_group_and_flag(cls): def _create_group_and_flag(cls):
''' Creates cls.GROUP, and restricts cls.PROD_1 only to users who are ''' Creates cls.GROUP_1, and restricts cls.PROD_1 only to users who are
members of the group. ''' members of the group. Likewise GROUP_2 and PROD_2 '''
group = Group.objects.create(
name="TEST GROUP",
)
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( flag = conditions.GroupMemberFlag.objects.create(
description="Group member flag", description="Group member flag " + str(i),
condition=conditions.FlagBase.ENABLE_IF_TRUE, condition=conditions.FlagBase.ENABLE_IF_TRUE,
) )
flag.group.add(group) flag.group.add(group)
flag.products.add(cls.PROD_1) flag.products.add(product)
cls.GROUP = group groups.append(group)
cls.GROUP_1 = groups[0]
cls.GROUP_2 = groups[1]
def test_product_not_enabled_until_user_joins_group(self): def test_product_not_enabled_until_user_joins_group(self):
''' Tests that GroupMemberFlag disables a product for a user until ''' Tests that GroupMemberFlag disables a product for a user until
@ -41,25 +44,30 @@ class GroupMemberTestCase(RegistrationCartTestCase):
self._create_group_and_flag() self._create_group_and_flag()
# USER_1 cannot see PROD_1 until they're in GROUP. groups = [self.GROUP_1, self.GROUP_2]
available = ProductController.available_products( products = [self.PROD_1, self.PROD_2]
self.USER_1,
products=[self.PROD_1],
)
self.assertNotIn(self.PROD_1, available)
self.USER_1.groups.add(self.GROUP) for group, product in zip(groups, products):
# USER_1 cannot see PROD_1 until they're in GROUP. # USER_1 cannot see PROD_1 until they're in GROUP.
available = ProductController.available_products( available = ProductController.available_products(
self.USER_1, self.USER_1,
products=[self.PROD_1], products=[product],
) )
self.assertIn(self.PROD_1, available) 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 # USER_2 is still locked out
available = ProductController.available_products( available = ProductController.available_products(
self.USER_2, self.USER_2,
products=[self.PROD_1], products=[product],
) )
self.assertNotIn(self.PROD_1, available) self.assertNotIn(product, available)