Skip to content

Commit 2012f9c

Browse files
committed
transport credentials
1 parent c08f8ae commit 2012f9c

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

connection_e2e_test.go

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@ import (
77
"context"
88
"crypto/tls"
99
"fmt"
10+
grpcCredentials "google.golang.org/grpc/credentials"
11+
"google.golang.org/grpc/credentials/insecure"
1012
"net/url"
1113
"os"
1214
"testing"
@@ -335,7 +337,17 @@ func TestStaticCredentials(t *testing.T) {
335337
t.Fatal(err)
336338
}
337339

338-
staticCredentials := credentials.NewStaticCredentials("root", "", url.Host)
340+
staticCredentials := credentials.NewStaticCredentials("root", "", url.Host, func() grpc.DialOption {
341+
if url.Scheme == "grpcs" {
342+
transportCredentials, err := grpcCredentials.NewClientTLSFromFile(os.Getenv("YDB_SSL_ROOT_CERTIFICATES_FILE"), "")
343+
if err != nil {
344+
t.Fatalf("cannot create transport credentials: %v", err)
345+
}
346+
return grpc.WithTransportCredentials(transportCredentials)
347+
}
348+
return grpc.WithTransportCredentials(insecure.NewCredentials())
349+
}())
350+
339351
token, err := staticCredentials.Token(ctx)
340352
if err != nil {
341353
t.Fatalf("get token failed: %v", err)

0 commit comments

Comments
 (0)