Pass compile

Author: kunli2

src/main/java/org/openrewrite/java/spring/boot2/ChangeEmbeddedServletContainerCustomizer.java

@@ -82,7 +82,10 @@ private static J.ParameterizedType getWebFactoryCustomizerIdentifier(ExecutionCo
                     "import org.springframework.boot.web.server.WebServerFactoryCustomizer;\n" +
                             "import org.springframework.boot.web.servlet.server.ConfigurableServletWebServerFactory;\n" +
                             "public abstract class Template implements WebServerFactoryCustomizer<ConfigurableServletWebServerFactory> {}"
-            ).findFirst().orElseThrow(NoSuchElementException::new);
+            )
+                .map(J.CompilationUnit.class::cast)
+                .findFirst()
+                .get();
 
             webFactoryCustomizerIdentifier = (J.ParameterizedType) requireNonNull(cu.getClasses()
                     .get(0).getImplements()).get(0);

src/main/java/org/openrewrite/java/spring/boot2/MoveAutoConfigurationToImportsFile.java

@@ -100,7 +100,10 @@ public Collection<SourceFile> generate(Accumulator acc, ExecutionContext ctx) {
             Collections.sort(finalList);
 
             PlainTextParser parser = new PlainTextParser();
-            PlainText brandNewFile = parser.parse(String.join("\n", finalList)).findFirst().get();
+            PlainText brandNewFile = parser.parse(String.join("\n", finalList))
+                .map(PlainText.class::cast)
+                .findFirst()
+                .get();
             newImportFiles.add(brandNewFile
                     .withSourcePath(entry.getKey())
                     .withMarkers(brandNewFile.getMarkers().withMarkers(entry.getValue().getMarkers()))