Something went wrong. Try again.
Monorepo for Tangled
Something went wrong. Try again.
8.7 kB · 272 lines
Go
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273package db
import ( "context" "path/filepath" "slices" "testing"
"github.com/bluesky-social/indigo/atproto/syntax")
func newACLTestDB(t *testing.T) *DB { t.Helper() d, err := Setup(context.Background(), filepath.Join(t.TempDir(), "knot.db")) if err != nil { t.Fatalf("Setup: %v", err) } return d}
func countMembers(t *testing.T, d *DB, where string, args ...any) int { t.Helper() var n int if err := d.QueryRow("select count(1) from knot_members where "+where, args...).Scan(&n); err != nil { t.Fatalf("count members: %v", err) } return n}
func seedLegacyMember(t *testing.T, d *DB, owner, rkey, subject string) { t.Helper() if _, err := d.Exec( `insert into knot_members (did, rkey, subject) values (?, ?, ?)`, owner, rkey, subject, ); err != nil { t.Fatalf("seed legacy member: %v", err) }}
func TestAddKnotMemberDirect_IdempotentUnderPartialUnique(t *testing.T) { d := newACLTestDB(t) owner := syntax.DID("did:plc:akshay") subject := syntax.DID("did:plc:boltless")
if err := AddKnotMemberDirect(d, owner, subject); err != nil { t.Fatalf("first add: %v", err) } if err := AddKnotMemberDirect(d, owner, subject); err != nil { t.Fatalf("second add: %v", err) }
if got := countMembers(t, d, "subject = ? and rkey is null", subject); got != 1 { t.Errorf("direct rows = %d, want 1", got) }}
func TestRemoveKnotMemberDirect_PreservesLegacyRow(t *testing.T) { d := newACLTestDB(t) owner := syntax.DID("did:plc:akshay") subject := syntax.DID("did:plc:boltless")
seedLegacyMember(t, d, owner.String(), "legacy-rk", subject.String()) if err := AddKnotMemberDirect(d, owner, subject); err != nil { t.Fatalf("direct add: %v", err) } if got := countMembers(t, d, "subject = ?", subject.String()); got != 2 { t.Fatalf("rows = %d, want 2 legacy plus direct", got) }
if err := RemoveKnotMemberDirect(d, subject); err != nil { t.Fatalf("remove direct: %v", err) } if got := countMembers(t, d, "subject = ? and rkey is null", subject.String()); got != 0 { t.Errorf("direct rows after remove = %d, want 0", got) } if got := countMembers(t, d, "subject = ? and rkey = 'legacy-rk'", subject.String()); got != 1 { t.Errorf("legacy row = %d, want 1 preserved", got) }}
func TestRemoveKnotMemberBySubject_RemovesAllRows(t *testing.T) { d := newACLTestDB(t) owner := syntax.DID("did:plc:akshay") subject := syntax.DID("did:plc:boltless")
seedLegacyMember(t, d, owner.String(), "legacy-rk", subject.String()) if err := AddKnotMemberDirect(d, owner, subject); err != nil { t.Fatalf("direct add: %v", err) }
if err := RemoveKnotMemberBySubject(d, subject); err != nil { t.Fatalf("remove by subject: %v", err) } if got := countMembers(t, d, "subject = ?", subject.String()); got != 0 { t.Errorf("rows after remove = %d, want 0", got) }}
func TestCollaborators_AddListRemoveScopedByRepo(t *testing.T) { d := newACLTestDB(t) owner := syntax.DID("did:plc:akshay") subject := syntax.DID("did:plc:boltless") repoA := syntax.DID("did:plc:limpet") repoB := syntax.DID("did:plc:scallop")
for _, repo := range []syntax.DID{repoA, repoB} { if err := AddCollaborator(d, Collaborator{RepoDid: repo, Subject: subject, AddedBy: owner}); err != nil { t.Fatalf("add collaborator on %s: %v", repo, err) } }
listA, _, err := ListCollaborators(d, repoA, ListPage{Limit: ListMaxLimit}) if err != nil { t.Fatalf("list A: %v", err) } if len(listA) != 1 || listA[0].Subject != subject || listA[0].RepoDid != repoA { t.Fatalf("repoA collaborators = %+v, want one boltless on limpet", listA) }
if err := RemoveCollaborator(d, repoA, subject); err != nil { t.Fatalf("remove on A: %v", err) } if listA, _, _ = ListCollaborators(d, repoA, ListPage{Limit: ListMaxLimit}); len(listA) != 0 { t.Errorf("repoA after remove = %d, want 0", len(listA)) } if listB, _, _ := ListCollaborators(d, repoB, ListPage{Limit: ListMaxLimit}); len(listB) != 1 { t.Errorf("repoB after removing from A = %d, want 1 scoped", len(listB)) }}
func TestAddCollaborator_IdempotentUnderUnique(t *testing.T) { d := newACLTestDB(t) owner := syntax.DID("did:plc:akshay") subject := syntax.DID("did:plc:boltless") repo := syntax.DID("did:plc:limpet")
if err := AddCollaborator(d, Collaborator{RepoDid: repo, Subject: subject, AddedBy: owner}); err != nil { t.Fatalf("first add: %v", err) } if err := AddCollaborator(d, Collaborator{RepoDid: repo, Subject: subject, AddedBy: owner}); err != nil { t.Fatalf("second add: %v", err) }
list, _, err := ListCollaborators(d, repo, ListPage{Limit: ListMaxLimit}) if err != nil { t.Fatalf("list: %v", err) } if len(list) != 1 { t.Errorf("rows = %d, want 1", len(list)) }}
func TestListKnotMembers_DedupsLegacyAndDirect(t *testing.T) { d := newACLTestDB(t) owner := syntax.DID("did:plc:akshay") subject := syntax.DID("did:plc:boltless")
seedLegacyMember(t, d, owner.String(), "legacy-rk", subject.String()) if err := AddKnotMemberDirect(d, owner, subject); err != nil { t.Fatalf("direct add: %v", err) } if got := countMembers(t, d, "subject = ?", subject.String()); got != 2 { t.Fatalf("raw rows = %d, want 2 (legacy + direct)", got) }
members, next, err := ListKnotMembers(d, ListPage{Limit: ListMaxLimit}) if err != nil { t.Fatalf("list: %v", err) } if len(members) != 1 { t.Fatalf("members = %d, want 1; legacy and direct rows for one subject must collapse", len(members)) } if members[0].Subject != subject { t.Errorf("subject = %s, want %s", members[0].Subject, subject) } if next != nil { t.Errorf("cursor = %v, want nil for a complete page", *next) }}
func TestListKnotMembers_OrderAndKeyset(t *testing.T) { d := newACLTestDB(t) owner := syntax.DID("did:plc:akshay") for _, s := range []string{"did:plc:limpet", "did:plc:whelk", "did:plc:scallop"} { if err := AddKnotMemberDirect(d, owner, syntax.DID(s)); err != nil { t.Fatalf("add %s: %v", s, err) } }
asc, _, err := ListKnotMembers(d, ListPage{Limit: ListMaxLimit, Desc: false}) if err != nil { t.Fatalf("asc: %v", err) } if !slices.IsSortedFunc(asc, func(a, b KnotMember) int { return a.Id - b.Id }) { t.Errorf("asc not ascending by id: %+v", asc) }
desc, _, err := ListKnotMembers(d, ListPage{Limit: ListMaxLimit, Desc: true}) if err != nil { t.Fatalf("desc: %v", err) } if !slices.IsSortedFunc(desc, func(a, b KnotMember) int { return b.Id - a.Id }) { t.Errorf("desc not descending by id: %+v", desc) }
page1, next, err := ListKnotMembers(d, ListPage{Limit: 2, Desc: false}) if err != nil { t.Fatalf("page1: %v", err) } if len(page1) != 2 || next == nil { t.Fatalf("page1 = %d next=%v, want 2 rows and a cursor", len(page1), next) } page2, next2, err := ListKnotMembers(d, ListPage{Limit: 2, Cursor: next, Desc: false}) if err != nil { t.Fatalf("page2: %v", err) } if len(page2) != 1 || next2 != nil { t.Fatalf("page2 = %d next=%v, want 1 row and no cursor", len(page2), next2) } if page1[len(page1)-1].Id >= page2[0].Id { t.Errorf("keyset overlap or gap: page1 last id %d, page2 first id %d", page1[len(page1)-1].Id, page2[0].Id) }}
func TestListKnotMembers_ZeroValuePageDefaultsLimit(t *testing.T) { d := newACLTestDB(t) owner := syntax.DID("did:plc:akshay") for _, s := range []string{"did:plc:limpet", "did:plc:whelk"} { if err := AddKnotMemberDirect(d, owner, syntax.DID(s)); err != nil { t.Fatalf("add %s: %v", s, err) } }
for _, p := range []ListPage{{}, {Limit: -3}} { members, next, err := ListKnotMembers(d, p) if err != nil { t.Fatalf("list with page %+v: %v", p, err) } if len(members) != 2 { t.Errorf("page %+v: members = %d, want 2 under the default limit", p, len(members)) } if next != nil { t.Errorf("page %+v: cursor = %v, want nil", p, *next) } }}
func TestDeleteRepoKeyRemovesCollaborators(t *testing.T) { d := newACLTestDB(t)
repoDid := syntax.DID("did:plc:whelk") owner := syntax.DID("did:plc:akshay") subject := syntax.DID("did:plc:boltless")
if err := d.StoreRepoKey(repoDid.String(), []byte("signing"), owner.String(), "reponame"); err != nil { t.Fatalf("StoreRepoKey: %v", err) } if err := AddCollaborator(d, Collaborator{RepoDid: repoDid, Subject: subject, AddedBy: owner}); err != nil { t.Fatalf("AddCollaborator: %v", err) } if ok, err := IsCollaborator(d, repoDid, subject); err != nil || !ok { t.Fatalf("collaborator missing before delete: ok=%v err=%v", ok, err) }
if err := d.DeleteRepoKey(repoDid.String()); err != nil { t.Fatalf("DeleteRepoKey: %v", err) }
if ok, err := IsCollaborator(d, repoDid, subject); err != nil || ok { t.Fatalf("collaborator not removed after repo delete: ok=%v err=%v", ok, err) }}