テストでAnnotation Processorを利用する

testAnnotationProcessor "..."
と書くと実行されてクラスは作られてテストも想定通り動くが、Android Studio上ではクラスが参照できずにエラーになる。
https://issuetracker.google.com/issues/37121918
関連したgistに回答があった。
gist.github.com
android {
  defaultConfig {
      android.libraryVariants.all {
        def aptOutputDir = new File(buildDir, "generated/source/apt/${it.unitTestVariant.dirName}")
        it.unitTestVariant.addJavaSourceFoldersToModel(aptOutputDir)
      }
  }
}