@@ -717,7 +717,7 @@ public void testPreparedStatementExecuteQueryWithLargeResults() throws Exception
717717 Job job = getJobMock (result , jobConfiguration , StatementType .SELECT );
718718
719719 doReturn (job ).when (bigquery ).queryWithTimeout (any (), any (), any ());
720- doReturn (jsonResultSet ).when (preparedStatementSpy ).processJsonResultSet (result );
720+ doReturn (jsonResultSet ).when (preparedStatementSpy ).processJsonResultSet (eq ( result ), any () );
721721
722722 Job dryRunJob = getJobMock (null , jobConfiguration , StatementType .SELECT );
723723 doReturn (dryRunJob ).when (bigquery ).create (any (JobInfo .class ));
@@ -757,7 +757,7 @@ public void testPreparedStatementExecuteWithLargeResults() throws Exception {
757757 Job job = getJobMock (result , jobConfiguration , StatementType .SELECT );
758758
759759 doReturn (job ).when (bigquery ).queryWithTimeout (any (), any (), any ());
760- doReturn (jsonResultSet ).when (preparedStatementSpy ).processJsonResultSet (result );
760+ doReturn (jsonResultSet ).when (preparedStatementSpy ).processJsonResultSet (eq ( result ), any () );
761761
762762 Job dryRunJob = getJobMock (null , jobConfiguration , StatementType .SELECT );
763763 doReturn (dryRunJob ).when (bigquery ).create (any (JobInfo .class ));
@@ -774,6 +774,9 @@ public void testPreparedStatementExecuteWithLargeResults() throws Exception {
774774 assertThat (capturedConfig .getDestinationTable ())
775775 .isEqualTo (TableId .of ("test_dataset" , "test_table" ));
776776 assertThat (capturedConfig .allowLargeResults ()).isTrue ();
777+ }
778+
779+ @ Test
777780 public void testSetFetchSizeNegativeThrows () {
778781 org .junit .jupiter .api .Assertions .assertThrows (
779782 SQLException .class , () -> bigQueryStatement .setFetchSize (-1 ));
0 commit comments