Function TestHandler_SendPrompt_WithClientAPIKey

server/handler_test.go:193–213

(t *testing.T)

}

func TestHandler_SendPrompt_WithClientAPIKey(t *testing.T) {
	logger := zap.NewNop()

	llmClient := &mockLLMClient{}
	llmClient.On("SendPrompt", mock.Anything, "Hello", []models.Message{}, 0).Return("Using your key!", nil)
	llmClient.On("GetModelName").Return("gpt-4o")

	mgr := &mockLLMManager{}
	mgr.On("CreateClientWithKey", "OPENAI", "", "sk-user-key-123").Return(llmClient, nil)

	handler := NewHandler(mgr, nil, logger, "OPENAI", "")

	resp, err := handler.SendPrompt(context.Background(), &pb.SendPromptRequest{
		Prompt:       "Hello",
		ClientApiKey: "sk-user-key-123",
	})
	assert.NoError(t, err)
	assert.Equal(t, "Using your key!", resp.Response)
	assert.Equal(t, "gpt-4o", resp.Model)
	mgr.AssertCalled(t, "CreateClientWithKey", "OPENAI", "", "sk-user-key-123")
}

func TestHandler_ListSessions(t *testing.T) {
	logger := zap.NewNop()