Function TestSendGenericRequestLoop

pkg/controller/rtctrl/runtime_test.go:146–236

(t *testing.T)

}

func TestSendGenericRequestLoop(t *testing.T) {
	id := "runtime-id"
	cmid := "commit-xxx"
	reqid := utilID.GetRequestID()
	rtMap := NewRuntimeManager(getFuncletNode(), &RuntimeManagerParameters{MaxRuntimeIdle: 10, MaxRunnerDefunct: 30})
	rtParams := &NewRuntimeParameters{
		RuntimeID:               id,
		ConcurrentMode:          false,
		StreamMode:              false,
		WaitRuntimeAliveTimeout: 3,
		Resource:                &api.Resource{},
	}
	rt := rtMap.NewRuntime(rtParams)
	req, resp := buildRequestResponse(id, cmid, reqid)
	conn, _, ok := setupHijackConn(resp, req)
	if !ok {
		t.Error("setup hijack connect failed")
		return
	}
	rt.SetState(RuntimeStateCold)
	params := &startRuntimeParams{
		commitID:   cmid,
		hostIP:     "127.0.0.1",
		conn:       conn,
		warmNotify: make(chan struct{}),
		urlParams:  req.URL.Query(),
	}
	if err := rt.initRuntime(params); err != nil {
		t.Errorf("init runtime failed: %s", err)
		return
	}

	now := time.Now().UnixNano()
	ver := "1"
	reqinfo := RequestInfo{
		RequestID:         reqid,
		Runtime:           rt,
		InvokeStartTimeNS: now,
		InvokeStartTimeMS: now / int64(time.Millisecond),
		SyncChannel:       make(chan struct{}, 1),
		TimeoutChannel:    make(chan struct{}, 1),
		Input: &InvocationInput{
			Runtime:   rt,
			RequestID: reqid,
			Configuration: &api.FunctionConfiguration{
				CommitID:              &cmid,
				FunctionConfiguration: lambda.FunctionConfiguration{Version: &ver},
			},
			User: &api.User{
				ID: "xxx",
			},
			WithStreamMode: false,
			Request: &api.InvokeProxyRequest{
				Headers: make(map[string]string, 0),
				Body:    []byte("{}"),
			},
			Response:      api.NewInvokeProxyResponse(),
			EnableMetrics: true,