mirror of
https://github.com/strukturag/nextcloud-spreed-signaling
synced 2024-05-04 14:53:10 +02:00
Merge pull request #105 from strukturag/nats-tests
Add tests for regular NATS client.
This commit is contained in:
commit
8f3db6da80
2
go.mod
2
go.mod
|
@ -9,7 +9,7 @@ require (
|
||||||
github.com/gorilla/securecookie v1.1.1
|
github.com/gorilla/securecookie v1.1.1
|
||||||
github.com/gorilla/websocket v1.4.2
|
github.com/gorilla/websocket v1.4.2
|
||||||
github.com/mailru/easyjson v0.7.7
|
github.com/mailru/easyjson v0.7.7
|
||||||
github.com/nats-io/nats-server/v2 v2.2.1 // indirect
|
github.com/nats-io/nats-server/v2 v2.2.1
|
||||||
github.com/nats-io/nats.go v1.10.1-0.20210330225420-a0b1f60162f8
|
github.com/nats-io/nats.go v1.10.1-0.20210330225420-a0b1f60162f8
|
||||||
github.com/notedit/janus-go v0.0.0-20200517101215-10eb8b95d1a0
|
github.com/notedit/janus-go v0.0.0-20200517101215-10eb8b95d1a0
|
||||||
github.com/oschwald/maxminddb-golang v1.8.0
|
github.com/oschwald/maxminddb-golang v1.8.0
|
||||||
|
|
|
@ -52,9 +52,9 @@ type NatsSubscription interface {
|
||||||
}
|
}
|
||||||
|
|
||||||
type NatsClient interface {
|
type NatsClient interface {
|
||||||
Subscribe(subject string, ch chan *nats.Msg) (NatsSubscription, error)
|
Close()
|
||||||
|
|
||||||
Request(subject string, data []byte, timeout time.Duration) (*nats.Msg, error)
|
Subscribe(subject string, ch chan *nats.Msg) (NatsSubscription, error)
|
||||||
|
|
||||||
Publish(subject string, message interface{}) error
|
Publish(subject string, message interface{}) error
|
||||||
PublishNats(subject string, message *NatsMessage) error
|
PublishNats(subject string, message *NatsMessage) error
|
||||||
|
@ -120,6 +120,10 @@ func NewNatsClient(url string) (NatsClient, error) {
|
||||||
return client, nil
|
return client, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *natsClient) Close() {
|
||||||
|
c.conn.Close()
|
||||||
|
}
|
||||||
|
|
||||||
func (c *natsClient) onClosed(conn *nats.Conn) {
|
func (c *natsClient) onClosed(conn *nats.Conn) {
|
||||||
log.Println("NATS client closed", conn.LastError())
|
log.Println("NATS client closed", conn.LastError())
|
||||||
}
|
}
|
||||||
|
@ -136,10 +140,6 @@ func (c *natsClient) Subscribe(subject string, ch chan *nats.Msg) (NatsSubscript
|
||||||
return c.nc.ChanSubscribe(subject, ch)
|
return c.nc.ChanSubscribe(subject, ch)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *natsClient) Request(subject string, data []byte, timeout time.Duration) (*nats.Msg, error) {
|
|
||||||
return c.nc.Request(subject, data, timeout)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *natsClient) Publish(subject string, message interface{}) error {
|
func (c *natsClient) Publish(subject string, message interface{}) error {
|
||||||
return c.conn.Publish(subject, message)
|
return c.conn.Publish(subject, message)
|
||||||
}
|
}
|
||||||
|
|
|
@ -22,13 +22,9 @@
|
||||||
package signaling
|
package signaling
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"log"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/nats-io/nats.go"
|
"github.com/nats-io/nats.go"
|
||||||
)
|
)
|
||||||
|
@ -36,7 +32,6 @@ import (
|
||||||
type LoopbackNatsClient struct {
|
type LoopbackNatsClient struct {
|
||||||
mu sync.Mutex
|
mu sync.Mutex
|
||||||
subscriptions map[string]map[*loopbackNatsSubscription]bool
|
subscriptions map[string]map[*loopbackNatsSubscription]bool
|
||||||
replyId uint64
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewLoopbackNatsClient() (NatsClient, error) {
|
func NewLoopbackNatsClient() (NatsClient, error) {
|
||||||
|
@ -45,6 +40,19 @@ func NewLoopbackNatsClient() (NatsClient, error) {
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *LoopbackNatsClient) Close() {
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
|
||||||
|
for _, subs := range c.subscriptions {
|
||||||
|
for sub := range subs {
|
||||||
|
sub.Unsubscribe() // nolint
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
c.subscriptions = nil
|
||||||
|
}
|
||||||
|
|
||||||
type loopbackNatsSubscription struct {
|
type loopbackNatsSubscription struct {
|
||||||
subject string
|
subject string
|
||||||
client *LoopbackNatsClient
|
client *LoopbackNatsClient
|
||||||
|
@ -105,6 +113,10 @@ func (c *LoopbackNatsClient) subscribe(subject string, ch chan *nats.Msg) (NatsS
|
||||||
return nil, nats.ErrBadSubject
|
return nil, nats.ErrBadSubject
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if c.subscriptions == nil {
|
||||||
|
return nil, nats.ErrConnectionClosed
|
||||||
|
}
|
||||||
|
|
||||||
s := &loopbackNatsSubscription{
|
s := &loopbackNatsSubscription{
|
||||||
subject: subject,
|
subject: subject,
|
||||||
client: c,
|
client: c,
|
||||||
|
@ -134,77 +146,6 @@ func (c *LoopbackNatsClient) unsubscribe(s *loopbackNatsSubscription) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *LoopbackNatsClient) Request(subject string, data []byte, timeout time.Duration) (*nats.Msg, error) {
|
|
||||||
if strings.HasSuffix(subject, ".") || strings.Contains(subject, " ") {
|
|
||||||
return nil, nats.ErrBadSubject
|
|
||||||
}
|
|
||||||
|
|
||||||
c.mu.Lock()
|
|
||||||
defer c.mu.Unlock()
|
|
||||||
var response *nats.Msg
|
|
||||||
var err error
|
|
||||||
subs, found := c.subscriptions[subject]
|
|
||||||
if !found {
|
|
||||||
c.mu.Unlock()
|
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), timeout)
|
|
||||||
defer cancel()
|
|
||||||
<-ctx.Done()
|
|
||||||
if ctx.Err() == context.DeadlineExceeded {
|
|
||||||
err = nats.ErrTimeout
|
|
||||||
} else {
|
|
||||||
err = ctx.Err()
|
|
||||||
}
|
|
||||||
c.mu.Lock()
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
replyId := c.replyId
|
|
||||||
c.replyId += 1
|
|
||||||
|
|
||||||
reply := "_reply_" + strconv.FormatUint(replyId, 10)
|
|
||||||
responder := make(chan *nats.Msg)
|
|
||||||
var replySubscriber NatsSubscription
|
|
||||||
replySubscriber, err = c.subscribe(reply, responder)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
go func() {
|
|
||||||
if err := replySubscriber.Unsubscribe(); err != nil {
|
|
||||||
log.Printf("Error closing reply subscriber %s: %s", reply, err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
}()
|
|
||||||
msg := &nats.Msg{
|
|
||||||
Subject: subject,
|
|
||||||
Data: data,
|
|
||||||
Reply: reply,
|
|
||||||
Sub: &nats.Subscription{
|
|
||||||
Subject: subject,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
for s := range subs {
|
|
||||||
s.queue(msg)
|
|
||||||
}
|
|
||||||
c.mu.Unlock()
|
|
||||||
|
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), timeout)
|
|
||||||
defer cancel()
|
|
||||||
select {
|
|
||||||
case response = <-responder:
|
|
||||||
err = nil
|
|
||||||
case <-ctx.Done():
|
|
||||||
if ctx.Err() == context.DeadlineExceeded {
|
|
||||||
err = nats.ErrTimeout
|
|
||||||
} else {
|
|
||||||
err = ctx.Err()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
c.mu.Lock()
|
|
||||||
return response, err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *LoopbackNatsClient) Publish(subject string, message interface{}) error {
|
func (c *LoopbackNatsClient) Publish(subject string, message interface{}) error {
|
||||||
if strings.HasSuffix(subject, ".") || strings.Contains(subject, " ") {
|
if strings.HasSuffix(subject, ".") || strings.Contains(subject, " ") {
|
||||||
return nats.ErrBadSubject
|
return nats.ErrBadSubject
|
||||||
|
@ -212,6 +153,10 @@ func (c *LoopbackNatsClient) Publish(subject string, message interface{}) error
|
||||||
|
|
||||||
c.mu.Lock()
|
c.mu.Lock()
|
||||||
defer c.mu.Unlock()
|
defer c.mu.Unlock()
|
||||||
|
if c.subscriptions == nil {
|
||||||
|
return nats.ErrConnectionClosed
|
||||||
|
}
|
||||||
|
|
||||||
if subs, found := c.subscriptions[subject]; found {
|
if subs, found := c.subscriptions[subject]; found {
|
||||||
msg := &nats.Msg{
|
msg := &nats.Msg{
|
||||||
Subject: subject,
|
Subject: subject,
|
||||||
|
|
|
@ -22,14 +22,9 @@
|
||||||
package signaling
|
package signaling
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
"context"
|
"context"
|
||||||
"runtime"
|
|
||||||
"sync/atomic"
|
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/nats-io/nats.go"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func (c *LoopbackNatsClient) waitForSubscriptionsEmpty(ctx context.Context, t *testing.T) {
|
func (c *LoopbackNatsClient) waitForSubscriptionsEmpty(ctx context.Context, t *testing.T) {
|
||||||
|
@ -62,167 +57,33 @@ func CreateLoopbackNatsClientForTest(t *testing.T) NatsClient {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLoopbackNatsClient_Subscribe(t *testing.T) {
|
func TestLoopbackNatsClient_Subscribe(t *testing.T) {
|
||||||
// Give time for things to settle before capturing the number of
|
ensureNoGoroutinesLeak(t, func() {
|
||||||
// go routines
|
client := CreateLoopbackNatsClientForTest(t)
|
||||||
time.Sleep(500 * time.Millisecond)
|
|
||||||
|
|
||||||
base := runtime.NumGoroutine()
|
testNatsClient_Subscribe(t, client)
|
||||||
|
})
|
||||||
client := CreateLoopbackNatsClientForTest(t)
|
|
||||||
dest := make(chan *nats.Msg)
|
|
||||||
sub, err := client.Subscribe("foo", dest)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
ch := make(chan bool)
|
|
||||||
|
|
||||||
received := int32(0)
|
|
||||||
max := int32(20)
|
|
||||||
quit := make(chan bool)
|
|
||||||
go func() {
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-dest:
|
|
||||||
total := atomic.AddInt32(&received, 1)
|
|
||||||
if total == max {
|
|
||||||
err := sub.Unsubscribe()
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("Unsubscribe failed with err: %s", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
ch <- true
|
|
||||||
}
|
|
||||||
case <-quit:
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
for i := int32(0); i < max; i++ {
|
|
||||||
if err := client.Publish("foo", []byte("hello")); err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
<-ch
|
|
||||||
|
|
||||||
r := atomic.LoadInt32(&received)
|
|
||||||
if r != max {
|
|
||||||
t.Fatalf("Received wrong # of messages: %d vs %d", r, max)
|
|
||||||
}
|
|
||||||
quit <- true
|
|
||||||
|
|
||||||
// Give time for things to settle before capturing the number of
|
|
||||||
// go routines
|
|
||||||
time.Sleep(500 * time.Millisecond)
|
|
||||||
|
|
||||||
delta := (runtime.NumGoroutine() - base)
|
|
||||||
if delta > 0 {
|
|
||||||
t.Fatalf("%d Go routines still exist post Close()", delta)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLoopbackNatsClient_Request(t *testing.T) {
|
func TestLoopbackClient_PublishAfterClose(t *testing.T) {
|
||||||
// Give time for things to settle before capturing the number of
|
ensureNoGoroutinesLeak(t, func() {
|
||||||
// go routines
|
client := CreateLoopbackNatsClientForTest(t)
|
||||||
time.Sleep(500 * time.Millisecond)
|
|
||||||
|
|
||||||
base := runtime.NumGoroutine()
|
testNatsClient_PublishAfterClose(t, client)
|
||||||
|
})
|
||||||
client := CreateLoopbackNatsClientForTest(t)
|
|
||||||
dest := make(chan *nats.Msg)
|
|
||||||
sub, err := client.Subscribe("foo", dest)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
msg := <-dest
|
|
||||||
if err := client.Publish(msg.Reply, []byte("world")); err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if err := sub.Unsubscribe(); err != nil {
|
|
||||||
t.Error("Unsubscribe failed with err:", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
reply, err := client.Request("foo", []byte("hello"), 1*time.Second)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var response []byte
|
|
||||||
if err := client.Decode(reply, &response); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
if !bytes.Equal(response, []byte("world")) {
|
|
||||||
t.Fatalf("expected 'world', got '%s'", string(reply.Data))
|
|
||||||
}
|
|
||||||
|
|
||||||
// Give time for things to settle before capturing the number of
|
|
||||||
// go routines
|
|
||||||
time.Sleep(500 * time.Millisecond)
|
|
||||||
|
|
||||||
delta := (runtime.NumGoroutine() - base)
|
|
||||||
if delta > 0 {
|
|
||||||
t.Fatalf("%d Go routines still exist post Close()", delta)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLoopbackNatsClient_RequestTimeout(t *testing.T) {
|
func TestLoopbackClient_SubscribeAfterClose(t *testing.T) {
|
||||||
// Give time for things to settle before capturing the number of
|
ensureNoGoroutinesLeak(t, func() {
|
||||||
// go routines
|
client := CreateLoopbackNatsClientForTest(t)
|
||||||
time.Sleep(500 * time.Millisecond)
|
|
||||||
|
|
||||||
base := runtime.NumGoroutine()
|
testNatsClient_SubscribeAfterClose(t, client)
|
||||||
|
})
|
||||||
client := CreateLoopbackNatsClientForTest(t)
|
|
||||||
dest := make(chan *nats.Msg)
|
|
||||||
sub, err := client.Subscribe("foo", dest)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
msg := <-dest
|
|
||||||
time.Sleep(200 * time.Millisecond)
|
|
||||||
if err := client.Publish(msg.Reply, []byte("world")); err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if err := sub.Unsubscribe(); err != nil {
|
|
||||||
t.Error("Unsubscribe failed with err:", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
reply, err := client.Request("foo", []byte("hello"), 100*time.Millisecond)
|
|
||||||
if err == nil {
|
|
||||||
t.Fatalf("Request should have timed out, reeived %+v", reply)
|
|
||||||
} else if err != nats.ErrTimeout {
|
|
||||||
t.Fatalf("Request should have timed out, received error %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Give time for things to settle before capturing the number of
|
|
||||||
// go routines
|
|
||||||
time.Sleep(500 * time.Millisecond)
|
|
||||||
|
|
||||||
delta := (runtime.NumGoroutine() - base)
|
|
||||||
if delta > 0 {
|
|
||||||
t.Fatalf("%d Go routines still exist post Close()", delta)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLoopbackNatsClient_RequestTimeoutNoReply(t *testing.T) {
|
func TestLoopbackClient_BadSubjects(t *testing.T) {
|
||||||
client := CreateLoopbackNatsClientForTest(t)
|
ensureNoGoroutinesLeak(t, func() {
|
||||||
timeout := 100 * time.Millisecond
|
client := CreateLoopbackNatsClientForTest(t)
|
||||||
start := time.Now()
|
|
||||||
reply, err := client.Request("foo", []byte("hello"), timeout)
|
testNatsClient_BadSubjects(t, client)
|
||||||
end := time.Now()
|
})
|
||||||
if err == nil {
|
|
||||||
t.Fatalf("Request should have timed out, reeived %+v", reply)
|
|
||||||
} else if err != nats.ErrTimeout {
|
|
||||||
t.Fatalf("Request should have timed out, received error %s", err)
|
|
||||||
}
|
|
||||||
if end.Sub(start) < timeout {
|
|
||||||
t.Errorf("Expected a delay of %s but had %s", timeout, end.Sub(start))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
169
natsclient_test.go
Normal file
169
natsclient_test.go
Normal file
|
@ -0,0 +1,169 @@
|
||||||
|
/**
|
||||||
|
* Standalone signaling server for the Nextcloud Spreed app.
|
||||||
|
* Copyright (C) 2021 struktur AG
|
||||||
|
*
|
||||||
|
* @author Joachim Bauch <bauch@struktur.de>
|
||||||
|
*
|
||||||
|
* @license GNU AGPL version 3 or any later version
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU Affero General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU Affero General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Affero General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
package signaling
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync/atomic"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/nats-io/nats.go"
|
||||||
|
|
||||||
|
natsserver "github.com/nats-io/nats-server/v2/test"
|
||||||
|
)
|
||||||
|
|
||||||
|
func startLocalNatsServer() (string, func()) {
|
||||||
|
opts := natsserver.DefaultTestOptions
|
||||||
|
opts.Port = -1
|
||||||
|
opts.Cluster.Name = "testing"
|
||||||
|
srv := natsserver.RunServer(&opts)
|
||||||
|
shutdown := func() {
|
||||||
|
srv.Shutdown()
|
||||||
|
srv.WaitForShutdown()
|
||||||
|
}
|
||||||
|
return srv.ClientURL(), shutdown
|
||||||
|
}
|
||||||
|
|
||||||
|
func CreateLocalNatsClientForTest(t *testing.T) (NatsClient, func()) {
|
||||||
|
url, shutdown := startLocalNatsServer()
|
||||||
|
result, err := NewNatsClient(url)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
return result, func() {
|
||||||
|
result.Close()
|
||||||
|
shutdown()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func testNatsClient_Subscribe(t *testing.T, client NatsClient) {
|
||||||
|
dest := make(chan *nats.Msg)
|
||||||
|
sub, err := client.Subscribe("foo", dest)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
ch := make(chan bool)
|
||||||
|
|
||||||
|
received := int32(0)
|
||||||
|
max := int32(20)
|
||||||
|
quit := make(chan bool)
|
||||||
|
go func() {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-dest:
|
||||||
|
total := atomic.AddInt32(&received, 1)
|
||||||
|
if total == max {
|
||||||
|
err := sub.Unsubscribe()
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unsubscribe failed with err: %s", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ch <- true
|
||||||
|
}
|
||||||
|
case <-quit:
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
for i := int32(0); i < max; i++ {
|
||||||
|
if err := client.Publish("foo", []byte("hello")); err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Allow NATS goroutines to process messages.
|
||||||
|
time.Sleep(time.Millisecond)
|
||||||
|
}
|
||||||
|
<-ch
|
||||||
|
|
||||||
|
r := atomic.LoadInt32(&received)
|
||||||
|
if r != max {
|
||||||
|
t.Fatalf("Received wrong # of messages: %d vs %d", r, max)
|
||||||
|
}
|
||||||
|
quit <- true
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNatsClient_Subscribe(t *testing.T) {
|
||||||
|
ensureNoGoroutinesLeak(t, func() {
|
||||||
|
client, shutdown := CreateLocalNatsClientForTest(t)
|
||||||
|
defer shutdown()
|
||||||
|
|
||||||
|
testNatsClient_Subscribe(t, client)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func testNatsClient_PublishAfterClose(t *testing.T, client NatsClient) {
|
||||||
|
client.Close()
|
||||||
|
|
||||||
|
if err := client.Publish("foo", "bar"); err != nats.ErrConnectionClosed {
|
||||||
|
t.Errorf("Expected %v, got %v", nats.ErrConnectionClosed, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNatsClient_PublishAfterClose(t *testing.T) {
|
||||||
|
ensureNoGoroutinesLeak(t, func() {
|
||||||
|
client, shutdown := CreateLocalNatsClientForTest(t)
|
||||||
|
defer shutdown()
|
||||||
|
|
||||||
|
testNatsClient_PublishAfterClose(t, client)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func testNatsClient_SubscribeAfterClose(t *testing.T, client NatsClient) {
|
||||||
|
client.Close()
|
||||||
|
|
||||||
|
ch := make(chan *nats.Msg)
|
||||||
|
if _, err := client.Subscribe("foo", ch); err != nats.ErrConnectionClosed {
|
||||||
|
t.Errorf("Expected %v, got %v", nats.ErrConnectionClosed, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNatsClient_SubscribeAfterClose(t *testing.T) {
|
||||||
|
ensureNoGoroutinesLeak(t, func() {
|
||||||
|
client, shutdown := CreateLocalNatsClientForTest(t)
|
||||||
|
defer shutdown()
|
||||||
|
|
||||||
|
testNatsClient_SubscribeAfterClose(t, client)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func testNatsClient_BadSubjects(t *testing.T, client NatsClient) {
|
||||||
|
subjects := []string{
|
||||||
|
"foo bar",
|
||||||
|
"foo.",
|
||||||
|
}
|
||||||
|
|
||||||
|
ch := make(chan *nats.Msg)
|
||||||
|
for _, s := range subjects {
|
||||||
|
if _, err := client.Subscribe(s, ch); err != nats.ErrBadSubject {
|
||||||
|
t.Errorf("Expected %v for subject %s, got %v", nats.ErrBadSubject, s, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNatsClient_BadSubjects(t *testing.T) {
|
||||||
|
ensureNoGoroutinesLeak(t, func() {
|
||||||
|
client, shutdown := CreateLocalNatsClientForTest(t)
|
||||||
|
defer shutdown()
|
||||||
|
|
||||||
|
testNatsClient_BadSubjects(t, client)
|
||||||
|
})
|
||||||
|
}
|
|
@ -191,7 +191,7 @@ func NewTestClient(t *testing.T, server *httptest.Server, hub *Hub) *TestClient
|
||||||
}
|
}
|
||||||
|
|
||||||
messageChan := make(chan []byte)
|
messageChan := make(chan []byte)
|
||||||
readErrorChan := make(chan error)
|
readErrorChan := make(chan error, 1)
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
for {
|
||||||
|
|
57
testutils_test.go
Normal file
57
testutils_test.go
Normal file
|
@ -0,0 +1,57 @@
|
||||||
|
/**
|
||||||
|
* Standalone signaling server for the Nextcloud Spreed app.
|
||||||
|
* Copyright (C) 2021 struktur AG
|
||||||
|
*
|
||||||
|
* @author Joachim Bauch <bauch@struktur.de>
|
||||||
|
*
|
||||||
|
* @license GNU AGPL version 3 or any later version
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU Affero General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU Affero General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Affero General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
package signaling
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"runtime/pprof"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func ensureNoGoroutinesLeak(t *testing.T, f func()) {
|
||||||
|
// Give time for things to settle before capturing the number of
|
||||||
|
// go routines
|
||||||
|
time.Sleep(500 * time.Millisecond)
|
||||||
|
before := pprof.Lookup("goroutine")
|
||||||
|
|
||||||
|
f()
|
||||||
|
|
||||||
|
var after *pprof.Profile
|
||||||
|
// Give time for things to settle before capturing the number of
|
||||||
|
// go routines
|
||||||
|
timeout := time.Now().Add(time.Second)
|
||||||
|
for time.Now().Before(timeout) {
|
||||||
|
after = pprof.Lookup("goroutine")
|
||||||
|
if after.Count() == before.Count() {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if after.Count() != before.Count() {
|
||||||
|
os.Stderr.WriteString("Before:\n")
|
||||||
|
before.WriteTo(os.Stderr, 1) // nolint
|
||||||
|
os.Stderr.WriteString("After:\n")
|
||||||
|
after.WriteTo(os.Stderr, 1) // nolint
|
||||||
|
t.Fatalf("Number of Go routines has changed in %s", t.Name())
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue