diff --git a/ksql-cli/src/test/java/io/confluent/ksql/cli/CliTest.java b/ksql-cli/src/test/java/io/confluent/ksql/cli/CliTest.java index a88666a56484..a7ea1d0c4aaa 100644 --- a/ksql-cli/src/test/java/io/confluent/ksql/cli/CliTest.java +++ b/ksql-cli/src/test/java/io/confluent/ksql/cli/CliTest.java @@ -58,6 +58,7 @@ import io.confluent.ksql.rest.server.TestKsqlRestApp; import io.confluent.ksql.rest.server.computation.CommandId; import io.confluent.ksql.rest.server.resources.Errors; +import io.confluent.ksql.rest.server.resources.RootDocument; import io.confluent.ksql.schema.ksql.KsqlSchema; import io.confluent.ksql.test.util.EmbeddedSingleNodeKafkaCluster; import io.confluent.ksql.test.util.KsqlIdentifierTestUtil; @@ -985,7 +986,7 @@ public void shouldResetStateWhenServerChanges() throws Exception { givenCommandSequenceNumber(mockRestClient, 5L); givenRequestPipelining("ON"); when(mockRestClient.makeRootRequest()).thenReturn( - RestResponse.successful(new ServerInfo("version", "clusterId", "serviceId"))); + RestResponse.successful(new RootDocument())); // When: runCliSpecificCommand("server foo"); diff --git a/ksql-cli/src/test/java/io/confluent/ksql/cli/console/cmd/RemoteServerSpecificCommandTest.java b/ksql-cli/src/test/java/io/confluent/ksql/cli/console/cmd/RemoteServerSpecificCommandTest.java index 2e7bd3811fc1..b0e9e488e9ba 100644 --- a/ksql-cli/src/test/java/io/confluent/ksql/cli/console/cmd/RemoteServerSpecificCommandTest.java +++ b/ksql-cli/src/test/java/io/confluent/ksql/cli/console/cmd/RemoteServerSpecificCommandTest.java @@ -30,6 +30,7 @@ import io.confluent.ksql.rest.client.exception.KsqlRestClientException; import io.confluent.ksql.rest.entity.ServerInfo; import io.confluent.ksql.rest.server.resources.Errors; +import io.confluent.ksql.rest.server.resources.RootDocument; import io.confluent.ksql.util.Event; import java.io.PrintWriter; import java.io.StringWriter; @@ -47,8 +48,7 @@ public class RemoteServerSpecificCommandTest { private static final String INITIAL_SERVER_ADDRESS = "http://192.168.0.1:8080"; private static final String VALID_SERVER_ADDRESS = "http://localhost:8088"; - private static final ServerInfo SERVER_INFO = - new ServerInfo("1.x", "myClusterId", "myKsqlServiceId"); + private static final RootDocument ROOT_DOCUMENT = new RootDocument(); @Mock private KsqlRestClient restClient; @@ -65,7 +65,7 @@ public void setUp() throws Exception { terminal = new PrintWriter(out); command = RemoteServerSpecificCommand.create(restClient, resetCliForNewServer); - when(restClient.makeRootRequest()).thenReturn(RestResponse.successful(SERVER_INFO)); + when(restClient.makeRootRequest()).thenReturn(RestResponse.successful(ROOT_DOCUMENT)); when(restClient.getServerAddress()).thenReturn(new URI(INITIAL_SERVER_ADDRESS)); } diff --git a/ksql-rest-app/src/main/java/io/confluent/ksql/rest/client/KsqlRestClient.java b/ksql-rest-app/src/main/java/io/confluent/ksql/rest/client/KsqlRestClient.java index 3de659eea633..103e63cbf692 100644 --- a/ksql-rest-app/src/main/java/io/confluent/ksql/rest/client/KsqlRestClient.java +++ b/ksql-rest-app/src/main/java/io/confluent/ksql/rest/client/KsqlRestClient.java @@ -30,6 +30,7 @@ import io.confluent.ksql.rest.entity.ServerInfo; import io.confluent.ksql.rest.entity.StreamedRow; import io.confluent.ksql.rest.server.resources.Errors; +import io.confluent.ksql.rest.server.resources.RootDocument; import io.confluent.ksql.rest.ssl.DefaultSslClientConfigurer; import io.confluent.ksql.rest.ssl.SslClientConfigurer; import io.confluent.rest.validation.JacksonMessageBodyProvider; @@ -154,8 +155,8 @@ public void setServerAddress(final String serverAddress) { this.serverAddresses = parseServerAddresses(serverAddress); } - public RestResponse makeRootRequest() { - return getServerInfo(); + public RestResponse makeRootRequest() { + return getRequest("/", RootDocument.class); } public RestResponse getServerInfo() {