diff --git a/client/auth.go b/client/auth.go index 0d62ad240..843ca077a 100644 --- a/client/auth.go +++ b/client/auth.go @@ -59,8 +59,10 @@ func (i *AuthInterceptor) WrapStreamingClient(next connect.StreamingClientFunc) ctx context.Context, spec connect.Spec, ) connect.StreamingClientConn { - ctx.Value(types.APIKeyKey) - return next(ctx, spec) + conn := next(ctx, spec) + conn.RequestHeader().Set(types.APIKeyKey, i.apiKey) + + return conn } } diff --git a/test/integration/main_test.go b/test/integration/main_test.go index e9a77e80e..867634420 100644 --- a/test/integration/main_test.go +++ b/test/integration/main_test.go @@ -1,4 +1,4 @@ -//gdo:build integration +//go:build integration /* * Copyright 2020 The Yorkie Authors. All rights reserved.