103 lines
2.1 KiB
Go
103 lines
2.1 KiB
Go
package nostr
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
const RELAY = "wss://relay.damus.io"
|
|
|
|
// test if we can fetch a couple of random events
|
|
func TestSubscribe(t *testing.T) {
|
|
rl := mustRelayConnect(RELAY)
|
|
defer rl.Close()
|
|
|
|
sub, err := rl.Subscribe(context.Background(), Filters{{Kinds: []int{1}, Limit: 2}})
|
|
if err != nil {
|
|
t.Errorf("subscription failed: %v", err)
|
|
return
|
|
}
|
|
|
|
timeout := time.After(5 * time.Second)
|
|
n := 0
|
|
|
|
for {
|
|
select {
|
|
case event := <-sub.Events:
|
|
if event == nil {
|
|
t.Errorf("event is nil: %v", event)
|
|
}
|
|
n++
|
|
case <-sub.EndOfStoredEvents:
|
|
goto end
|
|
case <-rl.Context().Done():
|
|
t.Errorf("connection closed: %v", rl.Context().Err())
|
|
goto end
|
|
case <-timeout:
|
|
t.Errorf("timeout")
|
|
goto end
|
|
}
|
|
}
|
|
|
|
end:
|
|
if n != 2 {
|
|
t.Errorf("expected 2 events, got %d", n)
|
|
}
|
|
}
|
|
|
|
// test if we can do multiple nested subscriptions
|
|
func TestNestedSubscriptions(t *testing.T) {
|
|
rl := mustRelayConnect(RELAY)
|
|
defer rl.Close()
|
|
|
|
// fetch any note
|
|
sub, err := rl.Subscribe(context.Background(), Filters{{Kinds: []int{1}, Limit: 1}})
|
|
if err != nil {
|
|
t.Errorf("subscription 1 failed: %v", err)
|
|
return
|
|
}
|
|
|
|
timeout := time.After(5 * time.Second)
|
|
|
|
for {
|
|
select {
|
|
case event := <-sub.Events:
|
|
// now fetch author of this event
|
|
sub, err := rl.Subscribe(context.Background(), Filters{{Kinds: []int{0}, Authors: []string{event.PubKey}, Limit: 1}})
|
|
if err != nil {
|
|
t.Errorf("subscription 2 failed: %v", err)
|
|
return
|
|
}
|
|
|
|
for {
|
|
select {
|
|
case <-sub.Events:
|
|
// now mentions of this person
|
|
sub, err := rl.Subscribe(context.Background(), Filters{{Kinds: []int{1}, Tags: TagMap{"p": []string{event.PubKey}}, Limit: 1}})
|
|
if err != nil {
|
|
t.Errorf("subscription 3 failed: %v", err)
|
|
return
|
|
}
|
|
|
|
for {
|
|
select {
|
|
case <-sub.Events:
|
|
// if we get here safely we won
|
|
return
|
|
case <-timeout:
|
|
t.Errorf("timeout 3")
|
|
}
|
|
}
|
|
case <-timeout:
|
|
t.Errorf("timeout 2")
|
|
}
|
|
}
|
|
case <-rl.Context().Done():
|
|
t.Errorf("connection closed: %v", rl.Context().Err())
|
|
case <-timeout:
|
|
t.Errorf("timeout 1")
|
|
}
|
|
}
|
|
}
|