diff options
Diffstat (limited to 'internal/hexaicli/run_more_test.go')
| -rw-r--r-- | internal/hexaicli/run_more_test.go | 51 |
1 files changed, 27 insertions, 24 deletions
diff --git a/internal/hexaicli/run_more_test.go b/internal/hexaicli/run_more_test.go index ae29563..bd88d56 100644 --- a/internal/hexaicli/run_more_test.go +++ b/internal/hexaicli/run_more_test.go @@ -1,44 +1,47 @@ package hexaicli import ( - "bytes" - "context" - "testing" + "bytes" + "context" + "testing" - "codeberg.org/snonux/hexai/internal/appconfig" - "codeberg.org/snonux/hexai/internal/llm" + "codeberg.org/snonux/hexai/internal/appconfig" + "codeberg.org/snonux/hexai/internal/llm" ) type streamClient struct{} func (streamClient) Chat(ctx context.Context, msgs []llm.Message, opts ...llm.RequestOption) (string, error) { - return "X", nil + return "X", nil } func (streamClient) Name() string { return "fake" } func (streamClient) DefaultModel() string { return "m" } func (streamClient) ChatStream(ctx context.Context, msgs []llm.Message, onDelta func(string), opts ...llm.RequestOption) error { - onDelta("A") - onDelta("B") - return nil + onDelta("A") + onDelta("B") + return nil } func TestRunChat_Streaming(t *testing.T) { - var out, errw bytes.Buffer - input := "hello" - msgs := []llm.Message{{Role: "user", Content: input}} - if err := runChat(context.Background(), streamClient{}, msgs, input, &out, &errw); err != nil { - t.Fatalf("runChat failed: %v", err) - } - if out.String() != "AB" { - t.Fatalf("unexpected stream output: %q", out.String()) - } + var out, errw bytes.Buffer + input := "hello" + msgs := []llm.Message{{Role: "user", Content: input}} + if err := runChat(context.Background(), streamClient{}, msgs, input, &out, &errw); err != nil { + t.Fatalf("runChat failed: %v", err) + } + if out.String() != "AB" { + t.Fatalf("unexpected stream output: %q", out.String()) + } } func TestBuildMessagesFromConfig(t *testing.T) { - cfg := appconfig.App{PromptCLIDefaultSystem: "DEF", PromptCLIExplainSystem: "EXP"} - msgs := buildMessagesFromConfig(cfg, "tell me") - if msgs[0].Content != "DEF" { t.Fatalf("default system wrong: %q", msgs[0].Content) } - msgs = buildMessagesFromConfig(cfg, "please explain") - if msgs[0].Content != "EXP" { t.Fatalf("explain system wrong: %q", msgs[0].Content) } + cfg := appconfig.App{PromptCLIDefaultSystem: "DEF", PromptCLIExplainSystem: "EXP"} + msgs := buildMessagesFromConfig(cfg, "tell me") + if msgs[0].Content != "DEF" { + t.Fatalf("default system wrong: %q", msgs[0].Content) + } + msgs = buildMessagesFromConfig(cfg, "please explain") + if msgs[0].Content != "EXP" { + t.Fatalf("explain system wrong: %q", msgs[0].Content) + } } - |
