expand lint rule to avoid dealing with context.WithCancel cases

Author: hasson82

scripts/vet.sh

@@ -50,8 +50,8 @@ not grep 'func Test[^(]' -- test/*.go
 git grep 'func (s) ' -- "*_test.go" | not grep -v 'func (s) Test'
 git grep 'func [A-Z]' -- "*_test.go" | not grep -v 'func Test\|Benchmark\|Example'
 
-# - Make sure not context usage are done without timeout.
-git grep 'context.Background()' -- "*_test.go" | not grep -v 'context.WithTimeout(context.Background()'
+# - Make sure all context usages are done timeout.
+git grep -e 'context.Background()' --or -e 'context.TODO()' -- "*_test.go" | not grep -v 'context.WithTimeout(context.Background()' | not grep -v 'context.WithCancel(context.Background' | not grep -v 'context.WithTimeOut(context.TODO()'
 
 # - Do not use time.After except in tests.  It has the potential to leak the
 #   timer since there is no way to stop it early.