def test_uses_request_options_on_wait(self): index = SearchClient.create("foo", "bar").init_index("foo") index.wait_task = mock.Mock(name="wait_task") index._sync = mock.Mock(name="_sync") index._sync.return_value = index response = IndexingResponse(index, [{"taskID": 1}]) response.wait({"bar": 2}) index.wait_task.assert_called_once_with(1, {"bar": 2})
def test_uses_request_options_on_wait(self): index = SearchClient.create('foo', 'bar').init_index('foo') index.wait_task = mock.Mock(name='wait_task') index._sync = mock.Mock(name='_sync') index._sync.return_value = index response = IndexingResponse(index, [{'taskID': 1}]) response.wait({'bar': 2}) index.wait_task.assert_called_once_with(1, {'bar': 2})