VERSE
6ffb9f83f2
- add transition scoring engine to learn sequential workflows per directory using command skeletons - prioritize current active git branch over older branches in suggestions - restore strict chronological order for history navigation and bypass AI re-ranking - add in-memory session history for instant access to just-run commands - fix PTY prompt not syncing when using arrow keys in the history menu - fix history tie-breaker to properly prioritize newer commands by assigning larger IDs - fix potential mutex deadlock between bufferMu and pty write during history navigation - fix workspace git detection to remove depth limits and correctly identify repositories with detached HEADs - fix SQLite connection leaks in frecency queries - fix global state leaks in history tests by snapshotting and restoring registry states - modernize string prefix checks using strings.CutPrefix
98 lines
2.0 KiB
Go
98 lines
2.0 KiB
Go
package spec
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestTryExtractSkeleton(t *testing.T) {
|
|
ResetRegistry()
|
|
Register(&Spec{
|
|
Name: "git",
|
|
Subcommands: []Subcommand{
|
|
{
|
|
Name: "checkout",
|
|
Aliases: []string{"co"},
|
|
},
|
|
{
|
|
Name: "remote",
|
|
Subcommands: []Subcommand{
|
|
{Name: "add"},
|
|
},
|
|
},
|
|
},
|
|
})
|
|
Register(&Spec{
|
|
Name: "docker",
|
|
Subcommands: []Subcommand{
|
|
{
|
|
Name: "compose",
|
|
Subcommands: []Subcommand{
|
|
{Name: "up"},
|
|
},
|
|
},
|
|
},
|
|
})
|
|
|
|
tests := []struct {
|
|
name string
|
|
input string
|
|
wantSkeleton string
|
|
wantOk bool
|
|
}{
|
|
{
|
|
name: "git checkout with argument",
|
|
input: "git checkout feature-x -b",
|
|
wantSkeleton: "git checkout",
|
|
wantOk: true,
|
|
},
|
|
{
|
|
name: "git co alias with argument",
|
|
input: "git co feature-y",
|
|
wantSkeleton: "git checkout",
|
|
wantOk: true,
|
|
},
|
|
{
|
|
name: "git remote add with arguments",
|
|
input: "git remote add origin https://github.com/test/test.git",
|
|
wantSkeleton: "git remote add",
|
|
wantOk: true,
|
|
},
|
|
{
|
|
name: "git global flag before subcommand",
|
|
input: "git -C /tmp checkout main",
|
|
wantSkeleton: "git checkout",
|
|
wantOk: true,
|
|
},
|
|
{
|
|
name: "docker compose up with flags",
|
|
input: "docker compose up -d",
|
|
wantSkeleton: "docker compose up",
|
|
wantOk: true,
|
|
},
|
|
{
|
|
name: "unregistered spec",
|
|
input: "cargo build --release",
|
|
wantSkeleton: "",
|
|
wantOk: false,
|
|
},
|
|
{
|
|
name: "empty input",
|
|
input: " ",
|
|
wantSkeleton: "",
|
|
wantOk: false,
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
got, ok := TryExtractSkeleton(tc.input)
|
|
if ok != tc.wantOk {
|
|
t.Fatalf("TryExtractSkeleton(%q) ok = %v, want %v", tc.input, ok, tc.wantOk)
|
|
}
|
|
if got != tc.wantSkeleton {
|
|
t.Errorf("TryExtractSkeleton(%q) = %q, want %q", tc.input, got, tc.wantSkeleton)
|
|
}
|
|
})
|
|
}
|
|
}
|