Fixed context cancel function issue

ashishxooa · Sep 17, 2018 · 880f8ec · 880f8ec
1 parent ee55b01
commit 880f8ec
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/monitor/monitor_test.go b/monitor/monitor_test.go
@@ -234,7 +234,8 @@ func TestRun(t *testing.T) {
 			Client:    mockClient,
 		}
 
-		ctx, _ := context.WithTimeout(context.Background(), time.Millisecond*25)
+		ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*25)
+		defer cancel()
 		err := Run(c, ctx)
 
 		assert.NoError(t, err)