Fix unit tests

Author: BoykoAlex

headless-services/spring-boot-language-server/src/main/java/org/springframework/ide/vscode/boot/java/data/GenAotQueryMethodImplProvider.java

@@ -102,7 +102,7 @@ public List<LocationLink> getLocationLinks(CancelChecker cancelToken, IJavaProje
 	}
 
 	private List<LocationLink> findImplLocations(IJavaProject project, GoToImplParams implParams) {
-		String genRepoFqn = implParams.repoFqName() + "Impl__Aot";
+		String genRepoFqn = implParams.repoFqName() + "Impl__AotRepository";
 		Path relativeGenSourcePath = Paths.get("%s.java".formatted(genRepoFqn.replace('.', '/')));
 		List<LocationLink> defs = findInSourceFolder(project, relativeGenSourcePath, genRepoFqn, implParams);
 		return defs.isEmpty() ? findInBuildFolder(project, relativeGenSourcePath, genRepoFqn, implParams) : defs;
@@ -149,7 +149,9 @@ public boolean visit(MethodDeclaration node) {
 	private List<LocationLink> findInSourceFolder(IJavaProject project, Path relativeGenSourcePath, String genRepoFqn, GoToImplParams params) {
 		for (File f : IClasspathUtil.getSourceFolders(project.getClasspath()).collect(Collectors.toSet())) {
 			Path genRepoSourcePath = f.toPath().resolve(relativeGenSourcePath);
-			return getLocationInGenFile(project, genRepoSourcePath, genRepoFqn, params);
+			if (Files.exists(relativeGenSourcePath)) {
+				return getLocationInGenFile(project, genRepoSourcePath, genRepoFqn, params);
+			}
 		}
 		return List.of();
 	}

headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/java/data/test/GenAotQueryMethodImplProviderTest.java

@@ -70,11 +70,11 @@ void nonAnnotatedMethod() throws Exception {
 
 		LocationLink ll = new LocationLink();
 		ll.setTargetUri(Paths.get(testProject.getLocationUri())
-				.resolve("target/spring-aot/main/sources/example/springdata/aot/UserRepositoryImpl__Aot.java").toUri()
+				.resolve("target/spring-aot/main/sources/example/springdata/aot/UserRepositoryImpl__AotRepository.java").toUri()
 				.toASCIIString());
 		ll.setOriginSelectionRange(new Range(new Position(43, 15), new Position(43, 61)));
-		ll.setTargetRange(new Range(new Position(144, 20), new Position(144, 66)));
-		ll.setTargetSelectionRange(new Range(new Position(144, 20), new Position(144, 66)));
+		ll.setTargetRange(new Range(new Position(140, 20), new Position(140, 66)));
+		ll.setTargetSelectionRange(new Range(new Position(140, 20), new Position(140, 66)));
 		editor.assertImplementationLinkTargets("findUserByLastnameStartingWith", List.of(ll));
 	}
 
@@ -86,11 +86,11 @@ void annotatedMethod() throws Exception {
 
 		LocationLink ll = new LocationLink();
 		ll.setTargetUri(Paths.get(testProject.getLocationUri())
-				.resolve("target/spring-aot/main/sources/example/springdata/aot/UserRepositoryImpl__Aot.java").toUri()
+				.resolve("target/spring-aot/main/sources/example/springdata/aot/UserRepositoryImpl__AotRepository.java").toUri()
 				.toASCIIString());
 		ll.setOriginSelectionRange(new Range(new Position(54, 15), new Position(54, 45)));
-		ll.setTargetRange(new Range(new Position(188, 20), new Position(188, 50)));
-		ll.setTargetSelectionRange(new Range(new Position(188, 20), new Position(188, 50)));
+		ll.setTargetRange(new Range(new Position(184, 20), new Position(184, 50)));
+		ll.setTargetSelectionRange(new Range(new Position(184, 20), new Position(184, 50)));
 		editor.assertImplementationLinkTargets("usersWithUsernamesStartingWith", List.of(ll));
 	}