Function TestInvokeWithContext

lambda/rpc_function_test.go:66–89

(t *testing.T)

}

func TestInvokeWithContext(t *testing.T) {
	key := struct{}{}
	srv := NewFunction(&handlerOptions{
		handlerFunc: func(ctx context.Context, _ []byte) (io.Reader, error) {
			assert.Equal(t, "dummy", ctx.Value(key))
			if deadline, ok := ctx.Deadline(); ok {
				return strings.NewReader(strconv.FormatInt(deadline.UnixNano(), 10)), nil
			}
			return nil, errors.New("!?!?!?!?!")
		},
		baseContext: context.WithValue(context.Background(), key, "dummy"),
	})
	deadline := time.Now()
	var response messages.InvokeResponse
	err := srv.Invoke(&messages.InvokeRequest{
		Deadline: messages.InvokeRequest_Timestamp{
			Seconds: deadline.Unix(),
			Nanos:   int64(deadline.Nanosecond()),
		}}, &response)
	assert.NoError(t, err)
	var responseValue int64
	assert.NoError(t, json.Unmarshal(response.Payload, &responseValue))
	assert.Equal(t, deadline.UnixNano(), responseValue)
}

type CustomError struct{}