From aeba8efa396d8fad19046b606105221b6e11bc77 Mon Sep 17 00:00:00 2001
From: Piotr Findeisen <piotr.findeisen@gmail.com>
Date: Fri, 10 May 2024 16:14:31 +0200
Subject: [PATCH] Fix StartupStatus in tests

`StartupStatus.startupComplete` was set in production code, but not in
tests.
---
 .../main/java/io/trino/server/testing/TestingTrinoServer.java   | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/core/trino-main/src/main/java/io/trino/server/testing/TestingTrinoServer.java b/core/trino-main/src/main/java/io/trino/server/testing/TestingTrinoServer.java
index afcf08994043..a62bcd513021 100644
--- a/core/trino-main/src/main/java/io/trino/server/testing/TestingTrinoServer.java
+++ b/core/trino-main/src/main/java/io/trino/server/testing/TestingTrinoServer.java
@@ -83,6 +83,7 @@
 import io.trino.server.SessionPropertyDefaults;
 import io.trino.server.SessionSupplier;
 import io.trino.server.ShutdownAction;
+import io.trino.server.StartupStatus;
 import io.trino.server.security.CertificateAuthenticatorManager;
 import io.trino.server.security.ServerSecurityModule;
 import io.trino.spi.ErrorType;
@@ -424,6 +425,7 @@ private TestingTrinoServer(
         eventListeners.forEach(eventListenerManager::addEventListener);
 
         getFutureValue(injector.getInstance(Announcer.class).forceAnnounce());
+        injector.getInstance(StartupStatus.class).startupComplete();
 
         refreshNodes();
     }