1 files changed, 7 insertions, 0 deletions
diff --git a/integration/intellij/src/test/java/ai/vespa/intellij/findUsages/UsagesTester.java b/integration/intellij/src/test/java/ai/vespa/intellij/findUsages/UsagesTester.java
index f48ef7f107a..319240b9ceb 100644
--- a/integration/intellij/src/test/java/ai/vespa/intellij/findUsages/UsagesTester.java
+++ b/integration/intellij/src/test/java/ai/vespa/intellij/findUsages/UsagesTester.java
@@ -36,6 +36,13 @@ class UsagesTester {
         return usageProcessor.usages;
     }
 
+    List<UsageInfo> assertProfileUsages(int expectedUsages, String profileName) {
+        var profile = schema.rankProfiles().get(profileName).definition();
+        findUsages(profile);
+        assertEquals(expectedUsages, usageProcessor.usages.size());
+        return usageProcessor.usages;
+    }
+
     /** Finds the function referred at the given character offset in the given profile. */
     UsageInfo findFunctionDefinition(String profileName, int offset) {
         PsiElement referringElement = schema.rankProfiles().get(profileName).definition().findElementAt(offset);