add mock for createStatement

CMSgov · Jun 11, 2024 · d8454f4 · d8454f4
1 parent f869186
commit d8454f4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/optout/src/test/java/gov/cms/ab2d/optout/OptOutProcessorTest.java b/optout/src/test/java/gov/cms/ab2d/optout/OptOutProcessorTest.java
@@ -49,6 +49,7 @@ static void beforeAll() throws SQLException {
         mockStatic(DriverManager.class)
                 .when(() ->  DriverManager.getConnection(anyString(), anyString(), anyString())).thenReturn(dbConnection);
         when(dbConnection.prepareStatement(anyString())).thenReturn(preparedStatement);
+        when(dbConnection.createStatement()).thenReturn(preparedStatement);
         when(preparedStatement.executeQuery(anyString())).thenReturn(resultSet);
         when(resultSet.next()).thenReturn(true).thenReturn(false);
         when(resultSet.getInt(any())).thenReturn(TEST_TOTAL_RESULT_COUNT);