Consolidated addition, removal, and listing of locks under a single narratio locks command

This commit is contained in:
2026-05-21 19:36:14 -05:00
parent 135407ba7c
commit 870c2d69d5
7 changed files with 282 additions and 72 deletions

View File

@@ -23,9 +23,7 @@ Implemented commands:
- `session validate`: run read-only preflight checks for a session. - `session validate`: run read-only preflight checks for a session.
- `session init`: create local or remote `session.yml`. - `session init`: create local or remote `session.yml`.
- `artifacts list`: list effective artifact source IDs. - `artifacts list`: list effective artifact source IDs.
- `locks`: list effective archive promotion locks. - `locks`: list, add, and remove archive promotion locks.
- `lock`: add or update a remote session lock.
- `unlock`: remove a remote session lock.
Unknown commands print usage and exit non-zero. Unknown commands print usage and exit non-zero.
@@ -136,17 +134,17 @@ Valid stage names:
- `--previous-session-id <value>` - `--previous-session-id <value>`
- `--remote`: check promoted remote object availability. - `--remote`: check promoted remote object availability.
### `locks`, `lock`, `unlock` ### `locks`
- `--config <path>` - `--session-id <value>`: required for list, add, and remove.
- `--campaign <path>` - `--config <path>`: optional explicit `pipeline.yml` path.
- `--session <path>` - `--campaign <path>`: optional explicit `campaign.yml` path.
- `--session-id <value>` - `--session <path>`: optional explicit `session.yml` path.
- `--previous-session-id <value>` - `--previous-session-id <value>`: optional session template value.
- `lock <source>` positional source ID. - `add <source>`: add a remote lock for one artifact or transcript source.
- `lock --reason <text>` optional remote lock reason. - `add --reason <text>`: record an optional remote lock reason.
- `lock --force` updates an existing remote lock. - `add --force`: update the reason for an existing remote lock.
- `unlock <source>` positional source ID. - `remove <source>`: remove one remote lock.
## Command Reference ## Command Reference
@@ -291,25 +289,34 @@ narratio artifacts list [--config <pipeline.yml>] [--campaign <campaign.yml>] [-
`--remote` checks promoted top-level object availability through the storage adapter. `--remote` checks promoted top-level object availability through the storage adapter.
### `locks`, `lock`, and `unlock` ### `locks`
Purpose: Purpose:
- Inspect and mutate source-based archive promotion locks. - Inspect and mutate source-based archive promotion locks for one session.
Syntax: Syntax:
```bash ```bash
narratio locks [--config <pipeline.yml>] [--campaign <campaign.yml>] [--session <session.yml>] [--session-id <id>] narratio locks --session-id <id>
narratio lock [flags] <source> narratio locks add --session-id <id> [--reason <text>] [--force] <source>
narratio unlock [flags] <source> narratio locks remove --session-id <id> <source>
``` ```
Behavior: Behavior:
- static locks from `pipeline.archive.locks` and remote locks from `{session_prefix}/locks.yml` are merged. - `--session-id` is required for list, add, and remove.
- static locks win when sources duplicate remote locks. - optional `--config`, `--campaign`, and `--session` override default config discovery.
- `lock` writes or updates only remote locks. - list mode prints effective locks from static `pipeline.archive.locks` and remote `{session_prefix}/locks.yml`.
- `unlock` removes only remote locks and cannot remove static pipeline locks. - `locks add` writes only the remote lock store and fails if the source is already locked by pipeline config.
- `lock --force` is required to update an existing remote lock reason. - `locks remove` removes only remote locks and cannot remove static pipeline locks.
- `locks add --force` is required to update an existing remote lock reason.
Examples:
```bash
narratio locks --session-id 2026-04-04
narratio locks add --session-id 2026-04-04 --reason "manual transcript review" narratio.transcript.trimmed
narratio locks remove --session-id 2026-04-04 narratio.transcript.trimmed
```
### `run-stage` ### `run-stage`

View File

@@ -394,7 +394,7 @@ Remote mutable lock store:
- path: `{root_prefix}/campaigns/{campaign}/sessions/{session_id}/locks.yml`. - path: `{root_prefix}/campaigns/{campaign}/sessions/{session_id}/locks.yml`.
- strict YAML shape: top-level `locks`, each with `source` and optional `reason`. - strict YAML shape: top-level `locks`, each with `source` and optional `reason`.
- `narratio lock` and `narratio unlock` mutate only the remote lock store. - `narratio locks add` and `narratio locks remove` mutate only the remote lock store.
- writes use existence checks plus `--force` for updates; they are not compare-and-swap atomic. - writes use existence checks plus `--force` for updates; they are not compare-and-swap atomic.
Restore-related implications: Restore-related implications:

View File

@@ -153,10 +153,10 @@ Archive promotion is explicit and source-based:
Lock helper behavior: Lock helper behavior:
- `narratio locks --session-id <id>` lists effective static and remote locks. - `narratio locks --session-id <id>` lists effective static and remote locks.
- `narratio lock <source> --session-id <id> --reason <text>` writes or updates a remote lock. - `narratio locks add --session-id <id> --reason <text> <source>` writes a remote lock.
- `narratio unlock <source> --session-id <id>` removes only a remote lock. - `narratio locks add --session-id <id> --force --reason <text> <source>` updates an existing remote lock reason.
- `lock --force` is required to update an existing remote lock reason. - `narratio locks remove --session-id <id> <source>` removes only a remote lock.
- `unlock` cannot remove static pipeline locks. - `locks remove` cannot remove static pipeline locks.
- remote lock writes check whether the lock store exists, but are not compare-and-swap atomic. - remote lock writes check whether the lock store exists, but are not compare-and-swap atomic.
## Resume, retry, restore, and safe rerun behavior ## Resume, retry, restore, and safe rerun behavior

View File

@@ -22,22 +22,22 @@ The operator helper command set is no longer conceptual. Current behavior is doc
- `narratio artifacts list` - `narratio artifacts list`
- `narratio artifacts list --remote` - `narratio artifacts list --remote`
- `narratio locks` - `narratio locks`
- `narratio lock <source>` - `narratio locks add <source>`
- `narratio unlock <source>` - `narratio locks remove <source>`
## Implemented Decisions ## Implemented Decisions
- Helper output is text-only. No JSON schema exists yet. - Helper output is text-only. No JSON schema exists yet.
- `status` remains a top-level command. - `status` remains a top-level command.
- `session validate`, `session init`, and `artifacts list` are nested helper commands. - `session validate`, `session init`, and `artifacts list` are nested helper commands.
- `lock`, `unlock`, and `locks` are top-level commands. - `locks` is the single top-level command for listing, adding, and removing archive promotion locks.
- Remote session initialization requires explicit `--remote`. - Remote session initialization requires explicit `--remote`.
- Local session initialization requires `--output`. - Local session initialization requires `--output`.
- Remote artifact availability is opt-in with `artifacts list --remote`. - Remote artifact availability is opt-in with `artifacts list --remote`.
- Mutable locks are source-based and stored at `{session_prefix}/locks.yml`. - Mutable locks are source-based and stored at `{session_prefix}/locks.yml`.
- The remote lock store uses strict YAML with top-level `locks`. - The remote lock store uses strict YAML with top-level `locks`.
- Static `pipeline.archive.locks` and remote locks are merged; static locks win on duplicate sources. - Static `pipeline.archive.locks` and remote locks are merged; static locks win on duplicate sources.
- `unlock` removes only remote locks. - `locks remove` removes only remote locks.
- Ordinary execution `--force` does not override locks. - Ordinary execution `--force` does not override locks.
- Remote lock writes use existence checks and `--force` for updates; there is no compare-and-swap protection. - Remote lock writes use existence checks and `--force` for updates; there is no compare-and-swap protection.
@@ -50,4 +50,3 @@ These are intentionally not implemented:
- Rich remote artifact availability across historical run-local objects. - Rich remote artifact availability across historical run-local objects.
- Session-lock acquisition for remote mutation helpers. - Session-lock acquisition for remote mutation helpers.
- Broader campaign helper commands such as `campaign validate` or `campaign publish`. - Broader campaign helper commands such as `campaign validate` or `campaign publish`.

View File

@@ -7,7 +7,7 @@ import (
"strings" "strings"
) )
var supportedCommands = []string{"run", "plan", "status", "resume", "run-stage", "restore", "session", "artifacts", "locks", "lock", "unlock"} var supportedCommands = []string{"run", "plan", "status", "resume", "run-stage", "restore", "session", "artifacts", "locks"}
// Execute dispatches CLI commands and returns a process exit code. // Execute dispatches CLI commands and returns a process exit code.
func Execute(args []string, stdout, stderr io.Writer) int { func Execute(args []string, stdout, stderr io.Writer) int {
@@ -40,10 +40,6 @@ func Execute(args []string, stdout, stderr io.Writer) int {
err = Artifacts(ctx, cmdArgs, stdout) err = Artifacts(ctx, cmdArgs, stdout)
case "locks": case "locks":
err = Locks(ctx, cmdArgs, stdout) err = Locks(ctx, cmdArgs, stdout)
case "lock":
err = Lock(ctx, cmdArgs, stdout)
case "unlock":
err = Unlock(ctx, cmdArgs, stdout)
default: default:
fmt.Fprintf(stderr, "unknown command: %q\n\n", cmd) fmt.Fprintf(stderr, "unknown command: %q\n\n", cmd)
printUsage(stderr) printUsage(stderr)

View File

@@ -379,8 +379,23 @@ func ArtifactsList(ctx context.Context, args []string, out io.Writer) error {
return nil return nil
} }
// Locks lists effective archive locks. // Locks dispatches archive lock list and mutation helpers.
func Locks(ctx context.Context, args []string, out io.Writer) error { func Locks(ctx context.Context, args []string, out io.Writer) error {
if len(args) > 0 && !strings.HasPrefix(args[0], "-") {
switch args[0] {
case "add":
return LocksAdd(ctx, args[1:], out)
case "remove":
return LocksRemove(ctx, args[1:], out)
default:
return fmt.Errorf("locks: unknown subcommand %q", args[0])
}
}
return LocksList(ctx, args, out)
}
// LocksList lists effective archive locks.
func LocksList(ctx context.Context, args []string, out io.Writer) error {
fs := flag.NewFlagSet("locks", flag.ContinueOnError) fs := flag.NewFlagSet("locks", flag.ContinueOnError)
fs.SetOutput(io.Discard) fs.SetOutput(io.Discard)
var flags commonConfigFlags var flags commonConfigFlags
@@ -391,6 +406,9 @@ func Locks(ctx context.Context, args []string, out io.Writer) error {
if fs.NArg() != 0 { if fs.NArg() != 0 {
return fmt.Errorf("locks: unexpected positional arguments") return fmt.Errorf("locks: unexpected positional arguments")
} }
if strings.TrimSpace(flags.sessionID) == "" {
return fmt.Errorf("locks: --session-id is required")
}
cfg, _, locks, _, err := loadHelperContext(ctx, flags, true) cfg, _, locks, _, err := loadHelperContext(ctx, flags, true)
if err != nil { if err != nil {
return fmt.Errorf("locks: %w", err) return fmt.Errorf("locks: %w", err)
@@ -399,9 +417,9 @@ func Locks(ctx context.Context, args []string, out io.Writer) error {
return nil return nil
} }
// Lock adds or updates one remote lock. // LocksAdd adds or updates one remote lock.
func Lock(ctx context.Context, args []string, out io.Writer) error { func LocksAdd(ctx context.Context, args []string, out io.Writer) error {
fs := flag.NewFlagSet("lock", flag.ContinueOnError) fs := flag.NewFlagSet("locks add", flag.ContinueOnError)
fs.SetOutput(io.Discard) fs.SetOutput(io.Discard)
var flags commonConfigFlags var flags commonConfigFlags
var reason string var reason string
@@ -410,71 +428,77 @@ func Lock(ctx context.Context, args []string, out io.Writer) error {
fs.StringVar(&reason, "reason", "", "lock reason") fs.StringVar(&reason, "reason", "", "lock reason")
fs.BoolVar(&force, "force", false, "update existing remote lock") fs.BoolVar(&force, "force", false, "update existing remote lock")
if err := fs.Parse(args); err != nil { if err := fs.Parse(args); err != nil {
return fmt.Errorf("lock: invalid flags: %w", err) return fmt.Errorf("locks add: invalid flags: %w", err)
} }
if fs.NArg() != 1 { if fs.NArg() != 1 {
return fmt.Errorf("lock: expected exactly one source id") return fmt.Errorf("locks add: expected exactly one source id")
}
if strings.TrimSpace(flags.sessionID) == "" {
return fmt.Errorf("locks add: --session-id is required")
} }
source := strings.TrimSpace(fs.Arg(0)) source := strings.TrimSpace(fs.Arg(0))
cfg, store, locks, _, err := loadHelperContext(ctx, flags, true) cfg, store, locks, _, err := loadHelperContext(ctx, flags, true)
if err != nil { if err != nil {
return fmt.Errorf("lock: %w", err) return fmt.Errorf("locks add: %w", err)
} }
if _, err := config.ValidateArchiveLockRules([]config.ArchiveLockRule{{Source: source}}, cfg.Pipeline.Scriptorium, "lock"); err != nil { if _, err := config.ValidateArchiveLockRules([]config.ArchiveLockRule{{Source: source}}, cfg.Pipeline.Scriptorium, "locks add"); err != nil {
return fmt.Errorf("lock: %w", err) return fmt.Errorf("locks add: %w", err)
} }
if _, ok := lockSourceSet(locks.Static)[source]; ok { if _, ok := lockSourceSet(locks.Static)[source]; ok {
return fmt.Errorf("lock: source %q is locked by pipeline config and cannot be modified remotely", source) return fmt.Errorf("locks add: source %q is locked by pipeline config and cannot be modified remotely", source)
} }
remoteSet := lockSourceSet(locks.Remote) remoteSet := lockSourceSet(locks.Remote)
if _, exists := remoteSet[source]; exists && !force { if _, exists := remoteSet[source]; exists && !force {
return fmt.Errorf("lock: remote lock for %q already exists; pass --force to update", source) return fmt.Errorf("locks add: remote lock for %q already exists; pass --force to update", source)
} }
remoteSet[source] = config.ArchiveLockRule{Source: source, Reason: strings.TrimSpace(reason)} remoteSet[source] = config.ArchiveLockRule{Source: source, Reason: strings.TrimSpace(reason)}
remoteLocks := lockMapValues(remoteSet) remoteLocks := lockMapValues(remoteSet)
if _, err := config.ValidateArchiveLockRules(remoteLocks, cfg.Pipeline.Scriptorium, "locks"); err != nil { if _, err := config.ValidateArchiveLockRules(remoteLocks, cfg.Pipeline.Scriptorium, "locks"); err != nil {
return fmt.Errorf("lock: %w", err) return fmt.Errorf("locks add: %w", err)
} }
if err := uploadRemoteLockStore(ctx, store, locks.Key, &config.ArchiveLockStore{Locks: remoteLocks}); err != nil { if err := uploadRemoteLockStore(ctx, store, locks.Key, &config.ArchiveLockStore{Locks: remoteLocks}); err != nil {
return fmt.Errorf("lock: %w", err) return fmt.Errorf("locks add: %w", err)
} }
_, err = fmt.Fprintf(out, "narratio lock: locked %s\n", source) _, err = fmt.Fprintf(out, "narratio locks add: locked %s\n", source)
return err return err
} }
// Unlock removes one remote lock. // LocksRemove removes one remote lock.
func Unlock(ctx context.Context, args []string, out io.Writer) error { func LocksRemove(ctx context.Context, args []string, out io.Writer) error {
fs := flag.NewFlagSet("unlock", flag.ContinueOnError) fs := flag.NewFlagSet("locks remove", flag.ContinueOnError)
fs.SetOutput(io.Discard) fs.SetOutput(io.Discard)
var flags commonConfigFlags var flags commonConfigFlags
addCommonConfigFlags(fs, &flags) addCommonConfigFlags(fs, &flags)
if err := fs.Parse(args); err != nil { if err := fs.Parse(args); err != nil {
return fmt.Errorf("unlock: invalid flags: %w", err) return fmt.Errorf("locks remove: invalid flags: %w", err)
} }
if fs.NArg() != 1 { if fs.NArg() != 1 {
return fmt.Errorf("unlock: expected exactly one source id") return fmt.Errorf("locks remove: expected exactly one source id")
}
if strings.TrimSpace(flags.sessionID) == "" {
return fmt.Errorf("locks remove: --session-id is required")
} }
source := strings.TrimSpace(fs.Arg(0)) source := strings.TrimSpace(fs.Arg(0))
cfg, store, locks, _, err := loadHelperContext(ctx, flags, true) cfg, store, locks, _, err := loadHelperContext(ctx, flags, true)
if err != nil { if err != nil {
return fmt.Errorf("unlock: %w", err) return fmt.Errorf("locks remove: %w", err)
} }
if _, err := config.ValidateArchiveLockRules([]config.ArchiveLockRule{{Source: source}}, cfg.Pipeline.Scriptorium, "unlock"); err != nil { if _, err := config.ValidateArchiveLockRules([]config.ArchiveLockRule{{Source: source}}, cfg.Pipeline.Scriptorium, "locks remove"); err != nil {
return fmt.Errorf("unlock: %w", err) return fmt.Errorf("locks remove: %w", err)
} }
remoteSet := lockSourceSet(locks.Remote) remoteSet := lockSourceSet(locks.Remote)
if _, ok := remoteSet[source]; !ok { if _, ok := remoteSet[source]; !ok {
if _, static := lockSourceSet(locks.Static)[source]; static { if _, static := lockSourceSet(locks.Static)[source]; static {
return fmt.Errorf("unlock: source %q is locked by pipeline config and cannot be unlocked remotely", source) return fmt.Errorf("locks remove: source %q is locked by pipeline config and cannot be unlocked remotely", source)
} }
return fmt.Errorf("unlock: remote lock for %q does not exist", source) return fmt.Errorf("locks remove: remote lock for %q does not exist", source)
} }
delete(remoteSet, source) delete(remoteSet, source)
remoteLocks := lockMapValues(remoteSet) remoteLocks := lockMapValues(remoteSet)
if err := uploadRemoteLockStore(ctx, store, locks.Key, &config.ArchiveLockStore{Locks: remoteLocks}); err != nil { if err := uploadRemoteLockStore(ctx, store, locks.Key, &config.ArchiveLockStore{Locks: remoteLocks}); err != nil {
return fmt.Errorf("unlock: %w", err) return fmt.Errorf("locks remove: %w", err)
} }
_, err = fmt.Fprintf(out, "narratio unlock: unlocked %s\n", source) _, err = fmt.Fprintf(out, "narratio locks remove: unlocked %s\n", source)
return err return err
} }

View File

@@ -91,7 +91,7 @@ inputs:
} }
} }
func TestExecuteLockAndUnlockUseRemoteLockStore(t *testing.T) { func TestExecuteLocksAddListAndRemoveUseRemoteLockStore(t *testing.T) {
workspaceRoot := t.TempDir() workspaceRoot := t.TempDir()
pipelinePath, campaignPath, sessionPath := writeValidConfigFiles(t, workspaceRoot) pipelinePath, campaignPath, sessionPath := writeValidConfigFiles(t, workspaceRoot)
fake := &storage.FakeBackend{} fake := &storage.FakeBackend{}
@@ -101,15 +101,16 @@ func TestExecuteLockAndUnlockUseRemoteLockStore(t *testing.T) {
var stdout bytes.Buffer var stdout bytes.Buffer
var stderr bytes.Buffer var stderr bytes.Buffer
code := Execute([]string{ code := Execute([]string{
"lock", "locks", "add",
"--config", pipelinePath, "--config", pipelinePath,
"--campaign", campaignPath, "--campaign", campaignPath,
"--session", sessionPath, "--session", sessionPath,
"--session-id", "2026-05-03",
"--reason", "manual edit", "--reason", "manual edit",
"narratio.transcript.trimmed", "narratio.transcript.trimmed",
}, &stdout, &stderr) }, &stdout, &stderr)
if code != 0 { if code != 0 {
t.Fatalf("lock exit code = %d, want 0; stderr=%q", code, stderr.String()) t.Fatalf("locks add exit code = %d, want 0; stderr=%q", code, stderr.String())
} }
key := artifacts.S3SessionLocksKey(artifacts.S3SessionPrefix("dnd", "sample-campaign", "2026-05-03")) key := artifacts.S3SessionLocksKey(artifacts.S3SessionPrefix("dnd", "sample-campaign", "2026-05-03"))
obj, ok := fake.Objects[key] obj, ok := fake.Objects[key]
@@ -123,24 +124,187 @@ func TestExecuteLockAndUnlockUseRemoteLockStore(t *testing.T) {
stdout.Reset() stdout.Reset()
stderr.Reset() stderr.Reset()
code = Execute([]string{ code = Execute([]string{
"unlock", "locks",
"--config", pipelinePath, "--config", pipelinePath,
"--campaign", campaignPath, "--campaign", campaignPath,
"--session", sessionPath, "--session", sessionPath,
"--session-id", "2026-05-03",
}, &stdout, &stderr)
if code != 0 {
t.Fatalf("locks list exit code = %d, want 0; stderr=%q", code, stderr.String())
}
if !strings.Contains(stdout.String(), "- narratio.transcript.trimmed origin=remote") {
t.Fatalf("stdout = %q, want remote lock", stdout.String())
}
stdout.Reset()
stderr.Reset()
code = Execute([]string{
"locks", "remove",
"--config", pipelinePath,
"--campaign", campaignPath,
"--session", sessionPath,
"--session-id", "2026-05-03",
"narratio.transcript.trimmed", "narratio.transcript.trimmed",
}, &stdout, &stderr) }, &stdout, &stderr)
if code != 0 { if code != 0 {
t.Fatalf("unlock exit code = %d, want 0; stderr=%q", code, stderr.String()) t.Fatalf("locks remove exit code = %d, want 0; stderr=%q", code, stderr.String())
} }
store, err := config.LoadArchiveLockStoreBytes("locks.yml", fake.Objects[key].Data, nil) store, err := config.LoadArchiveLockStoreBytes("locks.yml", fake.Objects[key].Data, nil)
if err != nil { if err != nil {
t.Fatalf("LoadArchiveLockStoreBytes() error = %v", err) t.Fatalf("LoadArchiveLockStoreBytes() error = %v", err)
} }
if len(store.Locks) != 0 { if len(store.Locks) != 0 {
t.Fatalf("locks after unlock = %#v, want empty", store.Locks) t.Fatalf("locks after remove = %#v, want empty", store.Locks)
} }
if storeInitCalls != 2 { if storeInitCalls != 3 {
t.Fatalf("object store init calls = %d, want 2", storeInitCalls) t.Fatalf("object store init calls = %d, want 3", storeInitCalls)
}
}
func TestExecuteLocksAddDuplicateRequiresForce(t *testing.T) {
workspaceRoot := t.TempDir()
pipelinePath, campaignPath, sessionPath := writeValidConfigFiles(t, workspaceRoot)
fake := &storage.FakeBackend{}
var storeInitCalls int
restoreAppConfigTestGlobals(t, fake, &storeInitCalls, []string{sessionPath})
var stdout bytes.Buffer
var stderr bytes.Buffer
code := Execute([]string{
"locks", "add",
"--config", pipelinePath,
"--campaign", campaignPath,
"--session", sessionPath,
"--session-id", "2026-05-03",
"--reason", "first",
"narratio.transcript.trimmed",
}, &stdout, &stderr)
if code != 0 {
t.Fatalf("initial locks add exit code = %d, want 0; stderr=%q", code, stderr.String())
}
stdout.Reset()
stderr.Reset()
code = Execute([]string{
"locks", "add",
"--config", pipelinePath,
"--campaign", campaignPath,
"--session", sessionPath,
"--session-id", "2026-05-03",
"--reason", "second",
"narratio.transcript.trimmed",
}, &stdout, &stderr)
if code == 0 {
t.Fatal("duplicate locks add exit code = 0, want non-zero")
}
if !strings.Contains(stderr.String(), "pass --force to update") {
t.Fatalf("stderr = %q, want force guidance", stderr.String())
}
stdout.Reset()
stderr.Reset()
code = Execute([]string{
"locks", "add",
"--config", pipelinePath,
"--campaign", campaignPath,
"--session", sessionPath,
"--session-id", "2026-05-03",
"--reason", "second",
"--force",
"narratio.transcript.trimmed",
}, &stdout, &stderr)
if code != 0 {
t.Fatalf("forced locks add exit code = %d, want 0; stderr=%q", code, stderr.String())
}
key := artifacts.S3SessionLocksKey(artifacts.S3SessionPrefix("dnd", "sample-campaign", "2026-05-03"))
if !strings.Contains(string(fake.Objects[key].Data), "reason: second") {
t.Fatalf("lock store data = %q, want updated reason", string(fake.Objects[key].Data))
}
}
func TestExecuteLocksRequireSessionID(t *testing.T) {
tests := []struct {
name string
args []string
want string
}{
{"list", []string{"locks"}, "locks: --session-id is required"},
{"add", []string{"locks", "add", "narratio.transcript.trimmed"}, "locks add: --session-id is required"},
{"remove", []string{"locks", "remove", "narratio.transcript.trimmed"}, "locks remove: --session-id is required"},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
var stdout bytes.Buffer
var stderr bytes.Buffer
code := Execute(tt.args, &stdout, &stderr)
if code == 0 {
t.Fatal("exit code = 0, want non-zero")
}
if !strings.Contains(stderr.String(), tt.want) {
t.Fatalf("stderr = %q, want %q", stderr.String(), tt.want)
}
})
}
}
func TestExecuteLocksCannotModifyStaticLocks(t *testing.T) {
workspaceRoot := t.TempDir()
pipelinePath, campaignPath, sessionPath := writeValidConfigFiles(t, workspaceRoot)
addStaticArchiveLockToPipelineConfig(t, pipelinePath, "narratio.transcript.trimmed")
fake := &storage.FakeBackend{}
var storeInitCalls int
restoreAppConfigTestGlobals(t, fake, &storeInitCalls, []string{sessionPath})
var stdout bytes.Buffer
var stderr bytes.Buffer
code := Execute([]string{
"locks", "add",
"--config", pipelinePath,
"--campaign", campaignPath,
"--session", sessionPath,
"--session-id", "2026-05-03",
"narratio.transcript.trimmed",
}, &stdout, &stderr)
if code == 0 {
t.Fatal("locks add static lock exit code = 0, want non-zero")
}
if !strings.Contains(stderr.String(), "locked by pipeline config") {
t.Fatalf("stderr = %q, want static lock error", stderr.String())
}
stdout.Reset()
stderr.Reset()
code = Execute([]string{
"locks", "remove",
"--config", pipelinePath,
"--campaign", campaignPath,
"--session", sessionPath,
"--session-id", "2026-05-03",
"narratio.transcript.trimmed",
}, &stdout, &stderr)
if code == 0 {
t.Fatal("locks remove static lock exit code = 0, want non-zero")
}
if !strings.Contains(stderr.String(), "locked by pipeline config") {
t.Fatalf("stderr = %q, want static lock error", stderr.String())
}
}
func TestExecuteTopLevelLockAndUnlockAreRemoved(t *testing.T) {
tests := []string{"lock", "unlock"}
for _, cmd := range tests {
t.Run(cmd, func(t *testing.T) {
var stdout bytes.Buffer
var stderr bytes.Buffer
code := Execute([]string{cmd, "narratio.transcript.trimmed"}, &stdout, &stderr)
if code == 0 {
t.Fatal("exit code = 0, want non-zero")
}
if !strings.Contains(stderr.String(), `unknown command: "`+cmd+`"`) {
t.Fatalf("stderr = %q, want unknown command", stderr.String())
}
})
} }
} }
@@ -237,3 +401,23 @@ func writeValidArchiveConfigFiles(t *testing.T, workspaceRoot string) (string, s
} }
return pipelinePath, campaignPath, sessionPath return pipelinePath, campaignPath, sessionPath
} }
func addStaticArchiveLockToPipelineConfig(t *testing.T, pipelinePath, source string) {
t.Helper()
data, err := os.ReadFile(pipelinePath)
if err != nil {
t.Fatalf("read pipeline: %v", err)
}
updated := strings.Replace(
string(data),
"archive:\n enabled: true\n upload_run: false\n",
"archive:\n enabled: true\n upload_run: false\n locks:\n - source: "+source+"\n reason: static review\n",
1,
)
if updated == string(data) {
t.Fatalf("archive section not found in pipeline config")
}
if err := os.WriteFile(pipelinePath, []byte(updated), 0o644); err != nil {
t.Fatalf("write pipeline: %v", err)
}
}