ExampleLambda3.test.scala

package org.encalmo.lambda.example

import org.encalmo.utils.JsonUtils.*
import org.encalmo.lambda.ApiGatewayRequest
import org.encalmo.lambda.ApiGatewayResponse
import org.encalmo.lambda.SqsEvent
import java.util.UUID
import org.encalmo.aws.AwsClient

class ExampleLambda3Spec extends TestSuite {

  val environment = Map("LAMBDA_RUNTIME_DEBUG_MODE" -> "ON")

  test("ExampleLambda3 should respond to generic event") {

    val input = """|{
                   |  "functionName": "Greeting",
                   |  "functionInput": {
                   |     "name": "Wroclaw"
                   |  }
                   |}""".stripMargin

    val output = new ExampleLambda3().test(input, environment)

    assertEquals(output, """{"greeting":"Hello Wroclaw!"}""")
  }

  test("ExampleLambda3 should respond to GET /hello/{name}") {

    val input = ApiGatewayRequest("GET", "/hello/World").writeAsString
    val output = new ExampleLambda3().test(input, environment)

    assertEquals(output.readAs[ApiGatewayResponse].body, "Hello World!")
  }

  test("ExampleLambda3 should respond to POST /hello") {

    val input = ApiGatewayRequest("POST", "/hello", "World").writeAsString
    val output = new ExampleLambda3().test(input, environment)

    assertEquals(output.readAs[ApiGatewayResponse].body, "Hello World!")
  }

  test("ExampleLambda3 should process SQS events") {

    val awsClientStub = AwsClient.newStatefulTestingStub()
    val targetSqsQueueUrl = "/foobar"

    val input = SqsEvent(Records =
      Seq(
        SqsEvent.Record(
          body = """|{
                     |  "functionName": "SendTargetEvent",
                     |  "functionInput": {
                     |     "name": "Wroclaw"
                     |  }
                     |}""".stripMargin,
          messageId = UUID.randomUUID().toString,
          attributes = Map.empty,
          eventSource = "",
          eventSourceARN = ""
        )
      )
    ).writeAsString

    new ExampleLambda3(Some(awsClientStub))
      .test(
        input,
        environment + ("TARGET_SQS_QUEUE_URL" -> targetSqsQueueUrl)
      )

    val targetEvent = awsClientStub.sqs.poll("/foobar")
    assert(targetEvent.isDefined)
    assertEquals(targetEvent.get._1, "Hello Wroclaw!")
  }

}