160 lines
6.9 KiB
Go
160 lines
6.9 KiB
Go
package cli
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"errors"
|
|
"io"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"gitea.maximumdirect.net/eric/notarius/internal/core/artifacts"
|
|
"gitea.maximumdirect.net/eric/notarius/internal/core/config"
|
|
"gitea.maximumdirect.net/eric/notarius/internal/framework/contracts"
|
|
)
|
|
|
|
func TestConfigKitReadOnlyCommandsLoadImportsAndOverlays(t *testing.T) {
|
|
t.Parallel()
|
|
root := t.TempDir()
|
|
path := filepath.Join(root, "notarius.yml")
|
|
writeConfigKitFile(t, filepath.Join(root, "fragment.yml"), "pipelines:\n imported:\n input: {module: input}\n")
|
|
writeConfigKitFile(t, filepath.Join(root, "overlays", "development.yml"), "pipelines:\n overlaid:\n input: {module: input}\n")
|
|
writeConfigKitFile(t, path, "version: 5\nconfigkit:\n imports:\n - path: fragment.yml\npipelines:\n base:\n input: {module: input}\n")
|
|
var stdout, stderr bytes.Buffer
|
|
if code := RunWithOptions([]string{"pipelines", "list", "--config", path, "--config-overlay", "development"}, &stdout, &stderr, commandContractOptions(t)); code != 0 || stdout.String() != "base\nimported\noverlaid\n" {
|
|
t.Fatalf("list = %d, %q, %q", code, stdout.String(), stderr.String())
|
|
}
|
|
stdout.Reset()
|
|
stderr.Reset()
|
|
if code := RunWithOptions([]string{"config", "validate", "--config", path, "--config-overlay", "development"}, &stdout, &stderr, commandContractOptions(t)); code != 0 || !strings.Contains(stdout.String(), "is valid") {
|
|
t.Fatalf("validate = %d, %q, %q", code, stdout.String(), stderr.String())
|
|
}
|
|
for _, selection := range [][]string{nil, {"--config-overlay", ""}, {"--config-overlay", "default"}} {
|
|
stdout.Reset()
|
|
stderr.Reset()
|
|
args := append([]string{"pipelines", "list", "--config", path}, selection...)
|
|
if code := RunWithOptions(args, &stdout, &stderr, commandContractOptions(t)); code != 0 || stdout.String() != "base\nimported\n" || stderr.Len() != 0 {
|
|
t.Fatalf("base list %q = %d, %q, %q", selection, code, stdout.String(), stderr.String())
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRetiredConfigProfileFlagUsesNormalUnknownFlagBehavior(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
for _, args := range [][]string{
|
|
{"run", "sample", "--config-profile=development"},
|
|
{"config", "validate", "--config-profile=development"},
|
|
{"config", "effective", "--config-profile=development"},
|
|
{"config", "explain", "--config-profile=development", ""},
|
|
{"pipelines", "list", "--config-profile=development"},
|
|
} {
|
|
t.Run(strings.Join(args[:2], " "), func(t *testing.T) {
|
|
var stdout, stderr bytes.Buffer
|
|
if code := RunWithOptions(args, &stdout, &stderr, commandContractOptions(t)); code != 2 || stdout.Len() != 0 || !strings.Contains(stderr.String(), "not defined") {
|
|
t.Fatalf("args=%q code=%d stdout=%q stderr=%q", args, code, stdout.String(), stderr.String())
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestConfigKitReadOnlyCommandsRejectVersionFour(t *testing.T) {
|
|
t.Parallel()
|
|
path := filepath.Join(t.TempDir(), "notarius.yml")
|
|
writeConfigKitFile(t, path, "version: 4\n")
|
|
var stdout, stderr bytes.Buffer
|
|
if code := runPipelinesList([]string{"--config", path}, &stdout, &stderr, commandContractOptions(t)); code != 1 || !strings.Contains(stderr.String(), "version") {
|
|
t.Fatalf("list = %d, %q", code, stderr.String())
|
|
}
|
|
}
|
|
|
|
func TestConfigurationInspectionCommandsRenderSafeConfigKitViews(t *testing.T) {
|
|
t.Parallel()
|
|
root := t.TempDir()
|
|
path := filepath.Join(root, "notarius.yml")
|
|
writeConfigKitFile(t, path, `version: 5
|
|
concurrency:
|
|
total_llm: 4
|
|
pipelines:
|
|
"0":
|
|
input: {module: input}
|
|
`)
|
|
writeConfigKitFile(t, filepath.Join(root, "overlays", "development.yml"), "concurrency:\n total_llm: 3\n")
|
|
writeConfigKitFile(t, filepath.Join(root, "overlays", "retired.yml"), "pipelines: null\n")
|
|
|
|
providerCalled := false
|
|
opts := commandContractOptions(t)
|
|
opts.LLMClientFactory = func(context.Context, config.Config, string, LLMRuntimeOverrides) (contracts.StructuredLLMClient, []artifacts.LLMProfileManifest, error) {
|
|
providerCalled = true
|
|
return nil, nil, errors.New("LLM client must not be constructed")
|
|
}
|
|
|
|
var stdout, stderr bytes.Buffer
|
|
if code := RunWithOptions([]string{"config", "effective", "--config", path, "--config-overlay", "development"}, &stdout, &stderr, opts); code != 0 || !strings.Contains(stdout.String(), "total_llm: 3") || stderr.Len() != 0 {
|
|
t.Fatalf("effective = %d, %q, %q", code, stdout.String(), stderr.String())
|
|
}
|
|
if providerCalled {
|
|
t.Fatal("config effective constructed an LLM client")
|
|
}
|
|
|
|
stdout.Reset()
|
|
stderr.Reset()
|
|
if code := RunWithOptions([]string{"config", "explain", "--config", path, "--config-overlay", "development", "/concurrency/total_llm"}, &stdout, &stderr, opts); code != 0 || !strings.Contains(stdout.String(), `path: "/concurrency/total_llm"`) || stderr.Len() != 0 {
|
|
t.Fatalf("explain effective = %d, %q, %q", code, stdout.String(), stderr.String())
|
|
}
|
|
|
|
stdout.Reset()
|
|
stderr.Reset()
|
|
if code := RunWithOptions([]string{"config", "explain", "--config", path, ""}, &stdout, &stderr, opts); code != 0 || !strings.Contains(stdout.String(), `path: ""`) || stderr.Len() != 0 {
|
|
t.Fatalf("explain root = %d, %q, %q", code, stdout.String(), stderr.String())
|
|
}
|
|
|
|
stdout.Reset()
|
|
stderr.Reset()
|
|
if code := RunWithOptions([]string{"config", "explain", "--config", path, "--config-overlay", "retired", "/pipelines/0"}, &stdout, &stderr, opts); code != 0 || !strings.Contains(stdout.String(), `path: "/pipelines/0"`) || stderr.Len() != 0 {
|
|
t.Fatalf("explain removed mapping key = %d, %q, %q", code, stdout.String(), stderr.String())
|
|
}
|
|
}
|
|
|
|
func TestConfigExplainRejectsInvalidAndUnknownPointers(t *testing.T) {
|
|
t.Parallel()
|
|
path := writeCommandConfig(t, "session", "other")
|
|
for _, test := range []struct {
|
|
name string
|
|
pointer string
|
|
wantCode int
|
|
}{
|
|
{name: "missing", wantCode: 2},
|
|
{name: "malformed escape", pointer: "/pipelines~2session", wantCode: 2},
|
|
{name: "unknown", pointer: "/unknown", wantCode: 1},
|
|
{name: "invalid sequence index", pointer: "/pipelines/session/steps/+0", wantCode: 1},
|
|
} {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
var stdout, stderr bytes.Buffer
|
|
args := []string{"config", "explain", "--config", path}
|
|
if test.name != "missing" {
|
|
args = append(args, test.pointer)
|
|
}
|
|
if code := RunWithOptions(args, &stdout, &stderr, commandContractOptions(t)); code != test.wantCode || stdout.Len() != 0 || stderr.Len() == 0 {
|
|
t.Fatalf("code=%d stdout=%q stderr=%q", code, stdout.String(), stderr.String())
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestConfigInspectionCommandsReportWriterFailures(t *testing.T) {
|
|
t.Parallel()
|
|
path := writeCommandConfig(t, "session", "other")
|
|
var stderr bytes.Buffer
|
|
if code := RunWithOptions([]string{"config", "effective", "--config", path}, failingInspectionWriter{}, &stderr, commandContractOptions(t)); code != 1 || !strings.Contains(stderr.String(), "write effective configuration") {
|
|
t.Fatalf("effective writer failure = %d, %q", code, stderr.String())
|
|
}
|
|
}
|
|
|
|
type failingInspectionWriter struct{}
|
|
|
|
func (failingInspectionWriter) Write([]byte) (int, error) {
|
|
return 0, io.ErrClosedPipe
|
|
}
|