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