From 0e1baa4d3c2044ae8369f809d0d3f7a8aeeacde2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fabian=20Kr=C3=BCger?= <fkrueger@vmware.com>
Date: Sun, 24 Sep 2023 14:22:06 +0200
Subject: [PATCH] Make sbm-recipes-jee-to-boot compile

---
 .../ControlledInstantiationOfExecutionContextTest.java       | 2 +-
 .../sbm/jee/jaxrs/recipes/ResponseBuilderTest.java           | 4 +++-
 .../jee/jpa/actions/MigrateEclipseLinkToSpringBootTest.java  | 5 ++---
 ...stenceXmlToSpringBootApplicationPropertiesActionTest.java | 4 +---
 4 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/architecture/ControlledInstantiationOfExecutionContextTest.java b/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/architecture/ControlledInstantiationOfExecutionContextTest.java
index 4ca962036..70e08f74f 100644
--- a/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/architecture/ControlledInstantiationOfExecutionContextTest.java
+++ b/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/architecture/ControlledInstantiationOfExecutionContextTest.java
@@ -24,7 +24,7 @@
 import com.tngtech.archunit.lang.ArchRule;
 import org.openrewrite.ExecutionContext;
 import org.springframework.sbm.boot.autoconfigure.ScopeConfiguration;
-import org.springframework.sbm.openrewrite.RewriteExecutionContext;
+import org.springframework.sbm.parsers.RewriteExecutionContext;
 
 import static com.tngtech.archunit.lang.syntax.ArchRuleDefinition.noClasses;
 
diff --git a/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jaxrs/recipes/ResponseBuilderTest.java b/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jaxrs/recipes/ResponseBuilderTest.java
index 30cd81332..482372eb5 100644
--- a/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jaxrs/recipes/ResponseBuilderTest.java
+++ b/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jaxrs/recipes/ResponseBuilderTest.java
@@ -15,6 +15,7 @@
  */
 package org.springframework.sbm.jee.jaxrs.recipes;
 
+import org.openrewrite.SourceFile;
 import org.openrewrite.java.tree.J;
 import org.springframework.sbm.engine.recipe.AbstractAction;
 import org.springframework.sbm.engine.context.ProjectContext;
@@ -26,6 +27,7 @@
 import org.junit.jupiter.api.Test;
 
 import java.util.List;
+import java.util.stream.Stream;
 
 import static org.assertj.core.api.Assertions.assertThat;
 
@@ -375,7 +377,7 @@ void type() {
         String actual = projectContext.getProjectJavaSources().list().get(0).print();
 
         // verify it compiles
-        List<J.CompilationUnit> parse = new RewriteJavaParser(new SbmApplicationProperties(),
+        Stream<SourceFile> parse = new RewriteJavaParser(new SbmApplicationProperties(),
                                                               new RewriteExecutionContext()).parse(actual);
 
         assertThat(actual)
diff --git a/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jpa/actions/MigrateEclipseLinkToSpringBootTest.java b/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jpa/actions/MigrateEclipseLinkToSpringBootTest.java
index c1ef87d88..2c0da3627 100644
--- a/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jpa/actions/MigrateEclipseLinkToSpringBootTest.java
+++ b/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jpa/actions/MigrateEclipseLinkToSpringBootTest.java
@@ -82,7 +82,7 @@ void shouldAddDependencyToSpringBootStarterDataJpaIfNoneExists() {
 
         sut.apply(projectContext);
 
-        assertThat(projectContext.getBuildFile().print()).isEqualTo(
+        assertThat(projectContext.getApplicationModules().getRootModule().getBuildFile().print()).isEqualTo(
                 "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" +
                 "<project xmlns=\"http://maven.apache.org/POM/4.0.0\"\n" +
                 "         xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"\n" +
@@ -152,7 +152,7 @@ void shouldAddExcludesToSpringBootStarterDataJpaIfExists() {
 
         sut.apply(projectContext);
 
-        assertThat(projectContext.getBuildFile().print()).isEqualTo(
+        assertThat(projectContext.getApplicationModules().getRootModule().getBuildFile().print()).isEqualTo(
                 "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" +
                 "<project xmlns=\"http://maven.apache.org/POM/4.0.0\"\n" +
                 "         xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"\n" +
@@ -338,7 +338,6 @@ void shouldAddEclipseLinkConfigurationClass() {
         ProjectContext projectContext = TestProjectContext.buildProjectContext()
                 .withMavenRootBuildFileSource(pomXml)
                 .withProjectResource("src/main/resources/META-INF/persistence.xml", persistenceXml)
-                .addRegistrar(new PersistenceXmlProjectResourceRegistrar())
                 .build();
 
         sut.apply(projectContext);
diff --git a/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jpa/actions/PersistenceXmlToSpringBootApplicationPropertiesActionTest.java b/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jpa/actions/PersistenceXmlToSpringBootApplicationPropertiesActionTest.java
index c2b083edc..33206fc31 100644
--- a/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jpa/actions/PersistenceXmlToSpringBootApplicationPropertiesActionTest.java
+++ b/components/sbm-recipes-jee-to-boot/src/test/java/org/springframework/sbm/jee/jpa/actions/PersistenceXmlToSpringBootApplicationPropertiesActionTest.java
@@ -76,7 +76,6 @@ void whenNoProjectContextExistThenExceptionThrown() {
                     .withMavenRootBuildFileSource(parentPom)
                     .withProjectResource("pom1/pom.xml", pom1)
                     .withProjectResource("pom2/pom.xml", pom2)
-                    .addRegistrar(new PersistenceXmlProjectResourceRegistrar())
                     .build();
 
             MigratePersistenceXmlToApplicationPropertiesAction sut = new MigratePersistenceXmlToApplicationPropertiesAction();
@@ -142,7 +141,6 @@ void whenNoApplicationPropertiesExistThenPropertiesShouldBeCreated() {
                     .withProjectResource("pom1/pom.xml", pom1)
                     .withProjectResource("pom2/pom.xml", pom2)
                     .withProjectResource("pom2/src/main/resources/META-INF/persistence.xml", persistenceXml)
-                    .addRegistrar(new PersistenceXmlProjectResourceRegistrar())
                     .build();
 
             assertThat(projectContext.search(new SpringBootApplicationPropertiesResourceListFilter())).isEmpty();
@@ -176,7 +174,7 @@ void migrateJpaToSpringBoot() {
                                     </persistence-unit>
                                 </persistence>
                                 """)
-                        .addRegistrar(new PersistenceXmlProjectResourceRegistrar()))
+                )
                 .actionUnderTest(new MigratePersistenceXmlToApplicationPropertiesAction())
                 .verify(context -> {
                     List<SpringBootApplicationProperties> applicationProperties = context.search(new SpringBootApplicationPropertiesResourceListFilter());