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_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_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_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"]
