From 0224aeb462f8c82caef55df4df47050e71c63e0c Mon Sep 17 00:00:00 2001 From: bilalcaliskan Date: Mon, 3 Jan 2022 14:19:35 +0300 Subject: [PATCH] call cancel() function manually --- internal/raw/raw_test.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/internal/raw/raw_test.go b/internal/raw/raw_test.go index 519e7a1..166a0e6 100644 --- a/internal/raw/raw_test.go +++ b/internal/raw/raw_test.go @@ -36,7 +36,6 @@ func TestStartFlooding(t *testing.T) { for _, tc := range cases { t.Run(tc.name, func(t *testing.T) { ctx, cancel := context.WithTimeout(context.TODO(), time.Duration(tc.floodMilliSeconds)*time.Millisecond) - defer cancel() t.Logf("starting flood, caseName=%s, floodType=%s, floodMilliSeconds=%d\n", tc.name, tc.floodType, tc.floodMilliSeconds) go func() { err := StartFlooding(tc.dstIp, tc.dstPort, tc.payloadLength, tc.floodType) @@ -49,8 +48,10 @@ func TestStartFlooding(t *testing.T) { select { case <-time.After(120 * time.Second): + cancel() t.Log("overslept") case <-ctx.Done(): + cancel() t.Logf("context closed, caseName=%s, floodType=%s, floodMilliSeconds=%d\n", tc.name, tc.floodType, tc.floodMilliSeconds) } })