Convenience builder for dependencies from classpath

openrewrite · Aug 4, 2020 · f440a71 · f440a71
1 parent 80dfc6a
commit f440a71
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/rewrite-java/src/main/java/org/openrewrite/java/JavaParser.java b/rewrite-java/src/main/java/org/openrewrite/java/JavaParser.java
@@ -18,7 +18,6 @@
 import io.micrometer.core.instrument.MeterRegistry;
 import io.micrometer.core.instrument.Metrics;
 import org.openrewrite.Parser;
-import org.openrewrite.SourceFile;
 import org.openrewrite.internal.lang.Nullable;
 import org.openrewrite.java.style.ImportLayoutStyle;
 import org.openrewrite.java.style.TabAndIndentStyle;
@@ -166,6 +165,11 @@ public B classpath(List<Path> classpath) {
             return (B) this;
         }
 
+        public B classpath(String... classpath) {
+            this.classpath = dependenciesFromClasspath(classpath);
+            return (B) this;
+        }
+
         public B importStyle(ImportLayoutStyle importStyle) {
             this.styles.add(importStyle);
             return (B) this;

diff --git a/rewrite-test/src/main/kotlin/org/openrewrite/java/AddImportTest.kt b/rewrite-test/src/main/kotlin/org/openrewrite/java/AddImportTest.kt
@@ -16,9 +16,10 @@
 package org.openrewrite.java
 
 import org.junit.jupiter.api.Test
+import org.openrewrite.RefactorVisitorTest
 import org.openrewrite.whenParsedBy
 
-interface AddImportTest {
+interface AddImportTest: RefactorVisitorTest {
     @Test
     fun addMultipleImports(jp: JavaParser) {
         "class A {}"