mirror of
https://github.com/strukturag/nextcloud-spreed-signaling
synced 2024-06-08 08:52:27 +02:00
0020076f2b
This makes sure the test has been completely teared down before the goroutines are counted and thus should prevent flaky failed tests.
93 lines
2.3 KiB
Go
93 lines
2.3 KiB
Go
/**
|
|
* Standalone signaling server for the Nextcloud Spreed app.
|
|
* Copyright (C) 2018 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 (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func (c *LoopbackNatsClient) waitForSubscriptionsEmpty(ctx context.Context, t *testing.T) {
|
|
for {
|
|
c.mu.Lock()
|
|
count := len(c.subscriptions)
|
|
c.mu.Unlock()
|
|
if count == 0 {
|
|
break
|
|
}
|
|
|
|
select {
|
|
case <-ctx.Done():
|
|
c.mu.Lock()
|
|
t.Errorf("Error waiting for subscriptions %+v to terminate: %s", c.subscriptions, ctx.Err())
|
|
c.mu.Unlock()
|
|
return
|
|
default:
|
|
time.Sleep(time.Millisecond)
|
|
}
|
|
}
|
|
}
|
|
|
|
func CreateLoopbackNatsClientForTest(t *testing.T) NatsClient {
|
|
result, err := NewLoopbackNatsClient()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(func() {
|
|
result.Close()
|
|
})
|
|
return result
|
|
}
|
|
|
|
func TestLoopbackNatsClient_Subscribe(t *testing.T) {
|
|
ensureNoGoroutinesLeak(t, func(t *testing.T) {
|
|
client := CreateLoopbackNatsClientForTest(t)
|
|
|
|
testNatsClient_Subscribe(t, client)
|
|
})
|
|
}
|
|
|
|
func TestLoopbackClient_PublishAfterClose(t *testing.T) {
|
|
ensureNoGoroutinesLeak(t, func(t *testing.T) {
|
|
client := CreateLoopbackNatsClientForTest(t)
|
|
|
|
testNatsClient_PublishAfterClose(t, client)
|
|
})
|
|
}
|
|
|
|
func TestLoopbackClient_SubscribeAfterClose(t *testing.T) {
|
|
ensureNoGoroutinesLeak(t, func(t *testing.T) {
|
|
client := CreateLoopbackNatsClientForTest(t)
|
|
|
|
testNatsClient_SubscribeAfterClose(t, client)
|
|
})
|
|
}
|
|
|
|
func TestLoopbackClient_BadSubjects(t *testing.T) {
|
|
ensureNoGoroutinesLeak(t, func(t *testing.T) {
|
|
client := CreateLoopbackNatsClientForTest(t)
|
|
|
|
testNatsClient_BadSubjects(t, client)
|
|
})
|
|
}
|