datahub/datahub-frontend/test/utils/CustomHttpClientFactoryTest.java

package utils;

import static org.junit.jupiter.api.Assertions.*;

import java.io.File;
import java.io.IOException;
import java.net.http.HttpClient;
import java.nio.file.Path;
import java.util.List;
import javax.net.ssl.SSLContext;
import org.apache.http.impl.client.CloseableHttpClient;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.io.TempDir;

class CustomHttpClientFactoryTest {

  @TempDir Path tempDir;

  private static final String TRUSTSTORE_PASSWORD = "testpassword";
  private static final String TRUSTSTORE_TYPE = "PKCS12";

  // Helper: Generate a temp PKCS12 truststore using keytool
  Path generateTempTruststore() throws IOException, InterruptedException {
    Path truststorePath = tempDir.resolve("test-truststore-" + System.nanoTime() + ".p12");
    String keytoolPath = System.getenv("KEYTOOL_PATH");
    if (keytoolPath == null) {
      keytoolPath = System.getProperty("java.home") + "/bin/keytool";
    }
    File keytoolFile = new File(keytoolPath);
    if (!keytoolFile.exists()) {
      keytoolPath = "keytool"; // fallback to system path
    }
    List<String> command =
        List.of(
            keytoolPath,
            "-genkeypair",
            "-alias",
            "testcert",
            "-keyalg",
            "RSA",
            "-keysize",
            "2048",
            "-storetype",
            "PKCS12",
            "-keystore",
            truststorePath.toString(),
            "-validity",
            "3650",
            "-storepass",
            TRUSTSTORE_PASSWORD,
            "-dname",
            "CN=Test, OU=Dev, O=Example, L=Test, S=Test, C=US");
    Process process = new ProcessBuilder(command).redirectErrorStream(true).start();
    String output = new String(process.getInputStream().readAllBytes());
    int exitCode = process.waitFor();
    if (exitCode != 0)
      throw new RuntimeException(
          "Could not generate truststore, exit code: " + exitCode + ", output: " + output);
    return truststorePath;
  }

  @Test
  void testCreateSslContextWithValidTruststore() throws Exception {
    Path truststorePath = generateTempTruststore();
    SSLContext context =
        CustomHttpClientFactory.createSslContext(
            truststorePath.toString(), TRUSTSTORE_PASSWORD, TRUSTSTORE_TYPE);
    assertNotNull(context);
    assertEquals("TLS", context.getProtocol());
  }

  @Test
  void testCreateSslContextWithInvalidTruststoreThrows() {
    assertThrows(
        Exception.class,
        () ->
            CustomHttpClientFactory.createSslContext(
                "doesnotexist.p12", "wrongpassword", TRUSTSTORE_TYPE));
  }

  @Test
  void testGetJavaHttpClientWithValidTruststore() throws Exception {
    Path truststorePath = generateTempTruststore();
    HttpClient client =
        CustomHttpClientFactory.getJavaHttpClient(
            truststorePath.toString(), TRUSTSTORE_PASSWORD, TRUSTSTORE_TYPE);
    assertNotNull(client);
  }

  @Test
  void testGetJavaHttpClientWithInvalidTruststoreFallsBack() {
    HttpClient client =
        CustomHttpClientFactory.getJavaHttpClient(
            "doesnotexist.p12", "wrongpassword", TRUSTSTORE_TYPE);
    assertNotNull(client);
  }

  @Test
  void testGetApacheHttpClientWithValidTruststore() throws Exception {
    Path truststorePath = generateTempTruststore();
    CloseableHttpClient client =
        CustomHttpClientFactory.getApacheHttpClient(
            truststorePath.toString(), TRUSTSTORE_PASSWORD, TRUSTSTORE_TYPE);
    assertNotNull(client);
  }

  @Test
  void testGetApacheHttpClientWithInvalidTruststoreFallsBack() {
    CloseableHttpClient client =
        CustomHttpClientFactory.getApacheHttpClient(
            "doesnotexist.p12", "wrongpassword", TRUSTSTORE_TYPE);
    assertNotNull(client);
  }
}