wehub-resource-sync
63 lines
2.1 KiB
Go
63 lines
2.1 KiB
Go
package evidence
|
|
|
|
import "testing"
|
|
|
|
func TestHasSuccessfulTodoWriteNilLedger(t *testing.T) {
|
|
var l *Ledger
|
|
if l.HasSuccessfulTodoWrite() {
|
|
t.Fatal("nil ledger must report no todo_write")
|
|
}
|
|
}
|
|
|
|
func TestHasSuccessfulTodoWriteMatchesOnlySuccessfulTodoWrite(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
receipt Receipt
|
|
want bool
|
|
}{
|
|
{"successful todo_write", Receipt{ToolName: "todo_write", Success: true}, true},
|
|
{"failed todo_write does not count", Receipt{ToolName: "todo_write", Success: false}, false},
|
|
{"successful non-todo tool does not count", Receipt{ToolName: "write_file", Success: true, Write: true}, false},
|
|
}
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
l := NewLedger()
|
|
l.Record(tc.receipt)
|
|
if got := l.HasSuccessfulTodoWrite(); got != tc.want {
|
|
t.Fatalf("HasSuccessfulTodoWrite() = %v, want %v", got, tc.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestHasSuccessfulTodoProgressReceipt(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
receipts []Receipt
|
|
want bool
|
|
}{
|
|
{"nil", nil, false},
|
|
{"todo only", []Receipt{{ToolName: "todo_write", Success: true}}, false},
|
|
{"read-only context only", []Receipt{{ToolName: "read_file", Success: true, Read: true}}, false},
|
|
{"failed execution only", []Receipt{{ToolName: "bash", Success: false}}, false},
|
|
{"successful command counts", []Receipt{{ToolName: "bash", Success: true}}, true},
|
|
{"complete_step counts", []Receipt{{ToolName: "complete_step", Success: true, Step: "done"}}, true},
|
|
{"todo plus read-only context still does not count", []Receipt{{ToolName: "todo_write", Success: true}, {ToolName: "read_file", Success: true, Read: true}}, false},
|
|
{"todo plus writer counts", []Receipt{{ToolName: "todo_write", Success: true}, {ToolName: "write_file", Success: true, Write: true}}, true},
|
|
}
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
var l *Ledger
|
|
if tc.receipts != nil {
|
|
l = NewLedger()
|
|
for _, r := range tc.receipts {
|
|
l.Record(r)
|
|
}
|
|
}
|
|
if got := l.HasSuccessfulTodoProgressReceipt(); got != tc.want {
|
|
t.Fatalf("HasSuccessfulTodoProgressReceipt() = %v, want %v", got, tc.want)
|
|
}
|
|
})
|
|
}
|
|
}
|