khatru/util_test.go

package khatru

import (
	"context"
	"testing"

	"github.com/nbd-wtf/go-nostr"
)

func startTestRelay(t *testing.T, tr *testRelay) *Server {
	t.Helper()
	srv, _ := NewServer(tr)
	started := make(chan bool)
	go srv.Start("127.0.0.1", 0, started)
	<-started
	return srv
}

type testRelay struct {
	name        string
	storage     Storage
	init        func() error
	onShutdown  func(context.Context)
	acceptEvent func(*nostr.Event) bool
}

func (tr *testRelay) Name() string                    { return tr.name }
func (tr *testRelay) Storage(context.Context) Storage { return tr.storage }

func (tr *testRelay) Init() error {
	if fn := tr.init; fn != nil {
		return fn()
	}
	return nil
}

func (tr *testRelay) OnShutdown(ctx context.Context) {
	if fn := tr.onShutdown; fn != nil {
		fn(ctx)
	}
}

func (tr *testRelay) AcceptEvent(ctx context.Context, e *nostr.Event) bool {
	if fn := tr.acceptEvent; fn != nil {
		return fn(e)
	}
	return true
}

type testStorage struct {
	init        func() error
	queryEvents func(context.Context, *nostr.Filter) (chan *nostr.Event, error)
	deleteEvent func(ctx context.Context, id string, pubkey string) error
	saveEvent   func(context.Context, *nostr.Event) error
	countEvents func(context.Context, *nostr.Filter) (int64, error)
}

func (st *testStorage) Init() error {
	if fn := st.init; fn != nil {
		return fn()
	}
	return nil
}

func (st *testStorage) QueryEvents(ctx context.Context, f *nostr.Filter) (chan *nostr.Event, error) {
	if fn := st.queryEvents; fn != nil {
		return fn(ctx, f)
	}
	return nil, nil
}

func (st *testStorage) DeleteEvent(ctx context.Context, id string, pubkey string) error {
	if fn := st.deleteEvent; fn != nil {
		return fn(ctx, id, pubkey)
	}
	return nil
}

func (st *testStorage) SaveEvent(ctx context.Context, e *nostr.Event) error {
	if fn := st.saveEvent; fn != nil {
		return fn(ctx, e)
	}
	return nil
}

func (st *testStorage) CountEvents(ctx context.Context, f *nostr.Filter) (int64, error) {
	if fn := st.countEvents; fn != nil {
		return fn(ctx, f)
	}
	return 0, nil
}