Making the test framework SQL prettier

OHDSI · Apr 29, 2016 · facb154 · facb154
1 parent 4cd56b6
commit facb154
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/org/ohdsi/rabbitInAHat/ETLTestFrameWorkGenerator.java b/src/org/ohdsi/rabbitInAHat/ETLTestFrameWorkGenerator.java
@@ -50,6 +50,9 @@ private static void createDeclareTestFunction(List<String> r) {
 		r.add("declareTest <- function(id, description) {");
 		r.add("  assign(\"testId\", id, envir = globalenv()) ");
 		r.add("  assign(\"testDescription\", description, envir = globalenv()) ");
+		r.add("  sql <- c(\"\", paste0(\"-- \", id, \": \", description))");
+		r.add("  assign(\"insertSql\", c(get(\"insertSql\", envir = globalenv()), sql), envir = globalenv())");
+		r.add("  assign(\"testSql\", c(get(\"testSql\", envir = globalenv()), sql), envir = globalenv())");
 		r.add("}");
 		r.add("");
 	}