client/isolate: correctly propagate context.Context

client/cmd/isolate/checker_test.go

Index: client/cmd/isolate/checker_test.go
diff --git a/client/cmd/isolate/checker_test.go b/client/cmd/isolate/checker_test.go
index 5c5fdac420a52e00d0d91d608dd67cc6bd9bd653..b7a8e40ea92e38df4baa5cd5d1e02d04e9460ac9 100644
--- a/client/cmd/isolate/checker_test.go
+++ b/client/cmd/isolate/checker_test.go
@@ -36,7 +36,12 @@ type fakeIsolateService struct {
 	errc <-chan error
 }
 
-func (f *fakeIsolateService) Contains(_ context.Context, digests []*service.HandlersEndpointsV1Digest) ([]*isolatedclient.PushState, error) {
+func (f *fakeIsolateService) Contains(ctx context.Context, digests []*service.HandlersEndpointsV1Digest) ([]*isolatedclient.PushState, error) {
+	select {
+	case <-ctx.Done():
+		return nil, ctx.Err()
+	default:
+	}
 	f.mu.Lock()
 	defer f.mu.Unlock()
 	var states []*isolatedclient.PushState