diff options
author | Jacob Vosmaer (GitLab) <jacob@gitlab.com> | 2017-09-11 15:46:14 +0300 |
---|---|---|
committer | Jacob Vosmaer (GitLab) <jacob@gitlab.com> | 2017-09-11 15:46:14 +0300 |
commit | 5776a29cb41046ebbe4e0741b5ed50862980cdb1 (patch) | |
tree | 189e9dfeeda280124a7e0acd64bc85cd29653b88 /internal/server | |
parent | aea5e82175047b514dbd0e0a75053b1e014c7612 (diff) |
Use context cancellation instead of command.Close
Diffstat (limited to 'internal/server')
-rw-r--r-- | internal/server/auth_test.go | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/internal/server/auth_test.go b/internal/server/auth_test.go index f5c8ae197..a92d61188 100644 --- a/internal/server/auth_test.go +++ b/internal/server/auth_test.go @@ -1,7 +1,6 @@ package server import ( - "context" "net" "testing" "time" @@ -130,8 +129,11 @@ func dial(opts []grpc.DialOption) (*grpc.ClientConn, error) { } func healthCheck(conn *grpc.ClientConn) error { + ctx, cancel := testhelper.Context() + defer cancel() + client := healthpb.NewHealthClient(conn) - _, err := client.Check(context.Background(), &healthpb.HealthCheckRequest{}) + _, err := client.Check(ctx, &healthpb.HealthCheckRequest{}) return err } |