1 files changed, 20 insertions, 1 deletions
diff --git a/java/org/brotli/dec/build_defs.bzl b/java/org/brotli/dec/build_defs.bzl
index d4f280b..aa4a929 100644
--- a/java/org/brotli/dec/build_defs.bzl
+++ b/java/org/brotli/dec/build_defs.bzl
@@ -4,13 +4,19 @@ _TEST_JVM_FLAGS = [
     "-DBROTLI_ENABLE_ASSERTS=true",
 ]
 
-def brotli_java_test(name, main_class = None, jvm_flags = None, **kwargs):
+_KOTLIN_DECODER_DEPS = [
+    "//org/brotli/dec/kt:dec",
+]
+
+def brotli_java_test(name, main_class = None, jvm_flags = None, test_kotlin = False, runtime_deps = [], **kwargs):
     """test duplication rule that creates 32/64-bit test pair.
 
     Args:
        name: target name prefix
        main_class: override for test_class
        jvm_flags: base Java VM options
+       test_kotlin: add target for Kotlin BrotliInputStream
+       runtime_deps: runtime target dependencies
        **kwargs: pass-through
     """
 
@@ -31,6 +37,7 @@ def brotli_java_test(name, main_class = None, jvm_flags = None, **kwargs):
         test_class = test_class,
         jvm_flags = jvm_flags + ["-DBROTLI_32_BIT_CPU=true"],
         visibility = ["//visibility:private"],
+        runtime_deps = runtime_deps,
         **kwargs
     )
 
@@ -40,5 +47,17 @@ def brotli_java_test(name, main_class = None, jvm_flags = None, **kwargs):
         test_class = test_class,
         jvm_flags = jvm_flags + ["-DBROTLI_32_BIT_CPU=false"],
         visibility = ["//visibility:private"],
+        runtime_deps = runtime_deps,
         **kwargs
     )
+
+    if test_kotlin:
+        native.java_test(
+            name = name + "_kt",
+            main_class = main_class,
+            test_class = test_class,
+            jvm_flags = jvm_flags + ["-DBROTLI_INPUT_STREAM=org.brotli.dec.kt.BrotliInputStream"],
+            visibility = ["//visibility:private"],
+            runtime_deps = runtime_deps + _KOTLIN_DECODER_DEPS,
+            **kwargs
+        )