runtime_test.ml

open Test_common

module Runtime = struct
  include
    Lambda_runtime__.Runtime.Make
      (Lambda_runtime__.Json.Id)
      (Lambda_runtime__.Json.Id)

  type event = Lambda_runtime__.Json.Id.t
  type response = Lambda_runtime__.Json.Id.t
end

let request = `String "test"
let test_runtime = test_runtime_generic (module Runtime) request

exception User_code_exception

let suite =
  [ ( "successful handler invocation"
    , `Quick
    , test_runtime
        (fun _event _ctx -> Ok (`String "Hello"))
        (fun output ->
          match output with
          | Ok result ->
            Alcotest.check
              yojson
              "runtime invoke output"
              (`String "Hello")
              result
          | Error e -> Alcotest.fail e) )
  ; ( "failed handler invocation"
    , `Quick
    , test_runtime
        (fun _event _ctx -> Error "I failed")
        (fun output ->
          match output with
          | Ok result ->
            let result_str = Yojson.Safe.to_string result in
            Alcotest.fail
              (Printf.sprintf
                 "Expected to get an error but the call succeeded with: %s"
                 result_str)
          | Error e ->
            Alcotest.(check string "Runtime invoke error" "I failed" e)) )
  ; ( "simple asynchronous handler invocation"
    , `Quick
    , test_runtime
        (fun _event _ctx -> Ok (`String "Hello"))
        (fun output ->
          match output with
          | Ok result ->
            Alcotest.check
              yojson
              "runtime invoke output"
              (`String "Hello")
              result
          | Error e -> Alcotest.fail e) )
  ; ( "failed handler invocation"
    , `Quick
    , test_runtime
        (fun _event _ctx -> raise User_code_exception)
        (fun output ->
          match output with
          | Ok result ->
            let result_str = Yojson.Safe.to_string result in
            Alcotest.fail
              (Printf.sprintf
                 "Expected to get an error but the call succeeded with: %s"
                 result_str)
          | Error e ->
            let expected =
              "Handler raised: Dune__exe__Runtime_test.User_code_exception\n"
            in
            Alcotest.(
              check
                string
                "Runtime invoke error"
                expected
                (String.sub e 0 (String.length expected)))) )
  ]