Function TestStartRuntimeAPIWithContext

lambda/entry_test.go:17–33

(t *testing.T)

type ctxTestKey struct{}

func TestStartRuntimeAPIWithContext(t *testing.T) {
	server, _ := runtimeAPIServer("null", 1) // serve a single invoke, and then cause an internal error
	expected := "expected"
	actual := "unexpected"

	os.Setenv("AWS_LAMBDA_RUNTIME_API", strings.Split(server.URL, "://")[1])
	defer os.Unsetenv("AWS_LAMBDA_RUNTIME_API")
	logFatalf = func(format string, v ...interface{}) {}
	defer func() { logFatalf = log.Fatalf }()

	StartWithContext(context.WithValue(context.Background(), ctxTestKey{}, expected), func(ctx context.Context) error {
		actual, _ = ctx.Value(ctxTestKey{}).(string)
		return nil
	})

	assert.Equal(t, expected, actual)
}