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
261 lines
8.4 KiB
Go
261 lines
8.4 KiB
Go
package parser
|
|
|
|
import (
|
|
"context"
|
|
"crypto/sha256"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAmpProviderSourceMethods(t *testing.T) {
|
|
root := t.TempDir()
|
|
threadID := "T-019ca26f-aaaa-bbbb-cccc-dddddddddddd"
|
|
sourcePath := filepath.Join(root, threadID+".json")
|
|
writeSourceFile(t, sourcePath, ampProviderFixture(threadID))
|
|
writeSourceFile(t, filepath.Join(root, "T-.json"), "{}\n")
|
|
writeSourceFile(t, filepath.Join(root, "notes.json"), "{}\n")
|
|
writeSourceFile(t, filepath.Join(root, "nested", threadID+".json"), "{}\n")
|
|
|
|
provider, ok := NewProvider(AgentAmp, ProviderConfig{
|
|
Roots: []string{root},
|
|
Machine: "devbox",
|
|
})
|
|
require.True(t, ok)
|
|
|
|
discovered, err := provider.Discover(context.Background())
|
|
require.NoError(t, err)
|
|
require.Len(t, discovered, 1)
|
|
assert.Equal(t, AgentAmp, discovered[0].Provider)
|
|
assert.Equal(t, sourcePath, discovered[0].DisplayPath)
|
|
|
|
found, ok, err := provider.FindSource(context.Background(), FindSourceRequest{
|
|
FullSessionID: "host~amp:" + threadID,
|
|
})
|
|
require.NoError(t, err)
|
|
require.True(t, ok)
|
|
assert.Equal(t, sourcePath, found.DisplayPath)
|
|
|
|
require.NoError(t, os.Remove(sourcePath))
|
|
changed, err := provider.SourcesForChangedPath(
|
|
context.Background(),
|
|
ChangedPathRequest{Path: sourcePath, EventKind: "remove", WatchRoot: root},
|
|
)
|
|
require.NoError(t, err)
|
|
require.Len(t, changed, 1)
|
|
assert.Equal(t, sourcePath, changed[0].DisplayPath)
|
|
}
|
|
|
|
func TestAmpProviderSourceMethodsFollowSymlinkedSessionFile(t *testing.T) {
|
|
root := t.TempDir()
|
|
targetDir := t.TempDir()
|
|
threadID := "T-019ca26f-aaaa-bbbb-cccc-dddddddddddd"
|
|
targetPath := filepath.Join(targetDir, threadID+".json")
|
|
sourcePath := filepath.Join(root, threadID+".json")
|
|
writeSourceFile(t, targetPath, ampProviderFixture(threadID))
|
|
if err := os.Symlink(targetPath, sourcePath); err != nil {
|
|
t.Skipf("symlink not supported: %v", err)
|
|
}
|
|
|
|
provider, ok := NewProvider(AgentAmp, ProviderConfig{
|
|
Roots: []string{root},
|
|
Machine: "devbox",
|
|
})
|
|
require.True(t, ok)
|
|
|
|
discovered, err := provider.Discover(context.Background())
|
|
require.NoError(t, err)
|
|
require.Len(t, discovered, 1)
|
|
assert.Equal(t, sourcePath, discovered[0].DisplayPath)
|
|
|
|
found, ok, err := provider.FindSource(context.Background(), FindSourceRequest{
|
|
FullSessionID: "host~amp:" + threadID,
|
|
})
|
|
require.NoError(t, err)
|
|
require.True(t, ok)
|
|
assert.Equal(t, sourcePath, found.DisplayPath)
|
|
|
|
changed, err := provider.SourcesForChangedPath(
|
|
context.Background(),
|
|
ChangedPathRequest{Path: sourcePath, EventKind: "write", WatchRoot: root},
|
|
)
|
|
require.NoError(t, err)
|
|
require.Len(t, changed, 1)
|
|
assert.Equal(t, sourcePath, changed[0].DisplayPath)
|
|
}
|
|
|
|
func TestAmpProviderParse(t *testing.T) {
|
|
root := t.TempDir()
|
|
threadID := "T-019ca26f-aaaa-bbbb-cccc-dddddddddddd"
|
|
sourcePath := filepath.Join(root, threadID+".json")
|
|
content := ampProviderFixture(threadID)
|
|
writeSourceFile(t, sourcePath, content)
|
|
|
|
provider, ok := NewProvider(AgentAmp, 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)
|
|
|
|
fingerprint, err := provider.Fingerprint(context.Background(), sources[0])
|
|
require.NoError(t, err)
|
|
|
|
outcome, err := provider.Parse(context.Background(), ParseRequest{
|
|
Source: sources[0],
|
|
Fingerprint: fingerprint,
|
|
})
|
|
require.NoError(t, err)
|
|
require.True(t, outcome.ResultSetComplete)
|
|
require.Len(t, outcome.Results, 1)
|
|
assert.Equal(t, DataVersionCurrent, outcome.Results[0].DataVersion)
|
|
assert.Equal(t, "amp:"+threadID, outcome.Results[0].Result.Session.ID)
|
|
assert.Equal(t, "amp-project", outcome.Results[0].Result.Session.Project)
|
|
assert.Equal(t, "devbox", outcome.Results[0].Result.Session.Machine)
|
|
assert.Equal(t,
|
|
fmt.Sprintf("%x", sha256.Sum256([]byte(content))),
|
|
outcome.Results[0].Result.Session.File.Hash,
|
|
)
|
|
assert.Len(t, outcome.Results[0].Result.Messages, 2)
|
|
}
|
|
|
|
func TestZencoderProviderSourceMethods(t *testing.T) {
|
|
root := t.TempDir()
|
|
sourcePath := filepath.Join(root, "abc-def-123.jsonl")
|
|
writeSourceFile(t, sourcePath, zencoderProviderFixture("abc-def-123"))
|
|
writeSourceFile(t, filepath.Join(root, "notes.txt"), "{}\n")
|
|
writeSourceFile(t, filepath.Join(root, "nested", "abc-def-123.jsonl"), "{}\n")
|
|
|
|
provider, ok := NewProvider(AgentZencoder, ProviderConfig{
|
|
Roots: []string{root},
|
|
Machine: "devbox",
|
|
})
|
|
require.True(t, ok)
|
|
|
|
discovered, err := provider.Discover(context.Background())
|
|
require.NoError(t, err)
|
|
require.Len(t, discovered, 1)
|
|
assert.Equal(t, AgentZencoder, discovered[0].Provider)
|
|
assert.Equal(t, sourcePath, discovered[0].DisplayPath)
|
|
|
|
found, ok, err := provider.FindSource(context.Background(), FindSourceRequest{
|
|
FullSessionID: "host~zencoder:abc-def-123",
|
|
})
|
|
require.NoError(t, err)
|
|
require.True(t, ok)
|
|
assert.Equal(t, sourcePath, found.DisplayPath)
|
|
|
|
require.NoError(t, os.Remove(sourcePath))
|
|
changed, err := provider.SourcesForChangedPath(
|
|
context.Background(),
|
|
ChangedPathRequest{Path: sourcePath, EventKind: "remove", WatchRoot: root},
|
|
)
|
|
require.NoError(t, err)
|
|
require.Len(t, changed, 1)
|
|
assert.Equal(t, sourcePath, changed[0].DisplayPath)
|
|
}
|
|
|
|
func TestZencoderProviderSourceMethodsFollowSymlinkedSessionFile(t *testing.T) {
|
|
root := t.TempDir()
|
|
targetDir := t.TempDir()
|
|
targetPath := filepath.Join(targetDir, "abc-def-123.jsonl")
|
|
sourcePath := filepath.Join(root, "abc-def-123.jsonl")
|
|
writeSourceFile(t, targetPath, zencoderProviderFixture("abc-def-123"))
|
|
if err := os.Symlink(targetPath, sourcePath); err != nil {
|
|
t.Skipf("symlink not supported: %v", err)
|
|
}
|
|
|
|
provider, ok := NewProvider(AgentZencoder, ProviderConfig{
|
|
Roots: []string{root},
|
|
Machine: "devbox",
|
|
})
|
|
require.True(t, ok)
|
|
|
|
discovered, err := provider.Discover(context.Background())
|
|
require.NoError(t, err)
|
|
require.Len(t, discovered, 1)
|
|
assert.Equal(t, sourcePath, discovered[0].DisplayPath)
|
|
|
|
found, ok, err := provider.FindSource(context.Background(), FindSourceRequest{
|
|
FullSessionID: "host~zencoder:abc-def-123",
|
|
})
|
|
require.NoError(t, err)
|
|
require.True(t, ok)
|
|
assert.Equal(t, sourcePath, found.DisplayPath)
|
|
|
|
changed, err := provider.SourcesForChangedPath(
|
|
context.Background(),
|
|
ChangedPathRequest{Path: sourcePath, EventKind: "write", WatchRoot: root},
|
|
)
|
|
require.NoError(t, err)
|
|
require.Len(t, changed, 1)
|
|
assert.Equal(t, sourcePath, changed[0].DisplayPath)
|
|
}
|
|
|
|
func TestZencoderProviderParse(t *testing.T) {
|
|
root := t.TempDir()
|
|
sourcePath := filepath.Join(root, "abc-def-123.jsonl")
|
|
content := zencoderProviderFixture("abc-def-123")
|
|
writeSourceFile(t, sourcePath, content)
|
|
|
|
provider, ok := NewProvider(AgentZencoder, 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)
|
|
|
|
fingerprint, err := provider.Fingerprint(context.Background(), sources[0])
|
|
require.NoError(t, err)
|
|
|
|
outcome, err := provider.Parse(context.Background(), ParseRequest{
|
|
Source: sources[0],
|
|
Fingerprint: fingerprint,
|
|
})
|
|
require.NoError(t, err)
|
|
require.True(t, outcome.ResultSetComplete)
|
|
require.Len(t, outcome.Results, 1)
|
|
assert.Equal(t, DataVersionCurrent, outcome.Results[0].DataVersion)
|
|
assert.Equal(t, "zencoder:abc-def-123", outcome.Results[0].Result.Session.ID)
|
|
assert.Equal(t, "sample_project", outcome.Results[0].Result.Session.Project)
|
|
assert.Equal(t, "devbox", outcome.Results[0].Result.Session.Machine)
|
|
assert.Equal(t,
|
|
fmt.Sprintf("%x", sha256.Sum256([]byte(content))),
|
|
outcome.Results[0].Result.Session.File.Hash,
|
|
)
|
|
assert.Len(t, outcome.Results[0].Result.Messages, 3)
|
|
}
|
|
|
|
func ampProviderFixture(threadID string) string {
|
|
return `{
|
|
"v": 1,
|
|
"id": "` + threadID + `",
|
|
"created": 1704067200000,
|
|
"title": "Migrate database schema",
|
|
"messages": [
|
|
{"role": "user", "content": [{"type": "text", "text": "Migrate the DB schema."}]},
|
|
{"role": "assistant", "content": [{"type": "text", "text": "Sure, I will help."}]}
|
|
],
|
|
"env": {"initial": {"trees": [{"displayName": "amp-project"}]}},
|
|
"meta": {"traces": []}
|
|
}`
|
|
}
|
|
|
|
func zencoderProviderFixture(sessionID string) string {
|
|
return strings.Join([]string{
|
|
`{"id":"` + sessionID + `","createdAt":"2024-01-01T00:00:00Z","updatedAt":"2024-01-01T00:01:00Z"}`,
|
|
`{"role":"system","content":"Working directory: /Users/alice/code/sample-project"}`,
|
|
`{"role":"user","content":[{"type":"text","text":"hello"}]}`,
|
|
`{"role":"assistant","content":[{"type":"text","text":"OK."}]}`,
|
|
}, "\n")
|
|
}
|