Coracle/nostrlib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

eventstore: add some basic tests.

Browse Source
This commit is contained in:
fiatjaf
2025-06-10 13:28:35 -03:00
parent e12f2b3f90
commit 287d8c23c7
2 changed files with 214 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

213
eventstore/test/basic_test.go Normal file
View File

@@ -0,0 +1,213 @@
package test
import (
"slices"
"testing"
"fiatjaf.com/nostr"
"fiatjaf.com/nostr/eventstore"
"github.com/stretchr/testify/require"
)
func basicTest(t *testing.T, db eventstore.Store) {
err := db.Init()
require.NoError(t, err)
// from basic-test.patch
{
// create test events with different tags and authors
pk3 := nostr.GetPublicKey(sk3)
pk4 := nostr.GetPublicKey(sk4)
events := []nostr.Event{
// event with 'e' tag
{
CreatedAt: 100,
Content: "event with e tag",
Tags: nostr.Tags{{"e", "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"}},
Kind: 1,
},
// event with 'q' tag
{
CreatedAt: 101,
Content: "event with q tag",
Tags: nostr.Tags{{"q", "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"}},
Kind: 1,
},
// event with 'p' tag and kind 3
{
CreatedAt: 102,
Content: "event with p tag kind 3",
Tags: nostr.Tags{{"p", pk3.Hex()}},
Kind: 3,
},
// event with 'p' tag and kind 7
{
CreatedAt: 103,
Content: "event with p tag kind 7",
Tags: nostr.Tags{{"p", pk4.Hex()}},
Kind: 7,
},
// event from author pk3 with kind 1
{
CreatedAt: 104,
Content: "event from pk3 kind 1",
Tags: nostr.Tags{},
Kind: 1,
},
// event from author pk4 with kind 3
{
CreatedAt: 105,
Content: "event from pk4 kind 3",
Tags: nostr.Tags{},
Kind: 3,
},
}
// sign events with appropriate keys
events[0].Sign(sk3)
events[1].Sign(sk3)
events[2].Sign(sk3)
events[3].Sign(sk3)
events[4].Sign(sk3)
events[5].Sign(sk4)
// save all events
for _, evt := range events {
err = db.SaveEvent(evt)
require.NoError(t, err)
}
// test 0: query all
{
results := slices.Collect(db.QueryEvents(nostr.Filter{}, 1000))
require.NoError(t, err)
require.Len(t, results, 6)
}
// test 1: query by 'e' tag
{
results := slices.Collect(db.QueryEvents(nostr.Filter{
Tags: nostr.TagMap{"e": []string{"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"}},
}, 1000))
require.NoError(t, err)
require.Len(t, results, 1)
require.Equal(t, events[0].ID, results[0].ID, "e tag query error")
}
// test 2: query by 'q' tag
{
results := slices.Collect(db.QueryEvents(nostr.Filter{
Tags: nostr.TagMap{"q": []string{"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"}},
}, 1000))
require.NoError(t, err)
require.Len(t, results, 1)
require.Equal(t, events[1].ID, results[0].ID, "q tag query error")
}
// test 3: query by 'p' tag + kind
{
results := slices.Collect(db.QueryEvents(nostr.Filter{
Tags: nostr.TagMap{"p": []string{pk3.Hex()}},
Kinds: []nostr.Kind{3},
}, 1000))
require.NoError(t, err)
require.Len(t, results, 1)
require.Equal(t, events[2].ID, results[0].ID, "p tag + kind query error")
}
// test 4: query by author + kind
{
results := slices.Collect(db.QueryEvents(nostr.Filter{
Authors: []nostr.PubKey{pk4},
Kinds: []nostr.Kind{3},
}, 1000))
require.NoError(t, err)
require.Len(t, results, 1)
require.Equal(t, events[5].ID, results[0].ID, "author + kind query error")
}
}
// from another-basic-test.patch
{
evt1 := nostr.Event{
CreatedAt: nostr.Now(),
Content: "two tags",
Tags: nostr.Tags{
{"e", "f355341a03672c5b136a6002fb4b69ad52111a1646638771c3995fc0a4db2a78"},
{"q", "f355341a03672c5b136a6002fb4b69ad52111a1646638771c3995fc0a4db2a78"},
},
Kind: 23122,
}
evt1.Sign(sk3)
require.NoError(t, db.SaveEvent(evt1))
{
results := slices.Collect(db.QueryEvents(nostr.Filter{
Tags: nostr.TagMap{"e": []string{"f355341a03672c5b136a6002fb4b69ad52111a1646638771c3995fc0a4db2a78"}},
}, 1000))
require.NoError(t, err)
require.Len(t, results, 1)
require.ElementsMatch(t,
[]nostr.Event{evt1},
results,
"querying by 'e' tag")
}
{
results := slices.Collect(db.QueryEvents(nostr.Filter{
Tags: nostr.TagMap{"q": []string{"f355341a03672c5b136a6002fb4b69ad52111a1646638771c3995fc0a4db2a78"}},
}, 1000))
require.NoError(t, err)
require.Len(t, results, 1)
require.ElementsMatch(t,
[]nostr.Event{evt1},
results,
"querying by 'q' tag")
}
evt2 := nostr.Event{
CreatedAt: nostr.Now(),
Content: "e tag",
Tags: nostr.Tags{
{"e", "f355341a03672c5b136a6002fb4b69ad52111a1646638771c3995fc0a4db2a78"},
},
Kind: 23122,
}
evt2.Sign(sk3)
require.NoError(t, db.SaveEvent(evt2))
evt3 := nostr.Event{
CreatedAt: nostr.Now(),
Content: "q tag",
Tags: nostr.Tags{
{"q", "f355341a03672c5b136a6002fb4b69ad52111a1646638771c3995fc0a4db2a78"},
},
Kind: 23122,
}
evt3.Sign(sk3)
require.NoError(t, db.SaveEvent(evt3))
{
results := slices.Collect(db.QueryEvents(nostr.Filter{
Tags: nostr.TagMap{"e": []string{"f355341a03672c5b136a6002fb4b69ad52111a1646638771c3995fc0a4db2a78"}},
}, 1000))
require.NoError(t, err)
require.Len(t, results, 2)
require.ElementsMatch(t,
[]nostr.Event{evt1, evt2},
results,
"querying by 'e' tag")
}
{
results := slices.Collect(db.QueryEvents(nostr.Filter{
Tags: nostr.TagMap{"q": []string{"f355341a03672c5b136a6002fb4b69ad52111a1646638771c3995fc0a4db2a78"}},
}, 1000))
require.NoError(t, err)
require.Len(t, results, 2)
require.ElementsMatch(t,
[]nostr.Event{evt1, evt3},
results,
"querying by 'q' tag")
}
}
}

1
eventstore/test/db_test.go
View File

@@ -24,6 +24,7 @@ var tests = []struct {
name string
run func(*testing.T, eventstore.Store)
}{
{"basic", basicTest},
{"first", runFirstTestOn},
{"second", runSecondTestOn},
{"manyauthors", manyAuthorsTest},