diff --git a/headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/java/events/test/EventsReferencesProviderTest.java b/headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/java/events/test/EventsReferencesProviderTest.java index 3d4db88724..4ed9a7cc7b 100644 --- a/headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/java/events/test/EventsReferencesProviderTest.java +++ b/headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/java/events/test/EventsReferencesProviderTest.java @@ -84,18 +84,20 @@ public class CustomEventListener { }""", tempJavaDocUri); List references = editor.getReferences(); - assertEquals(2, references.size()); + assertEquals(3, references.size()); String expectedDefinitionUri1 = directory.toPath().resolve("src/main/java/com/example/events/demo/CustomEventPublisher.java").toUri().toString(); Location expectedLocation1 = new Location(expectedDefinitionUri1, new Range(new Position(15, 2), new Position(15, 48))); - assertTrue(references.contains(expectedLocation1)); // from type hierarchy of specialzed custom event String expectedDefinitionUri2 = directory.toPath().resolve("src/main/java/com/example/events/demo/SpecializedCustomEventPublisher.java").toUri().toString(); Location expectedLocation2 = new Location(expectedDefinitionUri2, new Range(new Position(15, 2), new Position(15, 59))); - assertTrue(references.contains(expectedLocation2)); + + String expectedDefinitionUri3 = directory.toPath().resolve("src/main/java/com/example/events/demo/CustomEventPublisherWithAdditionalElements.java").toUri().toString(); + Location expectedLocation3 = new Location(expectedDefinitionUri3, new Range(new Position(17, 2), new Position(17, 48))); + assertTrue(references.contains(expectedLocation3)); } @Test