Skip to content

Commit 7d5ba5e

Browse files
authored
test: adjusts test input body based on changes to google-api_core (#1588)
Adjusts the body of a specific test to adapt to some changes that come through from google-api-core.
1 parent d73cf49 commit 7d5ba5e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

‎tests/unit/job/test_base.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919
from google.api_core import exceptions
2020
import google.api_core.retry
21+
from google.api_core.future import polling
2122
import mock
2223
import pytest
2324

@@ -970,7 +971,7 @@ def test_result_default_wo_state(self):
970971
client = _make_client(project=self.PROJECT, connection=conn)
971972
job = self._make_one(self.JOB_ID, client)
972973

973-
self.assertIs(job.result(), job)
974+
self.assertIs(job.result(retry=polling.DEFAULT_RETRY), job)
974975

975976
begin_call = mock.call(
976977
method="POST",

0 commit comments

Comments
 (0)