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
96 lines
2.3 KiB
Go
96 lines
2.3 KiB
Go
package db_test
|
|
|
|
import (
|
|
"maps"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.kenn.io/agentsview/internal/dbtest"
|
|
)
|
|
|
|
func TestSkippedFiles_RoundTrip(t *testing.T) {
|
|
d := dbtest.OpenTestDB(t)
|
|
|
|
// Initially empty.
|
|
loaded, err := d.LoadSkippedFiles()
|
|
require.NoError(t, err, "LoadSkippedFiles")
|
|
require.Empty(t, loaded)
|
|
|
|
// Persist some entries.
|
|
entries := map[string]int64{
|
|
"/a/b/c.jsonl": 100,
|
|
"/d/e/f.jsonl": 200,
|
|
"/g/h/i.jsonl": 300,
|
|
}
|
|
require.NoError(t, d.ReplaceSkippedFiles(entries))
|
|
|
|
// Load them back.
|
|
loaded, err = d.LoadSkippedFiles()
|
|
require.NoError(t, err, "LoadSkippedFiles")
|
|
assert.True(t, maps.Equal(loaded, entries),
|
|
"loaded map %v, want %v", loaded, entries)
|
|
}
|
|
|
|
func TestSkippedFiles_ReplaceOverwrites(t *testing.T) {
|
|
d := dbtest.OpenTestDB(t)
|
|
|
|
first := map[string]int64{
|
|
"/a.jsonl": 100,
|
|
"/b.jsonl": 200,
|
|
}
|
|
require.NoError(t, d.ReplaceSkippedFiles(first))
|
|
|
|
// Replace with different entries.
|
|
second := map[string]int64{
|
|
"/c.jsonl": 300,
|
|
}
|
|
require.NoError(t, d.ReplaceSkippedFiles(second))
|
|
|
|
loaded, err := d.LoadSkippedFiles()
|
|
require.NoError(t, err, "LoadSkippedFiles")
|
|
require.Len(t, loaded, 1)
|
|
assert.Equal(t, int64(300), loaded["/c.jsonl"])
|
|
}
|
|
|
|
func TestSkippedFiles_DeleteSingle(t *testing.T) {
|
|
d := dbtest.OpenTestDB(t)
|
|
|
|
entries := map[string]int64{
|
|
"/a.jsonl": 100,
|
|
"/b.jsonl": 200,
|
|
}
|
|
require.NoError(t, d.ReplaceSkippedFiles(entries))
|
|
|
|
require.NoError(t, d.DeleteSkippedFile("/a.jsonl"))
|
|
|
|
loaded, err := d.LoadSkippedFiles()
|
|
require.NoError(t, err, "LoadSkippedFiles")
|
|
require.Len(t, loaded, 1)
|
|
_, ok := loaded["/a.jsonl"]
|
|
assert.False(t, ok, "/a.jsonl should have been deleted")
|
|
assert.Equal(t, int64(200), loaded["/b.jsonl"])
|
|
}
|
|
|
|
func TestSkippedFiles_DeleteNonexistent(t *testing.T) {
|
|
d := dbtest.OpenTestDB(t)
|
|
|
|
// Should not error.
|
|
require.NoError(t, d.DeleteSkippedFile("/nope"))
|
|
}
|
|
|
|
func TestSkippedFiles_EmptyReplace(t *testing.T) {
|
|
d := dbtest.OpenTestDB(t)
|
|
|
|
entries := map[string]int64{"/a.jsonl": 100}
|
|
require.NoError(t, d.ReplaceSkippedFiles(entries))
|
|
|
|
// Replace with empty map clears the table.
|
|
require.NoError(t, d.ReplaceSkippedFiles(map[string]int64{}))
|
|
|
|
loaded, err := d.LoadSkippedFiles()
|
|
require.NoError(t, err, "LoadSkippedFiles")
|
|
require.Empty(t, loaded)
|
|
}
|