Function TestInjectTools

aibridge/intercept/responses/reqpayload_internal_test.go:260–333

(t *testing.T)

}

func TestInjectTools(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name      string
		raw       []byte
		injected  []responses.ToolUnionParam
		wantNames []string
		wantErr   string
		wantSame  bool
	}{
		{
			name:      "appends to existing tools",
			raw:       []byte(`{"model":"gpt-4o","input":"hello","tools":[{"type":"function","name":"existing"}]}`),
			injected:  []responses.ToolUnionParam{injectedFunctionTool("injected")},
			wantNames: []string{"existing", "injected"},
		},
		{
			name:      "adds tools when none exist",
			raw:       []byte(`{"model":"gpt-4o","input":"hello"}`),
			injected:  []responses.ToolUnionParam{injectedFunctionTool("injected")},
			wantNames: []string{"injected"},
		},
		{
			name:      "adds to empty tools array",
			raw:       []byte(`{"model":"gpt-4o","input":"hello","tools":[]}`),
			injected:  []responses.ToolUnionParam{injectedFunctionTool("injected")},
			wantNames: []string{"injected"},
		},
		{
			name: "appends multiple injected tools",
			raw:  []byte(`{"model":"gpt-4o","input":"hello","tools":[{"type":"function","name":"existing"}]}`),
			injected: []responses.ToolUnionParam{
				injectedFunctionTool("injected-one"),
				injectedFunctionTool("injected-two"),
			},
			wantNames: []string{"existing", "injected-one", "injected-two"},
		},
		{
			name:     "empty injected tools is no op",
			raw:      []byte(`{"model":"gpt-4o","input":"hello","tools":[{"type":"function","name":"existing"}]}`),
			wantSame: true,
		},
		{
			name:     "errors on unsupported tools shape",
			raw:      []byte(`{"model":"gpt-4o","input":"hello","tools":"bad"}`),
			injected: []responses.ToolUnionParam{injectedFunctionTool("injected")},
			wantErr:  "failed to get existing tools: unsupported 'tools' type: String",
			wantSame: true,
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			p := mustPayload(t, tc.raw)
			updated, err := p.injectTools(tc.injected)