Return type

ReturnTypeRulesTest.java

import static com.tngtech.archunit.lang.syntax.ArchRuleDefinition.methods;

import org.springframework.http.ResponseEntity;

import com.tngtech.archunit.core.importer.ImportOption;
import com.tngtech.archunit.junit.AnalyzeClasses;
import com.tngtech.archunit.junit.ArchTest;
import com.tngtech.archunit.lang.ArchRule;

@AnalyzeClasses(packages = "com.example.archunit", importOptions = ImportOption.DoNotIncludeTests.class)
class ReturnTypeRulesTest {

	@ArchTest
	static final ArchRule methods_should_return_response_entity =
	     methods().that().arePublic()
	              .and().areDeclaredInClassesThat()
	              .resideInAPackage("com.example.controller")
	              .should().haveRawReturnType(ResponseEntity.class)
	              .because("Controller endpoints should return a ResponseEntity object");
}