def add_or_remove_nipsa(client, index, userid, action): """Add/remove the NIPSA flag to/from all of the user's annotations.""" assert action in ("add_nipsa", "remove_nipsa") if action == "add_nipsa": query = nipsa_search.not_nipsad_annotations(userid) action_func = add_nipsa_action else: query = nipsa_search.nipsad_annotations(userid) action_func = remove_nipsa_action annotations = helpers.scan(client=client, query=query) actions = [action_func(index, a) for a in annotations] helpers.bulk(client=client, actions=actions)
def test_nipsad_annotations_filters_by_nipsa(): query = search.nipsad_annotations("test_userid") must_clauses = query["query"]["filtered"]["filter"]["bool"]["must"] assert {"term": {"nipsa": True}} in must_clauses