Batch Test improvements

[aaf/authz.git] / auth / auth-batch / src / main / java / org / onap / aaf / auth / batch / helpers / CQLBatch.java

diff --git a/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/helpers/CQLBatch.java b/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/helpers/CQLBatch.java
index efd1ec9..9f685ad 100644 (file)
--- a/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/helpers/CQLBatch.java
+++ b/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/helpers/CQLBatch.java
@@ -63,12 +63,14 @@ public class CQLBatch {
        }
        
        public ResultSet execute(boolean dryRun) {
+               ResultSet rv = null;
                if(dryRun) {
                        end();
-                       return null;
                } else {
-                       return execute();
+                       rv = execute();
                }
+               sb.setLength(0);
+               return rv;
        }
        
        public void touch(String table, int begin, int end, boolean dryRun) {
@@ -82,4 +84,8 @@ public class CQLBatch {
                }
                execute(dryRun);
        }
+       
+       public String toString() {
+               return sb.toString();
+       }
 }