78 lines
2.1 KiB
Go
78 lines
2.1 KiB
Go
package store
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func auditFixture(t *testing.T) string {
|
|
t.Helper()
|
|
return filepath.Join(t.TempDir(), "audit.jsonl")
|
|
}
|
|
|
|
func sampleSummary(id string) SessionSummary {
|
|
return SessionSummary{
|
|
SessionID: id,
|
|
NextAction: "write plan",
|
|
SuccessCond: "plan committed",
|
|
Outcome: "completed",
|
|
StartedUnix: 1000,
|
|
EndedUnix: 2000,
|
|
SwitchCount: 3,
|
|
Buckets: []BucketTotal{
|
|
{Class: "code", Title: "antidrift", Seconds: 540},
|
|
{Class: "firefox", Title: "docs", Seconds: 120},
|
|
},
|
|
}
|
|
}
|
|
|
|
func TestAppendSessionChainsAndVerifies(t *testing.T) {
|
|
path := auditFixture(t)
|
|
for i := 0; i < 3; i++ {
|
|
if err := AppendSession(path, sampleSummary("session-"+string(rune('a'+i)))); err != nil {
|
|
t.Fatalf("append %d: %v", i, err)
|
|
}
|
|
}
|
|
if err := VerifyChain(path); err != nil {
|
|
t.Fatalf("chain should verify: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestGenesisPrevHashIsZeros(t *testing.T) {
|
|
path := auditFixture(t)
|
|
if err := AppendSession(path, sampleSummary("session-a")); err != nil {
|
|
t.Fatalf("append: %v", err)
|
|
}
|
|
data, _ := os.ReadFile(path)
|
|
if !strings.Contains(string(data), strings.Repeat("0", 64)) {
|
|
t.Fatalf("genesis prev_hash should be 64 zeros, got: %s", data)
|
|
}
|
|
}
|
|
|
|
func TestVerifyEmptyChain(t *testing.T) {
|
|
if err := VerifyChain(auditFixture(t)); err != nil {
|
|
t.Fatalf("empty/missing chain should verify, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestTamperDetected(t *testing.T) {
|
|
path := auditFixture(t)
|
|
_ = AppendSession(path, sampleSummary("session-a"))
|
|
_ = AppendSession(path, sampleSummary("session-b"))
|
|
_ = AppendSession(path, sampleSummary("session-c"))
|
|
|
|
// Corrupt the middle line's switch_count.
|
|
data, _ := os.ReadFile(path)
|
|
lines := strings.Split(strings.TrimRight(string(data), "\n"), "\n")
|
|
lines[1] = strings.Replace(lines[1], `"switch_count":3`, `"switch_count":99`, 1)
|
|
if err := os.WriteFile(path, []byte(strings.Join(lines, "\n")+"\n"), 0o644); err != nil {
|
|
t.Fatalf("rewrite: %v", err)
|
|
}
|
|
err := VerifyChain(path)
|
|
if err == nil || !strings.Contains(err.Error(), "line 2") {
|
|
t.Fatalf("tamper on line 2 should be reported, got %v", err)
|
|
}
|
|
}
|