GH-611 - Additional modulith packages are now considered for application classes.

Author: odrotbohm

spring-modulith-core/src/main/java/org/springframework/modulith/core/ApplicationModules.java

@@ -254,7 +254,6 @@ public Set<ApplicationModule> getSharedModules() {
 	 * Returns whether the given {@link JavaClass} is contained within the {@link ApplicationModules}.
 	 *
 	 * @param type must not be {@literal null}.
-	 * @return
 	 */
 	public boolean contains(JavaClass type) {
 
@@ -264,6 +263,18 @@ public boolean contains(JavaClass type) {
 				.anyMatch(module -> module.contains(type));
 	}
 
+	/**
+	 * Returns whether the given {@link Class} is contained within the {@link ApplicationModules}.
+	 *
+	 * @param type must not be {@literal null}.
+	 */
+	public boolean contains(Class<?> type) {
+
+		Assert.notNull(type, "Type must not be null!");
+
+		return allClasses.contain(type) && contains(allClasses.get(type));
+	}
+
 	/**
 	 * Returns whether the given type is contained in one of the root packages (not including sub-packages) of the
 	 * modules.

spring-modulith-runtime/pom.xml

@@ -40,6 +40,13 @@
 			<optional>true</optional>
 		</dependency>
 
+		<dependency>
+			<groupId>org.springframework.modulith</groupId>
+			<artifactId>spring-modulith-starter-test</artifactId>
+			<version>${project.version}</version>
+			<scope>test</scope>
+		</dependency>
+
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter-test</artifactId>

spring-modulith-runtime/src/main/java/org/springframework/modulith/runtime/ApplicationModulesRuntime.java

@@ -63,7 +63,9 @@ public ApplicationModules get() {
 	 * @return
 	 */
 	public boolean isApplicationClass(Class<?> type) {
-		return runtime.isApplicationClass(type);
+
+		return runtime.isApplicationClass(type)
+				|| modules.get().contains(type);
 	}
 
 	/**

spring-modulith-runtime/src/test/java/different/moduleB/ModuleBType.java

@@ -0,0 +1,21 @@
+/*
+ * Copyright 2024 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package different.moduleB;
+
+/**
+ * @author Oliver Drotbohm
+ */
+public class ModuleBType {}

spring-modulith-runtime/src/test/java/example/SampleApplication.java

@@ -0,0 +1,26 @@
+/*
+ * Copyright 2024 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package example;
+
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.modulith.Modulithic;
+
+/**
+ * @author Oliver Drotbohm
+ */
+@Modulithic(additionalPackages = "different")
+@SpringBootApplication
+public class SampleApplication {}

spring-modulith-runtime/src/test/java/example/moduleA/ModuleAType.java

@@ -0,0 +1,21 @@
+/*
+ * Copyright 2024 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package example.moduleA;
+
+/**
+ * @author Oliver Drotbohm
+ */
+public class ModuleAType {}

spring-modulith-runtime/src/test/java/org/springframework/modulith/runtime/ApplicationModulesRuntimeIntegrationTests.java

@@ -0,0 +1,52 @@
+/*
+ * Copyright 2024 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.modulith.runtime;
+
+import static org.assertj.core.api.Assertions.*;
+
+import different.moduleB.ModuleBType;
+import example.SampleApplication;
+import example.moduleA.ModuleAType;
+
+import java.util.stream.Stream;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.boot.SpringApplication;
+import org.springframework.modulith.core.ApplicationModules;
+import org.springframework.modulith.runtime.autoconfigure.TestSpringBootApplicationRuntime;
+import org.springframework.modulith.test.TestApplicationModules;
+
+/**
+ * Integration tests for {@link ApplicationModulesRuntime}.
+ *
+ * @author Oliver Drotbohm
+ */
+public class ApplicationModulesRuntimeIntegrationTests {
+
+	ApplicationModules modules = TestApplicationModules.of(SampleApplication.class);
+
+	@Test // GH-611
+	void detectsTypeInAdditionalPackageAsApplicationType() {
+
+		var context = SpringApplication.run(SampleApplication.class);
+		var applicationRuntime = new TestSpringBootApplicationRuntime(context);
+
+		var runtime = new ApplicationModulesRuntime(() -> modules, applicationRuntime);
+
+		Stream.of(ModuleAType.class, ModuleBType.class)
+				.forEach(it -> assertThat(runtime.isApplicationClass(it)).isTrue());
+	}
+}

spring-modulith-runtime/src/test/java/org/springframework/modulith/runtime/autoconfigure/TestSpringBootApplicationRuntime.java

@@ -0,0 +1,28 @@
+/*
+ * Copyright 2024 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.modulith.runtime.autoconfigure;
+
+import org.springframework.context.ApplicationContext;
+
+/**
+ * @author Oliver Drotbohm
+ */
+public class TestSpringBootApplicationRuntime extends SpringBootApplicationRuntime {
+
+	public TestSpringBootApplicationRuntime(ApplicationContext context) {
+		super(context);
+	}
+}

spring-modulith-runtime/src/test/resources/application.properties

@@ -0,0 +1 @@
+spring.main.banner-mode=OFF

spring-modulith-runtime/src/test/resources/logback.xml

@@ -9,6 +9,6 @@
 	</root>
 
 	<logger name="org.springframework.modulith" level="DEBUG"/>
-	<logger name="example" level="INFO" />
+	<logger name="example" level="WARN" />
 
 </configuration>