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