From b4a5902faff7a7849c66a9c2f5b11a4840a5ef0e Mon Sep 17 00:00:00 2001
From: Clinton Alexander <clinton.alexander@r3cev.com>
Date: Fri, 17 Mar 2017 12:17:50 +0000
Subject: [PATCH] Test build now compiles correctly with kapt removed from the
 build tree.

---
 node-schemas/build.gradle | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/node-schemas/build.gradle b/node-schemas/build.gradle
index 56e61cfd6c..df382cb787 100644
--- a/node-schemas/build.gradle
+++ b/node-schemas/build.gradle
@@ -27,8 +27,13 @@ sourceSets {
     generated {
         java {
             srcDir "${projectDir}/generated/source/kapt/main/"
+            compileClasspath += sourceSets.main.compileClasspath + sourceSets.main.output
         }
     }
+    test {
+        compileClasspath += sourceSets.generated.output + sourceSets.main.compileClasspath
+        runtimeClasspath += compileClasspath
+    }
 }
 
 dependencies {
@@ -60,8 +65,10 @@ project.afterEvaluate {
     jar.dependsOn.each { jar.dependsOn.remove it }
     buildGenerated.dependsOn.each { buildGenerated.dependsOn.remove it }
     jar.dependsOn buildGenerated
+    compileTestKotlin.dependsOn buildGenerated
     buildGenerated.dependsOn classes
     compileKotlin.dependsOn.remove kaptKotlin
+    compileTestKotlin.dependsOn.remove kaptTestKotlin
     buildKapt.dependsOn kaptKotlin
     buildKapt.mustRunAfter kaptKotlin
 }