2022-06-28 11:55:24 +00:00
|
|
|
/**
|
|
|
|
* Standalone signaling server for the Nextcloud Spreed app.
|
|
|
|
* Copyright (C) 2022 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"
|
|
|
|
|
|
|
|
"github.com/dlintw/goconf"
|
|
|
|
"go.etcd.io/etcd/server/v3/embed"
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
GrpcSelfTargetForTesting = "testing.grpc.target"
|
|
|
|
)
|
|
|
|
|
2022-06-28 15:18:35 +00:00
|
|
|
func NewGrpcClientsForTest(t *testing.T, addr string) *GrpcClients {
|
|
|
|
config := goconf.NewConfigFile()
|
|
|
|
config.AddOption("grpc", "targets", addr)
|
|
|
|
|
|
|
|
client, err := NewGrpcClients(config, nil)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
t.Cleanup(func() {
|
|
|
|
client.Close()
|
|
|
|
})
|
|
|
|
|
|
|
|
return client
|
|
|
|
}
|
|
|
|
|
2022-06-28 11:55:24 +00:00
|
|
|
func NewGrpcClientsWithEtcdForTest(t *testing.T, etcd *embed.Etcd) *GrpcClients {
|
|
|
|
config := goconf.NewConfigFile()
|
|
|
|
config.AddOption("etcd", "endpoints", etcd.Config().LCUrls[0].String())
|
|
|
|
|
|
|
|
config.AddOption("grpc", "targettype", "etcd")
|
|
|
|
config.AddOption("grpc", "targetprefix", "/grpctargets")
|
|
|
|
config.AddOption("grpc", "targetself", GrpcSelfTargetForTesting)
|
|
|
|
|
|
|
|
etcdClient, err := NewEtcdClient(config, "")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
t.Cleanup(func() {
|
|
|
|
if err := etcdClient.Close(); err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
|
|
|
|
client, err := NewGrpcClients(config, etcdClient)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
t.Cleanup(func() {
|
|
|
|
client.Close()
|
|
|
|
})
|
|
|
|
|
|
|
|
return client
|
|
|
|
}
|
|
|
|
|
|
|
|
func drainWakeupChannel(ch chan bool) {
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case <-ch:
|
|
|
|
default:
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Test_GrpcClients_EtcdInitial(t *testing.T) {
|
|
|
|
etcd := NewEtcdForTest(t)
|
|
|
|
|
|
|
|
_, addr1 := NewGrpcServerForTest(t)
|
|
|
|
SetEtcdValue(etcd, "/grpctargets/one", []byte("{\"address\":\""+addr1+"\"}"))
|
|
|
|
_, addr2 := NewGrpcServerForTest(t)
|
|
|
|
SetEtcdValue(etcd, "/grpctargets/two", []byte("{\"address\":\""+addr2+"\"}"))
|
|
|
|
|
|
|
|
client := NewGrpcClientsWithEtcdForTest(t, etcd)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
|
|
|
|
defer cancel()
|
|
|
|
if err := client.WaitForInitialized(ctx); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if clients := client.GetClients(); len(clients) != 2 {
|
|
|
|
t.Errorf("Expected two clients, got %+v", clients)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Test_GrpcClients_EtcdUpdate(t *testing.T) {
|
|
|
|
etcd := NewEtcdForTest(t)
|
|
|
|
client := NewGrpcClientsWithEtcdForTest(t, etcd)
|
|
|
|
ch := make(chan bool, 1)
|
|
|
|
client.wakeupChanForTesting = ch
|
|
|
|
|
|
|
|
if clients := client.GetClients(); len(clients) != 0 {
|
|
|
|
t.Errorf("Expected no clients, got %+v", clients)
|
|
|
|
}
|
|
|
|
|
|
|
|
drainWakeupChannel(ch)
|
|
|
|
_, addr1 := NewGrpcServerForTest(t)
|
|
|
|
SetEtcdValue(etcd, "/grpctargets/one", []byte("{\"address\":\""+addr1+"\"}"))
|
|
|
|
<-ch
|
|
|
|
if clients := client.GetClients(); len(clients) != 1 {
|
|
|
|
t.Errorf("Expected one client, got %+v", clients)
|
|
|
|
} else if clients[0].Target() != addr1 {
|
|
|
|
t.Errorf("Expected target %s, got %s", addr1, clients[0].Target())
|
|
|
|
}
|
|
|
|
|
|
|
|
drainWakeupChannel(ch)
|
|
|
|
_, addr2 := NewGrpcServerForTest(t)
|
|
|
|
SetEtcdValue(etcd, "/grpctargets/two", []byte("{\"address\":\""+addr2+"\"}"))
|
|
|
|
<-ch
|
|
|
|
if clients := client.GetClients(); len(clients) != 2 {
|
|
|
|
t.Errorf("Expected two clients, got %+v", clients)
|
|
|
|
} else if clients[0].Target() != addr1 {
|
|
|
|
t.Errorf("Expected target %s, got %s", addr1, clients[0].Target())
|
|
|
|
} else if clients[1].Target() != addr2 {
|
|
|
|
t.Errorf("Expected target %s, got %s", addr2, clients[1].Target())
|
|
|
|
}
|
|
|
|
|
|
|
|
drainWakeupChannel(ch)
|
|
|
|
DeleteEtcdValue(etcd, "/grpctargets/one")
|
|
|
|
<-ch
|
|
|
|
if clients := client.GetClients(); len(clients) != 1 {
|
|
|
|
t.Errorf("Expected one client, got %+v", clients)
|
|
|
|
} else if clients[0].Target() != addr2 {
|
|
|
|
t.Errorf("Expected target %s, got %s", addr2, clients[0].Target())
|
|
|
|
}
|
|
|
|
|
|
|
|
drainWakeupChannel(ch)
|
|
|
|
_, addr3 := NewGrpcServerForTest(t)
|
|
|
|
SetEtcdValue(etcd, "/grpctargets/two", []byte("{\"address\":\""+addr3+"\"}"))
|
|
|
|
<-ch
|
|
|
|
if clients := client.GetClients(); len(clients) != 1 {
|
|
|
|
t.Errorf("Expected one client, got %+v", clients)
|
|
|
|
} else if clients[0].Target() != addr3 {
|
|
|
|
t.Errorf("Expected target %s, got %s", addr3, clients[0].Target())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Test_GrpcClients_EtcdIgnoreSelf(t *testing.T) {
|
|
|
|
etcd := NewEtcdForTest(t)
|
|
|
|
client := NewGrpcClientsWithEtcdForTest(t, etcd)
|
|
|
|
ch := make(chan bool, 1)
|
|
|
|
client.wakeupChanForTesting = ch
|
|
|
|
|
|
|
|
if clients := client.GetClients(); len(clients) != 0 {
|
|
|
|
t.Errorf("Expected no clients, got %+v", clients)
|
|
|
|
}
|
|
|
|
|
|
|
|
drainWakeupChannel(ch)
|
|
|
|
_, addr1 := NewGrpcServerForTest(t)
|
|
|
|
SetEtcdValue(etcd, "/grpctargets/one", []byte("{\"address\":\""+addr1+"\"}"))
|
|
|
|
<-ch
|
|
|
|
if clients := client.GetClients(); len(clients) != 1 {
|
|
|
|
t.Errorf("Expected one client, got %+v", clients)
|
|
|
|
} else if clients[0].Target() != addr1 {
|
|
|
|
t.Errorf("Expected target %s, got %s", addr1, clients[0].Target())
|
|
|
|
}
|
|
|
|
|
|
|
|
drainWakeupChannel(ch)
|
|
|
|
SetEtcdValue(etcd, "/grpctargets/two", []byte("{\"address\":\""+GrpcSelfTargetForTesting+"\"}"))
|
|
|
|
<-ch
|
|
|
|
if clients := client.GetClients(); len(clients) != 1 {
|
|
|
|
t.Errorf("Expected one client, got %+v", clients)
|
|
|
|
} else if clients[0].Target() != addr1 {
|
|
|
|
t.Errorf("Expected target %s, got %s", addr1, clients[0].Target())
|
|
|
|
}
|
|
|
|
|
|
|
|
drainWakeupChannel(ch)
|
|
|
|
DeleteEtcdValue(etcd, "/grpctargets/two")
|
|
|
|
<-ch
|
|
|
|
if clients := client.GetClients(); len(clients) != 1 {
|
|
|
|
t.Errorf("Expected one client, got %+v", clients)
|
|
|
|
} else if clients[0].Target() != addr1 {
|
|
|
|
t.Errorf("Expected target %s, got %s", addr1, clients[0].Target())
|
|
|
|
}
|
|
|
|
}
|