From facb154908e74c96c83c6297fe6996aff0882d72 Mon Sep 17 00:00:00 2001
From: schuemie <schuemie@ohdsi.org>
Date: Fri, 29 Apr 2016 11:07:16 +0200
Subject: [PATCH] Making the test framework SQL prettier

---
 src/org/ohdsi/rabbitInAHat/ETLTestFrameWorkGenerator.java | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/org/ohdsi/rabbitInAHat/ETLTestFrameWorkGenerator.java b/src/org/ohdsi/rabbitInAHat/ETLTestFrameWorkGenerator.java
index 690a5880..5f321a44 100644
--- 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("");
 	}