/*
 * Copyright 2015-2019 The OpenZipkin Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */
package zipkin2.server.internal.elasticsearch;

import com.linecorp.armeria.client.Endpoint;
import org.junit.jupiter.api.Test;

import static com.linecorp.armeria.common.SessionProtocol.HTTP;
import static com.linecorp.armeria.common.SessionProtocol.HTTPS;
import static org.assertj.core.api.Assertions.assertThat;

class InitialEndpointSupplierTest {

  @Test void defaultIsLocalhost9200RegardlessOfSessionProtocol() {
    assertThat(new InitialEndpointSupplier(HTTP, null).get())
      .isEqualTo(Endpoint.of("localhost", 9200))
      .isEqualTo(new InitialEndpointSupplier(HTTPS, null).get());
  }

  @Test void usesNaturalHttpPortsWhenUrls() {
    assertThat(new InitialEndpointSupplier(HTTP, "http://localhost").get())
      .isEqualTo(Endpoint.of("localhost", 80));
    assertThat(new InitialEndpointSupplier(HTTPS, "https://localhost").get())
      .isEqualTo(Endpoint.of("localhost", 443));
  }

  @Test void defaultsPlainHostsToPort9200() {
    assertThat(new InitialEndpointSupplier(HTTP, "localhost").get())
      .isEqualTo(Endpoint.of("localhost", 9200));
    assertThat(new InitialEndpointSupplier(HTTPS, "localhost").get())
      .isEqualTo(Endpoint.of("localhost", 443));
  }

  /** This helps ensure old setups don't break (provided they have http port 9200 open) */
  @Test public void coersesPort9300To9200() {
    assertThat(new InitialEndpointSupplier(HTTP, "localhost:9300").get())
      .isEqualTo(Endpoint.of("localhost", 9200));
  }

  @Test void parsesListOfLocalhosts() {
    String hostList = "localhost:9201,localhost:9202";
    assertThat(new InitialEndpointSupplier(HTTP, hostList).get().endpoints())
      .containsExactly(Endpoint.of("localhost", 9201), Endpoint.of("localhost", 9202))
      .containsExactlyElementsOf(new InitialEndpointSupplier(HTTPS, hostList).get().endpoints());
  }
}