From 775a9e6a6fe4f3ef44b1051fd0813e7bb50958b2 Mon Sep 17 00:00:00 2001
From: Tink Team <tink-dev@google.com>
Date: Wed, 3 Apr 2024 05:19:25 -0700
Subject: [PATCH] Add `load()` statements for the builtin Bazel java rules

Loads are being added in preparation for moving the rules out of Bazel and into `rules_java`.

PiperOrigin-RevId: 621489359
---
 java_src/tools/gen_java_test_rules.bzl | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/java_src/tools/gen_java_test_rules.bzl b/java_src/tools/gen_java_test_rules.bzl
index 26a8f3d439..ab532d4c5c 100644
--- a/java_src/tools/gen_java_test_rules.bzl
+++ b/java_src/tools/gen_java_test_rules.bzl
@@ -13,7 +13,6 @@
 # limitations under the License.
 #
 ################################################################################
-
 """Generate Java test rules from given test_files.
 
 Instead of having to create one test rule per test in the BUILD file, this rule
@@ -22,6 +21,8 @@ files.
 
 """
 
+load("//third_party/bazel_rules/rules_java/java:java_test.bzl", "java_test")
+
 def gen_java_test_rules(
         test_files,
         deps,
@@ -70,7 +71,7 @@ def gen_java_test_rules(
         java_class = _package_from_path(
             native.package_name() + "/" + _strip_right(test, ".java"),
         )
-        native.java_test(
+        java_test(
             name = prefix + test,
             runtime_deps = deps,
             data = data,