Currently tests use `context.TODO()` - this should be replaced with something like: ```go ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) defer cancel() ```
Currently tests use
context.TODO()- this should be replaced with something like: