Function TestHandleSubagentCall_Failure

internal/gemini/gemini_planner_test.go:355–394

(t *testing.T)

}

func TestHandleSubagentCall_Failure(t *testing.T) {
	mockExec := &mockExecutor{
		execFunc: func(ctx context.Context, conversationID string, execID string, start *proto.AgentStart, o agent.OutputHandler) (proto.State, error) {
			return proto.State_STATE_UNSPECIFIED, fmt.Errorf("subagent execution failed")
		},
	}

	p := &geminiPlannerAgent{}
	fc := &genai.FunctionCall{
		Name: "test-subagent",
		Args: map[string]any{
			"history": "Previous history summary",
			"prompt":  "Current user prompt",
		},
	}

	var handlerOutput *proto.AgentOutputs
	handler := func(outgoing *proto.AgentOutputs) error {
		handlerOutput = outgoing
		return nil
	}

	err := p.handleSubagentCall(context.Background(), "test-conv", fc, nil, nil, mockExec, handler)
	if err != nil {
		t.Fatalf("handleSubagentCall failed: %v", err)
	}

	msg := handlerOutput.Messages[0]
	tr := msg.Content.GetToolResult()
	fr := tr.GetFunctionResult()
	resp := fr.GetResponse()
	resultVal := resp.Fields["result"].GetStringValue()
	if resultVal != "sub agent failed" {
		t.Errorf("expected 'sub agent failed', got %s", resultVal)
	}
	errVal := resp.Fields["error"].GetStringValue()
	if !strings.Contains(errVal, "subagent execution failed") {
		t.Errorf("expected error to contain 'subagent execution failed', got %s", errVal)
	}
}

func TestHandleSubagentCall_Panic(t *testing.T) {
	mockExec := &mockExecutor{