junit_suite_test.bzl

def canonacalize_package_name(prefix, pkg_name):
  return pkg_name.replace(prefix, "").replace("/", ".")

def junit_suite_test(name, srcs, deps, prefix = "src/test/java", size="small", resources=[], classpath_resources=[], jvm_flags=[], tags=[], data=[]):
  tests = []
  if not prefix.endswith("/"):
    prefix = prefix + "/"

  for src in srcs:
    if src.endswith("Test.java"):
      tests += [canonacalize_package_name(prefix, src.replace(".java", ".class"))]

  package = canonacalize_package_name(prefix, PACKAGE_NAME).replace("-", "_")

  native.genrule(
    name = name + "-AllTests-gen",
    outs = ["AllTests.java"],
    cmd = """
      cat <<EOF >> $@
package %s;

import org.junit.runner.RunWith;
import org.junit.runners.Suite;

@RunWith(Suite.class)
@Suite.SuiteClasses({
  %s
})
public class AllTests {
}
EOF
    """ % (package, ",\n  ".join(tests))
  )

  native.java_test(
    name = name,
    srcs = srcs + ["AllTests.java"],
    test_class = package + ".AllTests",
    resources = resources,
    classpath_resources = classpath_resources,
    data = data,
    size = size,
    tags = tags,
    jvm_flags = jvm_flags,
    deps = deps
  )