def test_remove_nipsa_gets_query(search, _): worker.add_or_remove_nipsa(client=mock.Mock(), index="foo", userid="test_userid", action="remove_nipsa") search.nipsad_annotations.assert_called_once_with("test_userid")
def test_remove_nipsa_passes_es_client_to_scan(_, helpers): client = mock.Mock() worker.add_or_remove_nipsa(client=client, index="foo", userid="test_userid", action="remove_nipsa") assert helpers.scan.call_args[1]["client"] == client
def test_remove_nipsa_passes_query_to_scan(search, helpers): query = mock.MagicMock() search.nipsad_annotations.return_value = query worker.add_or_remove_nipsa(client=mock.Mock(), index="foo", userid="test_userid", action="remove_nipsa") assert helpers.scan.call_args[1]["query"] == query
def test_remove_nipsa_passes_actions_to_bulk(_, helpers): helpers.scan.return_value = [ {"_id": "foo"}, {"_id": "bar"}, {"_id": "gar"}] worker.add_or_remove_nipsa(client=mock.Mock(), index="foo", userid="test_userid", action="remove_nipsa") actions = helpers.bulk.call_args[1]["actions"] assert [action["_id"] for action in actions] == ["foo", "bar", "gar"]
def test_remove_nipsa_passes_actions_to_bulk(_, helpers): helpers.scan.return_value = [{ "_id": "foo" }, { "_id": "bar" }, { "_id": "gar" }] worker.add_or_remove_nipsa(client=mock.Mock(), index="foo", userid="test_userid", action="remove_nipsa") actions = helpers.bulk.call_args[1]["actions"] assert [action["_id"] for action in actions] == ["foo", "bar", "gar"]