Make RpcServer independent of ArchiveLogMessagesMethod

1 files changed, 3 insertions, 2 deletions

diff --git a/logserver/src/test/java/ai/vespa/logserver/protocol/RpcServerTest.java b/logserver/src/test/java/ai/vespa/logserver/protocol/ArchiveLogMessagesMethodTest.java
index 321c21915fc..a30df6bb050 100644
--- a/logserver/src/test/java/ai/vespa/logserver/protocol/RpcServerTest.java
+++ b/logserver/src/test/java/ai/vespa/logserver/protocol/ArchiveLogMessagesMethodTest.java
@@ -28,7 +28,7 @@ import static org.mockito.Mockito.verify;
 /**
  * @author bjorncs
  */
-public class RpcServerTest {
+public class ArchiveLogMessagesMethodTest {
 
     private static final LogMessage MESSAGE_1 =
             LogMessage.of(Instant.EPOCH.plus(1000, ChronoUnit.DAYS), "localhost", 12, 3456, "my-service", "my-component", LogLevel.ERROR, "My error message");
@@ -39,7 +39,8 @@ public class RpcServerTest {
     public void server_dispatches_log_messages_from_log_request() {
         List<LogMessage> messages = List.of(MESSAGE_1, MESSAGE_2);
         LogDispatcher logDispatcher = mock(LogDispatcher.class);
-        try (RpcServer server = new RpcServer(0, logDispatcher)) {
+        try (RpcServer server = new RpcServer(0)) {
+            server.addMethod(new ArchiveLogMessagesMethod(logDispatcher).methodDefinition());
             server.start();
             try (TestClient client = new TestClient(server.listenPort())) {
                 client.logMessages(messages);