sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-04-25 03:36:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

add logs

Browse Source
This commit is contained in:
Pascal Fischer
2025-01-17 00:35:58 +01:00
parent 66f003d00f
commit 833fec0029
1 changed files with 75 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

148
management/server/management_test.go
View File

@@ -310,79 +310,81 @@ 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{}
// wgCounter := initialPeers + initialPeers*additionalPeers wgCounter := initialPeers + initialPeers*additionalPeers
// wg.Add(wgCounter) wg.Add(wgCounter)
//
// var clients []mgmtProto.ManagementService_SyncClient var clients []mgmtProto.ManagementService_SyncClient
// for _, peer := range peers { for _, peer := range peers {
// messageBytes, err := pb.Marshal(&mgmtProto.SyncRequest{Meta: &mgmtProto.PeerSystemMeta{}}) messageBytes, err := pb.Marshal(&mgmtProto.SyncRequest{Meta: &mgmtProto.PeerSystemMeta{}})
// Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
// encryptedBytes, err := encryption.Encrypt(messageBytes, serverPubKey, peer) encryptedBytes, err := encryption.Encrypt(messageBytes, serverPubKey, peer)
// Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
//
// // open stream // open stream
// sync, err := client.Sync(context.TODO(), &mgmtProto.EncryptedMessage{ sync, err := client.Sync(context.TODO(), &mgmtProto.EncryptedMessage{
// WgPubKey: peer.PublicKey().String(), WgPubKey: peer.PublicKey().String(),
// Body: encryptedBytes, Body: encryptedBytes,
// }) })
// Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
// clients = append(clients, sync) clients = append(clients, sync)
//
// // receive stream // receive stream
// peer := peer go func() {
// go func() { for {
// for { encryptedResponse := &mgmtProto.EncryptedMessage{}
// encryptedResponse := &mgmtProto.EncryptedMessage{} err = sync.RecvMsg(encryptedResponse)
// err = sync.RecvMsg(encryptedResponse) if err != nil {
// if err != nil { break
// break }
// } decryptedBytes, err := encryption.Decrypt(encryptedResponse.Body, serverPubKey, peer)
// decryptedBytes, err := encryption.Decrypt(encryptedResponse.Body, serverPubKey, peer) Expect(err).NotTo(HaveOccurred())
// Expect(err).NotTo(HaveOccurred())
// resp := &mgmtProto.SyncResponse{}
// resp := &mgmtProto.SyncResponse{} err = pb.Unmarshal(decryptedBytes, resp)
// err = pb.Unmarshal(decryptedBytes, resp) Expect(err).NotTo(HaveOccurred())
// Expect(err).NotTo(HaveOccurred()) log.Println("Received Sync response in tests")
// if len(resp.GetRemotePeers()) > 0 { if len(resp.GetRemotePeers()) > 0 {
// // only consider peer updates // only consider peer updates
// wg.Done() wg.Done()
// } } else {
// } log.Println("Received empty Sync response in tests")
// }() }
// } }
// }()
// time.Sleep(1 * time.Second) }
// for i := 0; i < additionalPeers; i++ {
// key, _ := wgtypes.GenerateKey() time.Sleep(1 * time.Second)
// loginPeerWithValidSetupKey(serverPubKey, key, client) for i := 0; i < additionalPeers; i++ {
// r := rand.New(rand.NewSource(time.Now().UnixNano())) key, _ := wgtypes.GenerateKey()
// n := r.Intn(200) loginPeerWithValidSetupKey(serverPubKey, key, client)
// time.Sleep(time.Duration(n) * time.Millisecond) r := rand.New(rand.NewSource(time.Now().UnixNano()))
// } n := r.Intn(200)
// time.Sleep(time.Duration(n) * time.Millisecond)
// wg.Wait() }
//
// for _, syncClient := range clients { wg.Wait()
// err := syncClient.CloseSend()
// Expect(err).NotTo(HaveOccurred()) for _, syncClient := range clients {
// } 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() {