From eeb688cb5efcf544782ac2091dc9dbcbee2537bf Mon Sep 17 00:00:00 2001 From: Pascal Fischer Date: Fri, 17 Jan 2025 01:08:29 +0100 Subject: [PATCH] cleanup after sync --- management/server/management_test.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/management/server/management_test.go b/management/server/management_test.go index 864a936e7..f7998482d 100644 --- a/management/server/management_test.go +++ b/management/server/management_test.go @@ -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()