Function TestRuntimeAPIContextPlumbing

lambda/invoke_loop_test.go:198–278

(t *testing.T)

}

func TestRuntimeAPIContextPlumbing(t *testing.T) {
	handler := NewHandler(func(ctx context.Context) (interface{}, error) {
		lc, _ := lambdacontext.FromContext(ctx)
		deadline, _ := ctx.Deadline()
		return struct {
			Context    *lambdacontext.LambdaContext
			TraceID    string
			EnvTraceID string
			Deadline   int64
		}{
			Context:    lc,
			TraceID:    ctx.Value("x-amzn-trace-id").(string),
			EnvTraceID: os.Getenv("_X_AMZN_TRACE_ID"),
			Deadline:   deadline.UnixNano() / nsPerMS,
		}, nil
	})

	metadata2 := defaultInvokeMetadata()
	metadata2.tenantID = "some-tenant-id"
	ts, record := runtimeAPIServer(``, 2, defaultInvokeMetadata(), metadata2)
	defer ts.Close()

	endpoint := strings.Split(ts.URL, "://")[1]
	expectedError := fmt.Sprintf("failed to GET http://%s/2018-06-01/runtime/invocation/next: got unexpected status code: 410", endpoint)
	assert.EqualError(t, startRuntimeAPILoop(endpoint, handler), expectedError)

	expected1 := `
	{
		"Context": {
			"AwsRequestID": "dummyid",
			"InvokedFunctionArn": "dummyarn",
			"Identity": {
				"CognitoIdentityID": "dummyident",
				"CognitoIdentityPoolID": "dummypool"
			},
			"ClientContext": {
				"Client": {
					"installation_id": "dummyinstallid",
					"app_title": "dummytitle",
					"app_version_code": "dummycode",
					"app_package_name": "dummyname"
				},
				"env": null,
				"custom": null
			}
		},
		"TraceID": "its-xray-time",
		"EnvTraceID": "its-xray-time",
		"Deadline": 22
	}
	`
	expected2 := `
	{
		"Context": {
			"AwsRequestID": "dummyid",
			"InvokedFunctionArn": "dummyarn",
			"TenantID": "some-tenant-id",
			"Identity": {