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
115 lines
3.3 KiB
Go
115 lines
3.3 KiB
Go
package scoring
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/versenilvis/iris/internal/workspace"
|
|
)
|
|
|
|
func TestApplyContextRules_MultiEcosystem(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
ws workspace.WorkspaceInfo
|
|
cmd string
|
|
expected int
|
|
}{
|
|
{
|
|
name: "git status inside git repo",
|
|
ws: workspace.WorkspaceInfo{HasGit: true},
|
|
cmd: "git status -s",
|
|
expected: 40,
|
|
},
|
|
{
|
|
name: "git init inside git repo penalized",
|
|
ws: workspace.WorkspaceInfo{HasGit: true},
|
|
cmd: "git init",
|
|
expected: -50,
|
|
},
|
|
{
|
|
name: "bun run dev inside node project",
|
|
ws: workspace.WorkspaceInfo{HasNodeProject: true},
|
|
cmd: "bun run dev",
|
|
expected: 50,
|
|
},
|
|
{
|
|
name: "go test inside go project",
|
|
ws: workspace.WorkspaceInfo{HasGoProject: true},
|
|
cmd: "go test ./...",
|
|
expected: 50,
|
|
},
|
|
{
|
|
name: "cargo check inside rust project",
|
|
ws: workspace.WorkspaceInfo{HasRustProject: true},
|
|
cmd: "cargo check",
|
|
expected: 50,
|
|
},
|
|
{
|
|
name: "pytest inside python project",
|
|
ws: workspace.WorkspaceInfo{HasPythonProject: true},
|
|
cmd: "pytest -v",
|
|
expected: 50,
|
|
},
|
|
{
|
|
name: "just build inside justfile project",
|
|
ws: workspace.WorkspaceInfo{HasJustfile: true},
|
|
cmd: "just build",
|
|
expected: 50,
|
|
},
|
|
{
|
|
name: "kubectl get pods inside k8s workspace",
|
|
ws: workspace.WorkspaceInfo{HasK8s: true},
|
|
cmd: "kubectl get pods",
|
|
expected: 40,
|
|
},
|
|
{
|
|
name: "git push active branch gets double bonus clamped to 100",
|
|
ws: workspace.WorkspaceInfo{HasGit: true, GitBranch: "fix/scoring"},
|
|
cmd: "git push origin fix/scoring",
|
|
expected: 100, // 40 (git push) + 60 (active branch) = 100
|
|
},
|
|
{
|
|
name: "git push other branch gets normal git push bonus only",
|
|
ws: workspace.WorkspaceInfo{HasGit: true, GitBranch: "fix/scoring"},
|
|
cmd: "git push origin fix/alias",
|
|
expected: 40,
|
|
},
|
|
{
|
|
name: "unrelated command gets no bonus",
|
|
ws: workspace.WorkspaceInfo{HasGit: true},
|
|
cmd: "echo hello",
|
|
expected: 0,
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
got := ApplyContextRules(tc.ws, tc.cmd)
|
|
if got != tc.expected {
|
|
t.Errorf("expected bonus %d, got %d for command %q", tc.expected, got, tc.cmd)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestApplyContextRules_Clamping(t *testing.T) {
|
|
ws := workspace.WorkspaceInfo{HasGit: true, HasNodeProject: true, HasMakefile: true}
|
|
// Create rules that sum above 100 and below -100
|
|
rules := []ContextRule{
|
|
&SimpleContextRule{check: func(w workspace.WorkspaceInfo, c string) bool { return true }, bonus: 80},
|
|
&SimpleContextRule{check: func(w workspace.WorkspaceInfo, c string) bool { return true }, bonus: 60},
|
|
}
|
|
got := ApplyCustomContextRules(ws, "any cmd", rules)
|
|
if got != 100 {
|
|
t.Errorf("expected clamp to 100, got %d", got)
|
|
}
|
|
|
|
negRules := []ContextRule{
|
|
&SimpleContextRule{check: func(w workspace.WorkspaceInfo, c string) bool { return true }, bonus: -80},
|
|
&SimpleContextRule{check: func(w workspace.WorkspaceInfo, c string) bool { return true }, bonus: -60},
|
|
}
|
|
gotNeg := ApplyCustomContextRules(ws, "any cmd", negRules)
|
|
if gotNeg != -100 {
|
|
t.Errorf("expected clamp to -100, got %d", gotNeg)
|
|
}
|
|
}
|