sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-10-23 13:59:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

cleanup after sync

Browse Source
This commit is contained in:
Pascal Fischer
2025-01-17 01:08:29 +01:00
parent f78b629f40
commit eeb688cb5e
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
management/server/management_test.go
+4 -1
View File
@@ -316,6 +316,8 @@ var _ = Describe("Management service", func() {
initialPeers := 10 initialPeers := 10
additionalPeers := 10 additionalPeers := 10
ctx, cancel := context.WithCancel(context.Background())
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()
@@ -335,7 +337,7 @@ var _ = Describe("Management service", func() {
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
// open stream // open stream
sync, err := client.Sync(context.TODO(), &mgmtProto.EncryptedMessage{ sync, err := client.Sync(ctx, &mgmtProto.EncryptedMessage{
WgPubKey: peer.PublicKey().String(), WgPubKey: peer.PublicKey().String(),
Body: encryptedBytes, Body: encryptedBytes,
}) })
@@ -377,6 +379,7 @@ var _ = Describe("Management service", func() {
} }
wg.Wait() wg.Wait()
cancel()
for _, syncClient := range clients { for _, syncClient := range clients {
err := syncClient.CloseSend() err := syncClient.CloseSend()