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
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
