Files
wehub-resource-sync f99010fae1
CI / lint (push) Failing after 1s
CI / frontend (push) Failing after 1s
CI / scripts (push) Failing after 1s
CI / Go Test (ubuntu-latest) (push) Failing after 0s
CI / frontend-node-25 (push) Failing after 1s
CI / docs (push) Failing after 0s
CI / coverage (push) Failing after 0s
CI / e2e (push) Failing after 0s
Docker / build-and-push (push) Failing after 1s
CI / integration (push) Failing after 4m43s
CI / Go Test (windows-latest) (push) Has been cancelled
CI / Desktop Unit Tests (Windows) (push) Has been cancelled
Desktop Artifacts / Desktop Build (Linux (arm64)) (push) Has been cancelled
Desktop Artifacts / Desktop Build (Linux) (push) Has been cancelled
Desktop Artifacts / Desktop Build (Windows) (push) Has been cancelled
Desktop Artifacts (macOS) / Desktop Build (macOS (aarch64)) (push) Has been cancelled
Desktop Artifacts (macOS) / Desktop Build (macOS (x86_64)) (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 12:30:36 +08:00

501 lines
16 KiB
Go

package parser
import (
"context"
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestHermesProviderTranscriptSourceMethods(t *testing.T) {
root := t.TempDir()
jsonlPath := filepath.Join(root, "child.jsonl")
jsonPath := filepath.Join(root, "session_jsononly.json")
writeSourceFile(t, jsonlPath, hermesProviderJSONLFixture("jsonl question"))
writeSourceFile(t, jsonPath, hermesProviderJSONFixture("json question"))
writeSourceFile(t, filepath.Join(root, "scratch.json"), "{}\n")
provider, ok := NewProvider(AgentHermes, ProviderConfig{
Roots: []string{root},
Machine: "devbox",
})
require.True(t, ok)
plan, err := provider.WatchPlan(context.Background())
require.NoError(t, err)
require.Len(t, plan.Roots, 1)
assert.Equal(t, root, plan.Roots[0].Path)
assert.True(t, plan.Roots[0].Recursive)
assert.Equal(t, []string{"state.db", "*.jsonl", "session_*.json"}, plan.Roots[0].IncludeGlobs)
discovered, err := provider.Discover(context.Background())
require.NoError(t, err)
require.Len(t, discovered, 2)
assert.ElementsMatch(t, []string{jsonlPath, jsonPath}, []string{
discovered[0].DisplayPath,
discovered[1].DisplayPath,
})
found, ok, err := provider.FindSource(context.Background(), FindSourceRequest{
FullSessionID: "remote~hermes:child",
})
require.NoError(t, err)
require.True(t, ok)
assert.Equal(t, jsonlPath, found.DisplayPath)
found, ok, err = provider.FindSource(context.Background(), FindSourceRequest{
RawSessionID: "jsononly",
})
require.NoError(t, err)
require.True(t, ok)
assert.Equal(t, jsonPath, found.DisplayPath)
fingerprint, err := provider.Fingerprint(context.Background(), found)
require.NoError(t, err)
assert.Equal(t, jsonPath, fingerprint.Key)
assert.Positive(t, fingerprint.Size)
assert.Positive(t, fingerprint.MTimeNS)
assert.NotEmpty(t, fingerprint.Hash)
changed, err := provider.SourcesForChangedPath(
context.Background(),
ChangedPathRequest{Path: jsonlPath, EventKind: "write", WatchRoot: root},
)
require.NoError(t, err)
require.Len(t, changed, 1)
assert.Equal(t, jsonlPath, changed[0].DisplayPath)
require.NoError(t, os.Remove(jsonlPath))
changed, err = provider.SourcesForChangedPath(
context.Background(),
ChangedPathRequest{Path: jsonlPath, EventKind: "remove", WatchRoot: root},
)
require.NoError(t, err)
require.Len(t, changed, 1)
assert.Equal(t, jsonlPath, changed[0].DisplayPath)
ignored, err := provider.SourcesForChangedPath(
context.Background(),
ChangedPathRequest{
Path: filepath.Join(root, "scratch.json"),
EventKind: "write",
WatchRoot: root,
},
)
require.NoError(t, err)
assert.Empty(t, ignored)
}
func TestHermesProviderStateDBSourceMethods(t *testing.T) {
root := t.TempDir()
sessionsDir := filepath.Join(root, "sessions")
require.NoError(t, os.MkdirAll(sessionsDir, 0o755))
createHermesStateDB(t, root)
transcriptPath := filepath.Join(sessionsDir, "session_child.json")
writeSourceFile(t, transcriptPath, hermesProviderJSONFixture("transcript question"))
stateDB := filepath.Join(root, "state.db")
provider, ok := NewProvider(AgentHermes, ProviderConfig{
Roots: []string{root},
Machine: "devbox",
})
require.True(t, ok)
plan, err := provider.WatchPlan(context.Background())
require.NoError(t, err)
require.Len(t, plan.Roots, 2)
assert.Equal(t, root, plan.Roots[0].Path)
assert.False(t, plan.Roots[0].Recursive)
assert.Equal(t, []string{"state.db"}, plan.Roots[0].IncludeGlobs)
assert.Equal(t, sessionsDir, plan.Roots[1].Path)
assert.True(t, plan.Roots[1].Recursive)
assert.Equal(t, []string{"*.jsonl", "session_*.json"}, plan.Roots[1].IncludeGlobs)
discovered, err := provider.Discover(context.Background())
require.NoError(t, err)
require.Len(t, discovered, 1)
assert.Equal(t, stateDB, discovered[0].DisplayPath)
found, ok, err := provider.FindSource(context.Background(), FindSourceRequest{
FullSessionID: "remote~hermes:child",
})
require.NoError(t, err)
require.True(t, ok)
assert.Equal(t, stateDB, found.DisplayPath)
stateInfo, err := os.Stat(stateDB)
require.NoError(t, err)
transcriptInfo, err := os.Stat(transcriptPath)
require.NoError(t, err)
fingerprint, err := provider.Fingerprint(context.Background(), found)
require.NoError(t, err)
assert.Equal(t, stateDB, fingerprint.Key)
assert.Equal(t, stateInfo.Size()+transcriptInfo.Size(), fingerprint.Size)
assert.Equal(
t,
max(stateInfo.ModTime().UnixNano(), transcriptInfo.ModTime().UnixNano()),
fingerprint.MTimeNS,
)
assert.NotEmpty(t, fingerprint.Hash)
for _, tc := range []struct {
name string
path string
}{
{name: "state db", path: stateDB},
{name: "archive transcript", path: transcriptPath},
} {
t.Run(tc.name, func(t *testing.T) {
changed, err := provider.SourcesForChangedPath(
context.Background(),
ChangedPathRequest{Path: tc.path, EventKind: "write", WatchRoot: root},
)
require.NoError(t, err)
require.Len(t, changed, 1)
assert.Equal(t, stateDB, changed[0].DisplayPath)
})
}
changed, err := provider.SourcesForChangedPath(
context.Background(),
ChangedPathRequest{Path: stateDB, EventKind: "write", WatchRoot: root},
)
require.NoError(t, err)
require.Len(t, changed, 1)
assert.Equal(t, stateDB, changed[0].DisplayPath)
require.NoError(t, os.Remove(transcriptPath))
changed, err = provider.SourcesForChangedPath(
context.Background(),
ChangedPathRequest{Path: transcriptPath, EventKind: "remove", WatchRoot: sessionsDir},
)
require.NoError(t, err)
require.Len(t, changed, 1)
assert.Equal(t, stateDB, changed[0].DisplayPath)
require.NoError(t, os.Remove(stateDB))
changed, err = provider.SourcesForChangedPath(
context.Background(),
ChangedPathRequest{Path: stateDB, EventKind: "remove", WatchRoot: root},
)
require.NoError(t, err)
require.Len(t, changed, 1)
assert.Equal(t, stateDB, changed[0].DisplayPath)
}
func TestHermesProviderArchiveWatchRoots(t *testing.T) {
root := t.TempDir()
sessionsDir := filepath.Join(root, "sessions")
require.NoError(t, os.MkdirAll(sessionsDir, 0o755))
createHermesStateDB(t, root)
stateDB := filepath.Join(root, "state.db")
for _, tc := range []struct {
name string
configRoot string
}{
{name: "archive parent", configRoot: root},
{name: "sessions directory", configRoot: sessionsDir},
{name: "state db file", configRoot: stateDB},
} {
t.Run(tc.name, func(t *testing.T) {
provider, ok := NewProvider(AgentHermes, ProviderConfig{
Roots: []string{tc.configRoot},
Machine: "devbox",
})
require.True(t, ok)
plan, err := provider.WatchPlan(context.Background())
require.NoError(t, err)
require.Len(t, plan.Roots, 2)
assert.Equal(t, root, plan.Roots[0].Path)
assert.False(t, plan.Roots[0].Recursive)
assert.Equal(t, []string{"state.db"}, plan.Roots[0].IncludeGlobs)
assert.Equal(t, sessionsDir, plan.Roots[1].Path)
assert.True(t, plan.Roots[1].Recursive)
assert.Equal(t, []string{"*.jsonl", "session_*.json"}, plan.Roots[1].IncludeGlobs)
changed, err := provider.SourcesForChangedPath(
context.Background(),
ChangedPathRequest{Path: stateDB, EventKind: "write", WatchRoot: root},
)
require.NoError(t, err)
require.Len(t, changed, 1)
assert.Equal(t, stateDB, changed[0].DisplayPath)
})
}
}
func TestHermesProviderArchiveWatchRootsBeforeArchiveComplete(t *testing.T) {
t.Run("state db exists before sessions directory", func(t *testing.T) {
root := t.TempDir()
createHermesStateDB(t, root)
stateDB := filepath.Join(root, "state.db")
sessionsDir := filepath.Join(root, "sessions")
provider, ok := NewProvider(AgentHermes, ProviderConfig{
Roots: []string{root},
Machine: "devbox",
})
require.True(t, ok)
plan, err := provider.WatchPlan(context.Background())
require.NoError(t, err)
require.Len(t, plan.Roots, 2)
assert.Equal(t, root, plan.Roots[0].Path)
assert.False(t, plan.Roots[0].Recursive)
assert.Equal(t, []string{"state.db"}, plan.Roots[0].IncludeGlobs)
assert.Equal(t, sessionsDir, plan.Roots[1].Path)
assert.True(t, plan.Roots[1].Recursive)
assert.Equal(t, []string{"*.jsonl", "session_*.json"}, plan.Roots[1].IncludeGlobs)
changed, err := provider.SourcesForChangedPath(
context.Background(),
ChangedPathRequest{Path: stateDB, EventKind: "write", WatchRoot: root},
)
require.NoError(t, err)
require.Len(t, changed, 1)
assert.Equal(t, stateDB, changed[0].DisplayPath)
})
t.Run("direct state db root before file exists", func(t *testing.T) {
root := t.TempDir()
stateDB := filepath.Join(root, "state.db")
sessionsDir := filepath.Join(root, "sessions")
provider, ok := NewProvider(AgentHermes, ProviderConfig{
Roots: []string{stateDB},
Machine: "devbox",
})
require.True(t, ok)
plan, err := provider.WatchPlan(context.Background())
require.NoError(t, err)
require.Len(t, plan.Roots, 2)
assert.Equal(t, root, plan.Roots[0].Path)
assert.False(t, plan.Roots[0].Recursive)
assert.Equal(t, []string{"state.db"}, plan.Roots[0].IncludeGlobs)
assert.Equal(t, sessionsDir, plan.Roots[1].Path)
assert.True(t, plan.Roots[1].Recursive)
assert.Equal(t, []string{"*.jsonl", "session_*.json"}, plan.Roots[1].IncludeGlobs)
createHermesStateDB(t, root)
changed, err := provider.SourcesForChangedPath(
context.Background(),
ChangedPathRequest{Path: stateDB, EventKind: "write", WatchRoot: root},
)
require.NoError(t, err)
require.Len(t, changed, 1)
assert.Equal(t, stateDB, changed[0].DisplayPath)
})
t.Run("sessions directory root before state db exists", func(t *testing.T) {
root := t.TempDir()
stateDB := filepath.Join(root, "state.db")
sessionsDir := filepath.Join(root, "sessions")
require.NoError(t, os.MkdirAll(sessionsDir, 0o755))
provider, ok := NewProvider(AgentHermes, ProviderConfig{
Roots: []string{sessionsDir},
Machine: "devbox",
})
require.True(t, ok)
plan, err := provider.WatchPlan(context.Background())
require.NoError(t, err)
require.Len(t, plan.Roots, 2)
assert.Equal(t, root, plan.Roots[0].Path)
assert.False(t, plan.Roots[0].Recursive)
assert.Equal(t, []string{"state.db"}, plan.Roots[0].IncludeGlobs)
assert.Equal(t, sessionsDir, plan.Roots[1].Path)
assert.True(t, plan.Roots[1].Recursive)
assert.Equal(t, []string{"*.jsonl", "session_*.json"}, plan.Roots[1].IncludeGlobs)
createHermesStateDB(t, root)
changed, err := provider.SourcesForChangedPath(
context.Background(),
ChangedPathRequest{Path: stateDB, EventKind: "write", WatchRoot: root},
)
require.NoError(t, err)
require.Len(t, changed, 1)
assert.Equal(t, stateDB, changed[0].DisplayPath)
})
}
func TestHermesProviderParse(t *testing.T) {
root := t.TempDir()
sourcePath := filepath.Join(root, "child.jsonl")
writeSourceFile(t, sourcePath, hermesProviderJSONLFixture("parse question"))
provider, ok := NewProvider(AgentHermes, ProviderConfig{
Roots: []string{root},
Machine: "devbox",
})
require.True(t, ok)
sources, err := provider.Discover(context.Background())
require.NoError(t, err)
require.Len(t, sources, 1)
outcome, err := provider.Parse(context.Background(), ParseRequest{
Source: sources[0],
Fingerprint: SourceFingerprint{Key: sourcePath, Hash: "abc123"},
})
require.NoError(t, err)
require.True(t, outcome.ResultSetComplete)
require.False(t, outcome.ForceReplace)
require.Len(t, outcome.Results, 1)
result := outcome.Results[0]
assert.Equal(t, DataVersionCurrent, result.DataVersion)
assert.Equal(t, "hermes:child", result.Result.Session.ID)
assert.Equal(t, AgentHermes, result.Result.Session.Agent)
assert.Equal(t, "devbox", result.Result.Session.Machine)
assert.Equal(t, sourcePath, result.Result.Session.File.Path)
assert.Equal(t, "abc123", result.Result.Session.File.Hash)
assert.Equal(t, "parse question", result.Result.Session.FirstMessage)
assert.Len(t, result.Result.Messages, 2)
}
func TestHermesProviderParseStateDB(t *testing.T) {
root := t.TempDir()
sessionsDir := filepath.Join(root, "sessions")
require.NoError(t, os.MkdirAll(sessionsDir, 0o755))
createHermesStateDB(t, root)
transcriptPath := filepath.Join(sessionsDir, "session_child.json")
writeSourceFile(
t,
transcriptPath,
hermesProviderJSONFixture("archive transcript"),
)
stateDB := filepath.Join(root, "state.db")
provider, ok := NewProvider(AgentHermes, ProviderConfig{
Roots: []string{root},
Machine: "devbox",
})
require.True(t, ok)
sources, err := provider.Discover(context.Background())
require.NoError(t, err)
require.Len(t, sources, 1)
outcome, err := provider.Parse(context.Background(), ParseRequest{
Source: sources[0],
Fingerprint: SourceFingerprint{Key: stateDB, Hash: "archive-hash"},
})
require.NoError(t, err)
require.True(t, outcome.ResultSetComplete)
require.True(t, outcome.ForceReplace)
require.Len(t, outcome.Results, 1)
result := outcome.Results[0]
assert.Equal(t, DataVersionCurrent, result.DataVersion)
assert.Equal(t, "hermes:child", result.Result.Session.ID)
assert.Equal(t, "hermes:parent", result.Result.Session.ParentSessionID)
assert.Equal(t, RelContinuation, result.Result.Session.RelationshipType)
assert.Equal(t, "Child Session", result.Result.Session.SessionName)
assert.Equal(t, "hermes-state-db", result.Result.Session.SourceVersion)
assert.Equal(t, "devbox", result.Result.Session.Machine)
require.Len(t, result.Result.UsageEvents, 1)
assert.Len(t, result.Result.Messages, 2)
// The provider reproduces the legacy engine's stampHermesArchiveResults:
// every archive session's stored file identity is the state.db path with
// the aggregate (state.db plus transcripts) size and mtime, so a
// transcript-only change still refreshes the archive's freshness.
stateInfo, err := os.Stat(stateDB)
require.NoError(t, err)
transcriptInfo, err := os.Stat(transcriptPath)
require.NoError(t, err)
assert.Equal(t, stateDB, result.Result.Session.File.Path)
assert.Equal(
t,
stateInfo.Size()+transcriptInfo.Size(),
result.Result.Session.File.Size,
)
assert.Equal(
t,
max(stateInfo.ModTime().UnixNano(), transcriptInfo.ModTime().UnixNano()),
result.Result.Session.File.Mtime,
)
}
func TestHermesProviderFindSourceDoesNotReturnStateDBForMissingRawID(t *testing.T) {
root := t.TempDir()
require.NoError(t, os.MkdirAll(filepath.Join(root, "sessions"), 0o755))
createHermesStateDB(t, root)
provider, ok := NewProvider(AgentHermes, ProviderConfig{
Roots: []string{root},
Machine: "devbox",
})
require.True(t, ok)
source, ok, err := provider.FindSource(context.Background(), FindSourceRequest{
RawSessionID: "missing-valid-id",
})
require.NoError(t, err)
assert.False(t, ok)
assert.Empty(t, source)
}
func TestHermesProviderFindSourceFallsBackToTranscriptWhenStateDBUnreadable(t *testing.T) {
root := t.TempDir()
sessionsDir := filepath.Join(root, "sessions")
require.NoError(t, os.MkdirAll(sessionsDir, 0o755))
// A present-but-unreadable state.db: hermesStateDBHasSession opens it
// lazily, then errors on the first query because the bytes are not a
// SQLite database. parseArchive logs and falls back to transcripts in this
// case, so FindSource must do the same rather than aborting the lookup.
stateDB := filepath.Join(root, "state.db")
writeSourceFile(t, stateDB, "not a sqlite database")
transcriptPath := filepath.Join(sessionsDir, "freshchild.jsonl")
writeSourceFile(t, transcriptPath, hermesProviderJSONLFixture("transcript question"))
provider, ok := NewProvider(AgentHermes, ProviderConfig{
Roots: []string{root},
Machine: "devbox",
})
require.True(t, ok)
source, ok, err := provider.FindSource(context.Background(), FindSourceRequest{
RawSessionID: "freshchild",
})
require.NoError(t, err, "unreadable state.db must not abort transcript lookup")
require.True(t, ok, "valid transcript next to a bad state.db must be found")
assert.Equal(t, transcriptPath, source.DisplayPath)
}
func hermesProviderJSONLFixture(firstMessage string) string {
return `{"role":"session_meta","platform":"cli","timestamp":"2026-05-14T10:00:00.000000"}` + "\n" +
`{"role":"user","content":"` + firstMessage + `","timestamp":"2026-05-14T10:01:00.000000"}` + "\n" +
`{"role":"assistant","content":"Done.","timestamp":"2026-05-14T10:02:00.000000"}` + "\n"
}
func hermesProviderJSONFixture(firstMessage string) string {
return `{
"platform":"cli",
"session_start":"2026-05-14T10:00:00Z",
"last_updated":"2026-05-14T10:02:00Z",
"messages":[
{"role":"user","content":"` + firstMessage + `","timestamp":"2026-05-14T10:01:00Z"},
{"role":"assistant","content":"Done.","timestamp":"2026-05-14T10:02:00Z"}
]
}`
}