fix: remove unused variable (#324)

* fix: remove unused variable * feat: apply setOf to copy with multiple arguments
openrewrite · Oct 22, 2023 · 512712c · 512712c
1 parent ab08480
commit 512712c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/main/java/org/openrewrite/java/migrate/util/UseEnumSetOf.java b/src/main/java/org/openrewrite/java/migrate/util/UseEnumSetOf.java
@@ -66,10 +66,11 @@ public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, Execu
                         if (isAssignmentSetOfEnum(type)) {
                             maybeAddImport("java.util.EnumSet");
 
-                            StringJoiner setOf = new StringJoiner(", ", "EnumSet.of(", ")");
                             List<Expression> args = m.getArguments();
+                            StringJoiner setOf = new StringJoiner(", ", "EnumSet.of(", ")");
                             args.forEach(o -> setOf.add("#{any()}"));
-                            return JavaTemplate.builder("EnumSet.of(#{any()})")
+
+                            return JavaTemplate.builder(setOf.toString())
                                     .contextSensitive()
                                     .imports("java.util.EnumSet")
                                     .build()

diff --git a/src/test/java/org/openrewrite/java/migrate/util/UseEnumSetOfTest.java b/src/test/java/org/openrewrite/java/migrate/util/UseEnumSetOfTest.java
@@ -44,7 +44,7 @@ public enum Color {
                         RED, GREEN, BLUE
                     }
                     public void method() {
-                        Set<Color> warm = Set.of(Color.RED);
+                        Set<Color> warm = Set.of(Color.RED, Color.GREEN);
                     }
                 }
                 """,
@@ -57,7 +57,7 @@ public enum Color {
                         RED, GREEN, BLUE
                     }
                     public void method() {
-                        Set<Color> warm = EnumSet.of(Color.RED);
+                        Set<Color> warm = EnumSet.of(Color.RED, Color.GREEN);
                     }
                 }
                 """