Enable output_buffer before calling ut_runner.run

viniciusam · viniciusam · commit 0d8501b6ac91 · 2017-10-22T17:04:51.000+01:00
diff --git a/src/main/java/org/utplsql/api/DBHelper.java b/src/main/java/org/utplsql/api/DBHelper.java
@@ -96,4 +96,38 @@ public static boolean versionCompatibilityCheck(Connection conn)
         return versionCompatibilityCheck(conn, UTPLSQL_VERSION);
     }
 
+    /**
+     * Enables the dbms_output buffer.
+     * @param conn the connection
+     * @param bufferLen the buffer length
+     */
+    public static void enableDBMSOutput(Connection conn, int bufferLen) {
+        try (CallableStatement call = conn.prepareCall("BEGIN dbms_output.enable(?); END;")) {
+            call.setInt(1, bufferLen);
+            call.execute();
+        } catch (SQLException e) {
+            System.out.println("Failed to enable dbms_output.");
+        }
+    }
+
+    /**
+     * Enables the dbms_output buffer.
+     * @param conn the connection
+     */
+    public static void enableDBMSOutput(Connection conn) {
+        enableDBMSOutput(conn, Integer.MAX_VALUE);
+    }
+
+    /**
+     * Disables the dbms_output buffer.
+     * @param conn the connection
+     */
+    public static void disableDBMSOutput(Connection conn) {
+        try (CallableStatement call = conn.prepareCall("BEGIN dbms_output.disable(); END;")) {
+            call.execute();
+        } catch (SQLException e) {
+            System.out.println("Failed to disable dbms_output.");
+        }
+    }
+
 }
diff --git a/src/main/java/org/utplsql/api/TestRunner.java b/src/main/java/org/utplsql/api/TestRunner.java
@@ -103,6 +103,8 @@ public void run(Connection conn) throws SomeTestsFailedException, SQLException {
         OracleConnection oraConn = conn.unwrap(OracleConnection.class);
         CallableStatement callableStatement = null;
         try {
+            DBHelper.enableDBMSOutput(conn);
+
             callableStatement = conn.prepareCall(
                     "BEGIN " +
                         "ut_runner.run(" +
@@ -172,8 +174,11 @@ public void run(Connection conn) throws SomeTestsFailedException, SQLException {
                 throw e;
             }
         } finally {
-            if (callableStatement != null)
+            if (callableStatement != null) {
                 callableStatement.close();
+            }
+
+            DBHelper.disableDBMSOutput(conn);
         }
     }
 
