Take skipped shards as successful in scan helper

This commit is contained in:
Honza Král
2019-10-04 11:46:48 +02:00
parent a8c858718e
commit fc78c992ef
2 changed files with 8 additions and 7 deletions
@@ -311,12 +311,12 @@ class TestScan(ElasticsearchTestCase):
mock_scroll_responses = [
{
"_scroll_id": "dummy_id",
"_shards": {"successful": 4, "total": 5},
"_shards": {"successful": 4, "total": 5, "skipped": 0},
"hits": {"hits": [{"scroll_data": 42}]},
},
{
"_scroll_id": "dummy_id",
"_shards": {"successful": 4, "total": 5},
"_shards": {"successful": 4, "total": 5, "skipped": 0},
"hits": {"hits": []},
},
]
@@ -398,7 +398,7 @@ class TestScan(ElasticsearchTestCase):
with patch.object(self, "client") as client_mock:
client_mock.search.return_value = {
"_scroll_id": "dummy_id",
"_shards": {"successful": 4, "total": 5},
"_shards": {"successful": 4, "total": 5, "skipped": 0},
"hits": {"hits": [{"search_data": 1}]},
}
client_mock.scroll.side_effect = self.mock_scroll_responses