82 lines
1.6 KiB
Go
82 lines
1.6 KiB
Go
package bluge
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"fiatjaf.com/nostr/eventstore/badger"
|
|
"fiatjaf.com/nostr"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestBlugeFlow(t *testing.T) {
|
|
os.RemoveAll("/tmp/blugetest-badger")
|
|
os.RemoveAll("/tmp/blugetest-bluge")
|
|
|
|
bb := &badger.BadgerBackend{Path: "/tmp/blugetest-badger"}
|
|
bb.Init()
|
|
defer bb.Close()
|
|
|
|
bl := BlugeBackend{
|
|
Path: "/tmp/blugetest-bluge",
|
|
RawEventStore: bb,
|
|
}
|
|
bl.Init()
|
|
defer bl.Close()
|
|
|
|
ctx := context.Background()
|
|
|
|
willDelete := make([]*nostr.Event, 0, 3)
|
|
|
|
for i, content := range []string{
|
|
"good morning mr paper maker",
|
|
"good night",
|
|
"I'll see you again in the paper house",
|
|
"tonight we dine in my house",
|
|
"the paper in this house if very good, mr",
|
|
} {
|
|
evt := &nostr.Event{Content: content, Tags: nostr.Tags{}}
|
|
evt.Sign("0000000000000000000000000000000000000000000000000000000000000001")
|
|
|
|
bb.SaveEvent(ctx, evt)
|
|
bl.SaveEvent(ctx, evt)
|
|
|
|
if i%2 == 0 {
|
|
willDelete = append(willDelete, evt)
|
|
}
|
|
}
|
|
|
|
{
|
|
ch, err := bl.QueryEvents(ctx, nostr.Filter{Search: "good"})
|
|
if err != nil {
|
|
t.Fatalf("QueryEvents error: %s", err)
|
|
return
|
|
}
|
|
n := 0
|
|
for range ch {
|
|
n++
|
|
}
|
|
assert.Equal(t, 3, n)
|
|
}
|
|
|
|
for _, evt := range willDelete {
|
|
bl.DeleteEvent(ctx, evt)
|
|
}
|
|
|
|
{
|
|
ch, err := bl.QueryEvents(ctx, nostr.Filter{Search: "good"})
|
|
if err != nil {
|
|
t.Fatalf("QueryEvents error: %s", err)
|
|
return
|
|
}
|
|
n := 0
|
|
for res := range ch {
|
|
n++
|
|
assert.Equal(t, res.Content, "good night")
|
|
assert.Equal(t, res.PubKey, "79be667ef9dcbbac55a06295ce870b07029bfcdb2dce28d959f2815b16f81798")
|
|
}
|
|
assert.Equal(t, 1, n)
|
|
}
|
|
}
|