sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-05-03 23:56:38 +00: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

5
management/server/management_test.go
View File

@@ -316,6 +316,8 @@ var _ = Describe("Management service", func() {
initialPeers := 10
additionalPeers := 10
ctx, cancel := context.WithCancel(context.Background())
var peers []wgtypes.Key
for i := 0; i < initialPeers; i++ {
key, _ := wgtypes.GenerateKey()
@@ -335,7 +337,7 @@ var _ = Describe("Management service", func() {
Expect(err).NotTo(HaveOccurred())
// open stream
sync, err := client.Sync(context.TODO(), &mgmtProto.EncryptedMessage{
sync, err := client.Sync(ctx, &mgmtProto.EncryptedMessage{
WgPubKey: peer.PublicKey().String(),
Body: encryptedBytes,
})
@@ -377,6 +379,7 @@ var _ = Describe("Management service", func() {
}
wg.Wait()
cancel()
for _, syncClient := range clients {
err := syncClient.CloseSend()