Also update TestMessageFactory

Graylog2 · Mar 12, 2024 · 96d4829 · 96d4829
1 parent 1e31d32
commit 96d4829
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/graylog2-server/src/test/java/org/graylog2/plugin/TestMessageFactory.java b/graylog2-server/src/test/java/org/graylog2/plugin/TestMessageFactory.java
@@ -16,6 +16,9 @@
  */
 package org.graylog2.plugin;
 
+import org.graylog2.indexer.IndexSet;
+import org.graylog2.indexer.messages.IndexingResultCallback;
+import org.jetbrains.annotations.Nullable;
 import org.joda.time.DateTime;
 
 import java.util.Map;
@@ -39,4 +42,9 @@ public Message createMessage(Map<String, Object> fields) {
     public Message createMessage(String id, Map<String, Object> newFields) {
         return new Message(id, newFields);
     }
+
+    @Override
+    public SystemMessage createSystemMessage(IndexSet indexSet, Map<String, Object> fields, @Nullable IndexingResultCallback resultCallback) {
+        return new SystemMessage(indexSet, fields, resultCallback);
+    }
 }