From 0ac2d186e84f77d73d924e2c0ce975a17c3a8078 Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Sun, 28 Sep 2025 00:20:05 +0300 Subject: Improve multi-provider completion streaming and CLI selector flags --- internal/hexaicli/run_test.go | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) (limited to 'internal/hexaicli/run_test.go') diff --git a/internal/hexaicli/run_test.go b/internal/hexaicli/run_test.go index f11545e..dfde068 100644 --- a/internal/hexaicli/run_test.go +++ b/internal/hexaicli/run_test.go @@ -225,6 +225,23 @@ func TestBuildCLIJobs_MultiEntries(t *testing.T) { } } +func TestFilterJobsBySelection(t *testing.T) { + jobs := []cliJob{{index: 0, provider: "openai"}, {index: 1, provider: "ollama"}, {index: 2, provider: "copilot"}} + filtered, err := filterJobsBySelection(jobs, []int{2, 0}) + if err != nil { + t.Fatalf("unexpected error: %v", err) + } + if len(filtered) != 2 || filtered[0].provider != "copilot" || filtered[1].provider != "openai" { + t.Fatalf("unexpected filtered order: %+v", filtered) + } + if filtered[0].index != 0 || filtered[1].index != 1 { + t.Fatalf("expected reindexed jobs, got %+v", filtered) + } + if _, err := filterJobsBySelection(jobs, []int{5}); err == nil { + t.Fatalf("expected out-of-range error") + } +} + func TestNewClientFromConfig_Ollama(t *testing.T) { cfg := appconfig.App{Provider: "ollama", OllamaBaseURL: "http://x", OllamaModel: "m"} c, err := newClientFromConfig(cfg) -- cgit v1.2.3