def test_update_es_user_with_groups(self):
group_doc = {
'name': 'g1',
'_id': 'group1',
'users': []
}
# no change if user not in group
update_es_user_with_groups(group_doc)
self._check_es_user(None, None)
# user added to group
group_doc['users'] = [self.user_id]
update_es_user_with_groups(group_doc)
self._check_es_user(['group1'], ['g1'])
# re-process group with no change
update_es_user_with_groups(group_doc)
self._check_es_user(['group1'], ['g1'])
# user added to new group
new_group = {
'name': 'g2',
'_id': 'group2',
'users': [self.user_id]
}
update_es_user_with_groups(new_group)
self._check_es_user(['group1', 'group2'], ['g1', 'g2'])
def test_update_es_user_with_groups(self):
group_doc = {
'name': 'g1',
'_id': 'group1',
'users': []
}
# no change if user not in group
update_es_user_with_groups(group_doc)
self._check_es_user(None, None)
# user added to group
group_doc['users'] = [self.user_id]
update_es_user_with_groups(group_doc)
self._check_es_user(['group1'], ['g1'])
# re-process group with no change
update_es_user_with_groups(group_doc)
self._check_es_user(['group1'], ['g1'])
# user added to new group
new_group = {
'name': 'g2',
'_id': 'group2',
'users': [self.user_id]
}
update_es_user_with_groups(new_group)
self._check_es_user(['group1', 'group2'], ['g1', 'g2'])
def test_update_es_user_with_groups_remove_user(self):
group_doc = {
'name': 'g1',
'_id': 'group1',
'users': [self.user_id],
'removed_users': set([]),
}
# re-process group with no change
update_es_user_with_groups(group_doc)
self._check_es_user(['group1'], ['g1'])
group_doc['removed_users'].add(self.user_id)
group_doc['users'] = []
update_es_user_with_groups(group_doc)
self._check_es_user(None, None)
def test_update_es_user_with_groups_remove_user(self):
group_doc = {
'name': 'g1',
'_id': 'group1',
'users': [self.user_id],
'removed_users': set([]),
}
# re-process group with no change
update_es_user_with_groups(group_doc)
self._check_es_user(['group1'], ['g1'])
group_doc['removed_users'].add(self.user_id)
group_doc['users'] = []
update_es_user_with_groups(group_doc)
self._check_es_user(None, None)
def test_remove_user_from_groups_partial_match(self):
original_id = uuid.uuid4().hex
group_doc = {
'name': 'original',
'_id': original_id,
'users': [self.user_id]
}
# set original groups on the user
update_es_user_with_groups(group_doc)
self._check_es_user([original_id], ['original'])
new_id = uuid.uuid4().hex
group_doc = {
'name': 'original',
'_id': new_id,
'users': [self.user_id]
}
remove_group_from_users(group_doc, self.es_client)
def test_remove_user_from_groups_partial_match(self):
original_id = uuid.uuid4().hex
group_doc = {
'name': 'original',
'_id': original_id,
'users': [self.user_id]
}
# set original groups on the user
update_es_user_with_groups(group_doc)
self._check_es_user([original_id], ['original'])
new_id = uuid.uuid4().hex
group_doc = {
'name': 'original',
'_id': new_id,
'users': [self.user_id]
}
remove_group_from_users(group_doc, self.es_client)
