def setUp(self): super().setUp() property_group_data = { # name, granted, denied 'active': ({'login', 'mail'}, set()), 'mail_only': ({'mail'}, set()), 'violation': (set(), {'login'}), } self.groups = {} for name, (granted, denied) in property_group_data.items(): self.groups[name] = PropertyGroupFactory.create(name=name, granted=granted, denied=denied) self.users = dict( zip(['active', 'mail', 'violator', 'former'], UserFactory.create_batch(4))) memberships = [ # user, group, delta_days_start, delta_days_end ('active', 'active', -1, None), ('mail', 'mail_only', -1, +1), ('mail', 'active', +2, None), ('violator', 'active', -1, None), ('violator', 'violation', -1, None), ('former', 'active', -10, -1), ('former', 'violation', -9, -5), ] for username, groupname, delta_days_start, delta_days_end in memberships: start = (datetime.now() + timedelta(delta_days_start) if delta_days_start is not None else None) end = (datetime.now() + timedelta(delta_days_end) if delta_days_end is not None else None) MembershipFactory.create(user=self.users[username], group=self.groups[groupname], begins_at=start, ends_at=end) session.session.commit()
def setUp(self): super().setUp() property_group_data = { # name, granted, denied 'active': ({'login', 'mail'}, set()), 'mail_only': ({'mail'}, set()), 'violation': (set(), {'login'}), } self.groups = {} for name, (granted, denied) in property_group_data.items(): self.groups[name] = PropertyGroupFactory.create(name=name, granted=granted, denied=denied) self.users = dict(zip(['active', 'mail', 'violator', 'former'], UserFactory.create_batch(4))) memberships = [ # user, group, delta_days_start, delta_days_end ('active', 'active', -1, None), ('mail', 'mail_only', -1, +1), ('mail', 'active', +2, None), ('violator', 'active', -1, None), ('violator', 'violation', -1, None), ('former', 'active', -10, -1), ('former', 'violation', -9, -5), ] for username, groupname, delta_days_start, delta_days_end in memberships: start = (datetime.now() + timedelta(delta_days_start) if delta_days_start is not None else None) end = (datetime.now() + timedelta(delta_days_end) if delta_days_end is not None else None) MembershipFactory.create(user=self.users[username], group=self.groups[groupname], begins_at=start, ends_at=end) session.session.commit()
def users(self): return UserFactory.create_batch(10)