Function TestBinaryResponseDoesNotLeakResources

lambda/invoke_loop_test.go:330–353

(t *testing.T)

}

func TestBinaryResponseDoesNotLeakResources(t *testing.T) {
	numResponses := 3
	responses := make([]*readCloser, numResponses)
	for i := 0; i < numResponses; i++ {
		responses[i] = &readCloser{closed: false, reader: strings.NewReader(fmt.Sprintf("hello %d", i))}
	}
	timesCalled := 0
	handler := NewHandler(func() (res io.Reader, _ error) {
		res = responses[timesCalled]
		timesCalled++
		return
	})

	ts, record := runtimeAPIServer(`{}`, numResponses)
	defer ts.Close()
	endpoint := strings.Split(ts.URL, "://")[1]
	_ = startRuntimeAPILoop(endpoint, handler)

	for i := 0; i < numResponses; i++ {
		assert.Equal(t, contentTypeBytes, record.contentTypes[i])
		assert.Equal(t, fmt.Sprintf("hello %d", i), string(record.responses[i]))
		assert.True(t, responses[i].closed)
	}
}

func TestContextDeserializationErrors(t *testing.T) {
	badClientContext := defaultInvokeMetadata()