status-go/metrics/node/subscribe_test.go

38 lines
780 B
Go
Raw Normal View History

package node
import (
"context"
"testing"
"github.com/ethereum/go-ethereum/node"
"github.com/stretchr/testify/require"
)
func TestSubscribeServerEventsWithoutServer(t *testing.T) {
node, err := node.New(&node.Config{})
require.NoError(t, err)
require.EqualError(t, SubscribeServerEvents(context.TODO(), node), "server is unavailable")
}
func TestSubscribeServerEvents(t *testing.T) {
node, err := node.New(&node.Config{})
require.NoError(t, err)
err = node.Start()
require.NoError(t, err)
defer func() {
err := node.Stop()
require.NoError(t, err)
}()
ctx, cancel := context.WithCancel(context.Background())
done := make(chan struct{})
go func() {
err := SubscribeServerEvents(ctx, node)
require.NoError(t, err)
close(done)
}()
cancel()
<-done
}