Fix IT.

Signed-off-by: Yury-Fridlyand <[email protected]>
Bit-Quill · Aug 21, 2023 · 35a13e5 · 35a13e5
1 parent db3e860
commit 35a13e5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/integ-test/src/test/java/org/opensearch/sql/correctness/tests/JDBCConnectionTest.java b/integ-test/src/test/java/org/opensearch/sql/correctness/tests/JDBCConnectionTest.java
@@ -50,6 +50,7 @@ public void setUp() throws SQLException {
     conn = new JDBCConnection("Test DB", "jdbc:testdb://localhost:12345");
     conn.setConnection(connection);
 
+    when(connection.createStatement()).thenReturn(statement);
     when(connection.createStatement(ResultSet.TYPE_FORWARD_ONLY, ResultSet.CONCUR_READ_ONLY))
         .thenReturn(statement);
   }