Search in sources :

Example 1 with ClientOptions

use of io.confluent.ksql.api.client.ClientOptions in project ksql by confluentinc.

the class MigrationsUtilTest method shouldCreateNonTlsClientOptions.

@Test
public void shouldCreateNonTlsClientOptions() {
    // Given:
    final ClientOptions clientOptions = createClientOptions(NON_TLS_URL, "user", "pass", null, "", null, null, "", "foo", false, true, null);
    // Then:
    assertThat(clientOptions.isUseTls(), is(false));
    assertThat(clientOptions.getBasicAuthUsername(), is("user"));
    assertThat(clientOptions.getBasicAuthPassword(), is("pass"));
    assertThat(clientOptions.getTrustStore(), is(""));
    assertThat(clientOptions.getTrustStorePassword(), is(""));
    assertThat(clientOptions.getKeyStore(), is(""));
    assertThat(clientOptions.getKeyStorePassword(), is(""));
    assertThat(clientOptions.getKeyPassword(), is(""));
    assertThat(clientOptions.getKeyAlias(), is(""));
    assertThat(clientOptions.isUseAlpn(), is(false));
    assertThat(clientOptions.isVerifyHost(), is(true));
    assertThat(clientOptions.getRequestHeaders(), is(Collections.emptyMap()));
}
Also used : MigrationsUtil.createClientOptions(io.confluent.ksql.tools.migrations.util.MigrationsUtil.createClientOptions) ClientOptions(io.confluent.ksql.api.client.ClientOptions) Test(org.junit.Test)

Example 2 with ClientOptions

use of io.confluent.ksql.api.client.ClientOptions in project ksql by confluentinc.

the class MigrationsUtil method createClientOptions.

@VisibleForTesting
static // CHECKSTYLE_RULES.OFF: ParameterNumberCheck
ClientOptions createClientOptions(// CHECKSTYLE_RULES.ON: ParameterNumberCheck
final String ksqlServerUrl, final String username, final String password, final String sslTrustStoreLocation, final String sslTrustStorePassword, final String sslKeystoreLocation, final String sslKeystorePassword, final String sslKeyPassword, final String sslKeyAlias, final boolean useAlpn, final boolean verifyHost, final Map<String, String> requestHeaders) {
    final URL url;
    try {
        url = new URL(ksqlServerUrl);
    } catch (MalformedURLException e) {
        throw new MigrationException("Invalid ksql server URL: " + ksqlServerUrl);
    }
    final ClientOptions options = ClientOptions.create().setHost(url.getHost()).setPort(url.getPort());
    if (username != null || password != null) {
        options.setBasicAuthCredentials(username, password);
    }
    final boolean useTls = ksqlServerUrl.trim().toLowerCase().startsWith("https://");
    options.setUseTls(useTls);
    if (useTls) {
        options.setTrustStore(sslTrustStoreLocation);
        options.setTrustStorePassword(sslTrustStorePassword);
        options.setKeyStore(sslKeystoreLocation);
        options.setKeyStorePassword(sslKeystorePassword);
        options.setKeyPassword(sslKeyPassword);
        options.setKeyAlias(sslKeyAlias);
        options.setUseAlpn(useAlpn);
        options.setVerifyHost(verifyHost);
    }
    if (requestHeaders != null) {
        options.setRequestHeaders(requestHeaders);
    }
    return options;
}
Also used : MigrationException(io.confluent.ksql.tools.migrations.MigrationException) ClientOptions(io.confluent.ksql.api.client.ClientOptions) MalformedURLException(java.net.MalformedURLException) URL(java.net.URL) VisibleForTesting(com.google.common.annotations.VisibleForTesting)

Example 3 with ClientOptions

use of io.confluent.ksql.api.client.ClientOptions in project ksql by confluentinc.

the class MigrationsUtilTest method shouldCreateTlsClientOptions.

@Test
public void shouldCreateTlsClientOptions() {
    // Given:
    final Map<String, String> requestHeaders = ImmutableMap.of("h1", "v1", "h2", "v2");
    final ClientOptions clientOptions = createClientOptions(TLS_URL, "user", "pass", "abc", null, null, null, null, null, true, true, requestHeaders);
    // Then:
    assertThat(clientOptions.isUseTls(), is(true));
    assertThat(clientOptions.getBasicAuthUsername(), is("user"));
    assertThat(clientOptions.getBasicAuthPassword(), is("pass"));
    assertThat(clientOptions.getTrustStore(), is("abc"));
    assertThat(clientOptions.getTrustStorePassword(), is(""));
    assertThat(clientOptions.getKeyStore(), is(""));
    assertThat(clientOptions.getKeyStorePassword(), is(""));
    assertThat(clientOptions.getKeyPassword(), is(""));
    assertThat(clientOptions.getKeyAlias(), is(""));
    assertThat(clientOptions.isUseAlpn(), is(true));
    assertThat(clientOptions.isVerifyHost(), is(true));
    assertThat(clientOptions.getRequestHeaders(), is(requestHeaders));
}
Also used : MigrationsUtil.createClientOptions(io.confluent.ksql.tools.migrations.util.MigrationsUtil.createClientOptions) ClientOptions(io.confluent.ksql.api.client.ClientOptions) Test(org.junit.Test)

Aggregations

ClientOptions (io.confluent.ksql.api.client.ClientOptions)3 MigrationsUtil.createClientOptions (io.confluent.ksql.tools.migrations.util.MigrationsUtil.createClientOptions)2 Test (org.junit.Test)2 VisibleForTesting (com.google.common.annotations.VisibleForTesting)1 MigrationException (io.confluent.ksql.tools.migrations.MigrationException)1 MalformedURLException (java.net.MalformedURLException)1 URL (java.net.URL)1