Context cancelling should be tied to the test execution (#96)

Signed-off-by: Francesco Guardiani <francescoguard@gmail.com>
knative-extensions · Feb 22, 2021 · 712fadf · 712fadf
1 parent be32f9c
commit 712fadf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/environment/magic.go b/pkg/environment/magic.go
@@ -150,6 +150,9 @@ func (mr *MagicEnvironment) Test(ctx context.Context, t *testing.T, f *feature.F
 					continue
 				}
 				t.Run(s.TestName(), func(t *testing.T) {
+					ctx, cancelFn := context.WithCancel(ctx)
+					t.Cleanup(cancelFn)
+
 					wg.Add(1)
 					defer wg.Done()