@@ -926,7 +926,6 @@ public void testGetModelNotFoundWhenThrowIsEnabled() throws IOException {
926
926
String expected = "Model not found" ;
927
927
when (bigqueryRpcMock .getModelSkipExceptionTranslation (
928
928
PROJECT , DATASET , MODEL , EMPTY_RPC_OPTIONS ))
929
- .thenReturn (null )
930
929
.thenThrow (new BigQueryException (404 , expected ));
931
930
options .setThrowNotFound (true );
932
931
bigquery = options .getService ();
@@ -1782,6 +1781,12 @@ public void testCreateJobWithBigQueryRetryConfigFailureShouldNotRetry() throws I
1782
1781
jobCapture .capture (), eq (bigQueryRpcOptions )))
1783
1782
.thenThrow (new BigQueryException (400 , RATE_LIMIT_ERROR_MSG ));
1784
1783
1784
+ // Job create will attempt to retrieve the job even in the case when the job is created in a
1785
+ // returned failure.
1786
+ when (bigqueryRpcMock .getJobSkipExceptionTranslation (
1787
+ nullable (String .class ), nullable (String .class ), nullable (String .class ), Mockito .any ()))
1788
+ .thenThrow (new BigQueryException (500 , "InternalError" ));
1789
+
1785
1790
bigquery = options .getService ();
1786
1791
bigquery =
1787
1792
options
@@ -1839,7 +1844,7 @@ public void testCreateJobWithRetryOptionsFailureShouldNotRetry() throws IOExcept
1839
1844
.thenThrow (new BigQueryException (500 , "InternalError" ))
1840
1845
.thenReturn (newJobPb ());
1841
1846
1842
- // Job create will attempt to retrieve the job even in the case when the job is created a
1847
+ // Job create will attempt to retrieve the job even in the case when the job is created in a
1843
1848
// returned failure.
1844
1849
when (bigqueryRpcMock .getJobSkipExceptionTranslation (
1845
1850
nullable (String .class ), nullable (String .class ), nullable (String .class ), Mockito .any ()))
0 commit comments