diff options
Diffstat (limited to 'internal/hexaicli/run_output_test.go')
| -rw-r--r-- | internal/hexaicli/run_output_test.go | 24 |
1 files changed, 21 insertions, 3 deletions
diff --git a/internal/hexaicli/run_output_test.go b/internal/hexaicli/run_output_test.go index f4e47fe..77a7c6a 100644 --- a/internal/hexaicli/run_output_test.go +++ b/internal/hexaicli/run_output_test.go @@ -350,8 +350,20 @@ func TestRunCLIJobs_MultiJob_WritesOutputs(t *testing.T) { t.Setenv("XDG_CACHE_HOME", t.TempDir()) jobs := []cliJob{ - {index: 0, provider: "a", cfg: appconfig.App{Provider: "a", OllamaBaseURL: "http://x", OllamaModel: "m"}, req: requestArgs{model: "m"}}, - {index: 1, provider: "b", cfg: appconfig.App{Provider: "b", OllamaBaseURL: "http://x", OllamaModel: "m"}, req: requestArgs{model: "m"}}, + {index: 0, provider: "a", cfg: appconfig.App{ + CoreConfig: appconfig.CoreConfig{Provider: "a"}, + ProviderConfig: appconfig.ProviderConfig{ + OllamaBaseURL: "http://x", + OllamaModel: "m", + }, + }, req: requestArgs{model: "m"}}, + {index: 1, provider: "b", cfg: appconfig.App{ + CoreConfig: appconfig.CoreConfig{Provider: "b"}, + ProviderConfig: appconfig.ProviderConfig{ + OllamaBaseURL: "http://x", + OllamaModel: "m", + }, + }, req: requestArgs{model: "m"}}, } msgs := buildMessages("hello") var stdout, stderr bytes.Buffer @@ -378,7 +390,13 @@ func TestRunCLIJobs_MultiJob_WritesOutputs(t *testing.T) { // Also test the runCLIJobs single-job (streaming) path. singleJobs := []cliJob{ - {index: 0, provider: "a", cfg: appconfig.App{Provider: "a", OllamaBaseURL: "http://x", OllamaModel: "m"}, req: requestArgs{model: "m"}}, + {index: 0, provider: "a", cfg: appconfig.App{ + CoreConfig: appconfig.CoreConfig{Provider: "a"}, + ProviderConfig: appconfig.ProviderConfig{ + OllamaBaseURL: "http://x", + OllamaModel: "m", + }, + }, req: requestArgs{model: "m"}}, } stdout.Reset() stderr.Reset() |