Add tests using implicit java_test args attr

dwtj · dwtj · commit 18fd0a4f2d04 · 2020-10-29T17:43:00.000-07:00
diff --git a/test/workspaces/smoke_test_java_rule_attributes/attrs/args/BUILD b/test/workspaces/smoke_test_java_rule_attributes/attrs/args/BUILD
@@ -0,0 +1,28 @@
+load("@dwtj_rules_java//java:defs.bzl", "dwtj_java_test")
+
+dwtj_java_test(
+    name = "expect_zero_args",
+    main_class = "attrs.args.ExpectNumArgs",
+    srcs = ["ExpectNumArgs.java"],
+    args = [],
+)
+
+dwtj_java_test(
+    name = "expect_one_arg",
+    main_class = "attrs.args.ExpectNumArgs",
+    srcs = ["ExpectNumArgs.java"],
+    args = ["1"],
+)
+
+dwtj_java_test(
+    name = "expect_five_args",
+    main_class = "attrs.args.ExpectNumArgs",
+    srcs = ["ExpectNumArgs.java"],
+    args = [
+        "5",
+        "a",
+        "b",
+        "c",
+        "d",
+    ],
+)
diff --git a/test/workspaces/smoke_test_java_rule_attributes/attrs/args/ExpectNumArgs.java b/test/workspaces/smoke_test_java_rule_attributes/attrs/args/ExpectNumArgs.java
@@ -0,0 +1,22 @@
+package attrs.args;
+
+public class ExpectNumArgs {
+    public static void main(String[] args) {
+        if (args.length == 0) {
+            System.exit(0);
+        }
+
+        int numArgs = Integer.parseInt(args[0]);
+        if (args.length != numArgs) {
+            System.err.println("Wrong number of arguments found.");
+            System.err.println("Expected number of args: " + numArgs);
+            System.err.println("Actual number of args: " + args.length);
+            System.exit(-1);
+        } else {
+            for (String arg : args) {
+                System.out.println(arg);
+            }
+            System.exit(0);
+        }
+    }
+}
