(t *testing.T)
| 171 | } |
| 172 | |
| 173 | func TestHandler_SendPrompt_Success(t *testing.T) { |
| 174 | logger := zap.NewNop() |
| 175 | |
| 176 | llmClient := &mockLLMClient{} |
| 177 | llmClient.On("SendPrompt", mock.Anything, "Hello", []models.Message{}, 0).Return("Hi there!", nil) |
| 178 | llmClient.On("GetModelName").Return("gpt-4") |
| 179 | |
| 180 | mgr := &mockLLMManager{} |
| 181 | mgr.On("GetClient", "openai", "").Return(llmClient, nil) |
| 182 | |
| 183 | handler := NewHandler(mgr, nil, logger, "openai", "") |
| 184 | |
| 185 | resp, err := handler.SendPrompt(context.Background(), &pb.SendPromptRequest{ |
| 186 | Prompt: "Hello", |
| 187 | }) |
| 188 | assert.NoError(t, err) |
| 189 | assert.Equal(t, "Hi there!", resp.Response) |
| 190 | assert.Equal(t, "gpt-4", resp.Model) |
| 191 | } |
| 192 | |
| 193 | func TestHandler_SendPrompt_WithClientAPIKey(t *testing.T) { |
| 194 | logger := zap.NewNop() |
nothing calls this directly
no test coverage detected