sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-10-24 06:19:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

temporarily disable old test suite test

Browse Source
This commit is contained in:
Pascal Fischer
2025-01-17 00:24:49 +01:00
parent 14a3461e86
commit 66f003d00f
1 changed files with 75 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files
management/server/management_test.go
+75 -73
View File
@@ -6,6 +6,7 @@ import (
"net" "net"
"os" "os"
"runtime" "runtime"
"strconv"
sync2 "sync" sync2 "sync"
"time" "time"
@@ -47,7 +48,7 @@ var _ = Describe("Management service", func() {
level, _ := log.ParseLevel("Debug") level, _ := log.ParseLevel("Debug")
log.SetLevel(level) log.SetLevel(level)
var err error var err error
dataDir, err = os.MkdirTemp("", "wiretrustee_mgmt_test_tmp_*") dataDir, err = os.MkdirTemp("", strconv.Itoa(rand.Int())+"wiretrustee_mgmt_test_tmp_*")
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
var listener net.Listener var listener net.Listener
@@ -309,78 +310,79 @@ var _ = Describe("Management service", func() {
}) })
}) })
Context("when there are 10 peers registered under one account", func() { // Context("when there are 10 peers registered under one account", func() {
Context("when there are 10 more peers registered under the same account", func() { // Context("when there are 10 more peers registered under the same account", func() {
Specify("all of the 10 peers will get updates of 10 newly registered peers", func() { // Specify("all of the 10 peers will get updates of 10 newly registered peers", func() {
initialPeers := 10 // initialPeers := 10
additionalPeers := 10 // additionalPeers := 10
//
var peers []wgtypes.Key // var peers []wgtypes.Key
for i := 0; i < initialPeers; i++ { // for i := 0; i < initialPeers; i++ {
key, _ := wgtypes.GenerateKey() // key, _ := wgtypes.GenerateKey()
loginPeerWithValidSetupKey(serverPubKey, key, client) // loginPeerWithValidSetupKey(serverPubKey, key, client)
peers = append(peers, key) // peers = append(peers, key)
} // }
//
wg := sync2.WaitGroup{} // wg := sync2.WaitGroup{}
wg.Add(initialPeers + initialPeers*additionalPeers) // wgCounter := initialPeers + initialPeers*additionalPeers
// wg.Add(wgCounter)
var clients []mgmtProto.ManagementService_SyncClient //
for _, peer := range peers { // var clients []mgmtProto.ManagementService_SyncClient
messageBytes, err := pb.Marshal(&mgmtProto.SyncRequest{Meta: &mgmtProto.PeerSystemMeta{}}) // for _, peer := range peers {
Expect(err).NotTo(HaveOccurred()) // messageBytes, err := pb.Marshal(&mgmtProto.SyncRequest{Meta: &mgmtProto.PeerSystemMeta{}})
encryptedBytes, err := encryption.Encrypt(messageBytes, serverPubKey, peer) // Expect(err).NotTo(HaveOccurred())
Expect(err).NotTo(HaveOccurred()) // encryptedBytes, err := encryption.Encrypt(messageBytes, serverPubKey, peer)
// Expect(err).NotTo(HaveOccurred())
// open stream //
sync, err := client.Sync(context.TODO(), &mgmtProto.EncryptedMessage{ // // open stream
WgPubKey: peer.PublicKey().String(), // sync, err := client.Sync(context.TODO(), &mgmtProto.EncryptedMessage{
Body: encryptedBytes, // WgPubKey: peer.PublicKey().String(),
}) // Body: encryptedBytes,
Expect(err).NotTo(HaveOccurred()) // })
clients = append(clients, sync) // Expect(err).NotTo(HaveOccurred())
// clients = append(clients, sync)
// receive stream //
peer := peer // // receive stream
go func() { // peer := peer
for { // go func() {
encryptedResponse := &mgmtProto.EncryptedMessage{} // for {
err = sync.RecvMsg(encryptedResponse) // encryptedResponse := &mgmtProto.EncryptedMessage{}
if err != nil { // err = sync.RecvMsg(encryptedResponse)
break // if err != nil {
} // break
decryptedBytes, err := encryption.Decrypt(encryptedResponse.Body, serverPubKey, peer) // }
Expect(err).NotTo(HaveOccurred()) // decryptedBytes, err := encryption.Decrypt(encryptedResponse.Body, serverPubKey, peer)
// Expect(err).NotTo(HaveOccurred())
resp := &mgmtProto.SyncResponse{} //
err = pb.Unmarshal(decryptedBytes, resp) // resp := &mgmtProto.SyncResponse{}
Expect(err).NotTo(HaveOccurred()) // err = pb.Unmarshal(decryptedBytes, resp)
if len(resp.GetRemotePeers()) > 0 { // Expect(err).NotTo(HaveOccurred())
// only consider peer updates // if len(resp.GetRemotePeers()) > 0 {
wg.Done() // // only consider peer updates
} // wg.Done()
} // }
}() // }
} // }()
// }
time.Sleep(1 * time.Second) //
for i := 0; i < additionalPeers; i++ { // time.Sleep(1 * time.Second)
key, _ := wgtypes.GenerateKey() // for i := 0; i < additionalPeers; i++ {
loginPeerWithValidSetupKey(serverPubKey, key, client) // key, _ := wgtypes.GenerateKey()
r := rand.New(rand.NewSource(time.Now().UnixNano())) // loginPeerWithValidSetupKey(serverPubKey, key, client)
n := r.Intn(200) // r := rand.New(rand.NewSource(time.Now().UnixNano()))
time.Sleep(time.Duration(n) * time.Millisecond) // n := r.Intn(200)
} // time.Sleep(time.Duration(n) * time.Millisecond)
// }
wg.Wait() //
// wg.Wait()
for _, syncClient := range clients { //
err := syncClient.CloseSend() // for _, syncClient := range clients {
Expect(err).NotTo(HaveOccurred()) // err := syncClient.CloseSend()
} // Expect(err).NotTo(HaveOccurred())
}) // }
}) // })
}) // })
// })
Context("when there are peers registered under one account concurrently", func() { Context("when there are peers registered under one account concurrently", func() {
Specify("then there are no duplicate IPs", func() { Specify("then there are no duplicate IPs", func() {