80 lines
2.0 KiB
Go
80 lines
2.0 KiB
Go
package test
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"slices"
|
|
"testing"
|
|
|
|
"fiatjaf.com/nostr"
|
|
"fiatjaf.com/nostr/eventstore"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func runSecondTestOn(t *testing.T, db eventstore.Store) {
|
|
db.Init()
|
|
|
|
for i := 0; i < 10000; i++ {
|
|
eTag := make([]byte, 32)
|
|
binary.BigEndian.PutUint16(eTag, uint16(i))
|
|
|
|
ref := nostr.GetPublicKey(sk3)
|
|
if i%3 == 0 {
|
|
ref = nostr.GetPublicKey(sk4)
|
|
}
|
|
|
|
evt := nostr.Event{
|
|
CreatedAt: nostr.Timestamp(i*10 + 2),
|
|
Content: fmt.Sprintf("hello %d", i),
|
|
Tags: nostr.Tags{
|
|
{"t", fmt.Sprintf("t%d", i)},
|
|
{"e", hex.EncodeToString(eTag)},
|
|
{"p", ref.Hex()},
|
|
},
|
|
Kind: nostr.Kind(i % 10),
|
|
}
|
|
sk := sk3
|
|
if i%3 == 0 {
|
|
sk = sk4
|
|
}
|
|
evt.Sign(sk)
|
|
err := db.SaveEvent(evt)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
pk3 := nostr.GetPublicKey(sk3)
|
|
pk4 := nostr.GetPublicKey(sk4)
|
|
eTags := make([]string, 20)
|
|
for i := 0; i < 20; i++ {
|
|
eTag := make([]byte, 32)
|
|
binary.BigEndian.PutUint16(eTag, uint16(i))
|
|
eTags[i] = hex.EncodeToString(eTag)
|
|
}
|
|
|
|
filters := []nostr.Filter{
|
|
{Kinds: []nostr.Kind{1, 4, 8, 16}},
|
|
{Authors: []nostr.PubKey{pk3, nostr.Generate().Public()}},
|
|
{Authors: []nostr.PubKey{pk3, nostr.Generate().Public()}, Kinds: []nostr.Kind{3, 4}},
|
|
{},
|
|
{Limit: 20},
|
|
{Kinds: []nostr.Kind{8, 9}, Tags: nostr.TagMap{"p": []string{pk3.Hex()}}},
|
|
{Kinds: []nostr.Kind{8, 9}, Tags: nostr.TagMap{"p": []string{pk3.Hex(), pk4.Hex()}}},
|
|
{Kinds: []nostr.Kind{8, 9}, Tags: nostr.TagMap{"p": []string{pk3.Hex(), pk4.Hex()}}},
|
|
{Kinds: []nostr.Kind{9}, Tags: nostr.TagMap{"e": eTags}},
|
|
{Kinds: []nostr.Kind{5}, Tags: nostr.TagMap{"e": eTags, "t": []string{"t5"}}},
|
|
{Tags: nostr.TagMap{"e": eTags}},
|
|
{Tags: nostr.TagMap{"e": eTags}, Limit: 50},
|
|
}
|
|
|
|
t.Run("filter", func(t *testing.T) {
|
|
for q, filter := range filters {
|
|
label := fmt.Sprintf("filter %d: %s", q, filter)
|
|
t.Run(fmt.Sprintf("q-%d", q), func(t *testing.T) {
|
|
results := slices.Collect(db.QueryEvents(filter))
|
|
require.NotEmpty(t, results, label)
|
|
})
|
|
}
|
|
})
|
|
}
|