Skip to content

Commit 4bc86a3

Browse files
committed
dexop: migrate from deprecated grpc.Dial
1 parent cb69d69 commit 4bc86a3

File tree

2 files changed

+12
-5
lines changed

2 files changed

+12
-5
lines changed

go/dexop/dex/client.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ func newDexClient(hostAndPort, caPath, clientKey, clientCrt string) (dexapi.DexC
107107
}
108108
creds := credentials.NewTLS(clientTLSConfig)
109109

110-
conn, err := grpc.Dial(hostAndPort, grpc.WithTransportCredentials(creds))
110+
conn, err := grpc.NewClient(hostAndPort, grpc.WithTransportCredentials(creds))
111111
if err != nil {
112112
return nil, fmt.Errorf("dial: %v", err)
113113
}
@@ -116,7 +116,7 @@ func newDexClient(hostAndPort, caPath, clientKey, clientCrt string) (dexapi.DexC
116116

117117
func NewInsecureTestManager(grpcAddr string) (*DexManager, error) {
118118
// Establish gRPC connection for tests to use
119-
grpcConn, err := grpc.Dial(grpcAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
119+
grpcConn, err := grpc.NewClient(grpcAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
120120
if err != nil {
121121
return nil, err
122122
}

go/dexop/internal/controller/suite_test.go

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ import (
2929
. "github.com/onsi/ginkgo/v2"
3030
. "github.com/onsi/gomega"
3131
"google.golang.org/grpc"
32+
"google.golang.org/grpc/connectivity"
3233
"google.golang.org/grpc/credentials/insecure"
3334

3435
"k8s.io/client-go/kubernetes/scheme"
@@ -38,6 +39,7 @@ import (
3839
logf "sigs.k8s.io/controller-runtime/pkg/log"
3940
"sigs.k8s.io/controller-runtime/pkg/log/zap"
4041

42+
dexapi "github.com/dexidp/dex/api/v2"
4143
dexv1alpha1 "github.com/rackerlabs/understack/go/dexop/api/v1alpha1"
4244
// +kubebuilder:scaffold:imports
4345
)
@@ -105,9 +107,15 @@ var _ = BeforeSuite(func() {
105107
// Wait for Dex gRPC server to be ready
106108
grpcAddr := testDexHostAddr
107109
Eventually(func(g Gomega) {
108-
conn, dialErr := grpc.Dial(grpcAddr, grpc.WithTransportCredentials(insecure.NewCredentials()), grpc.WithBlock(), grpc.WithTimeout(1*time.Second))
110+
conn, dialErr := grpc.NewClient(grpcAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
109111
g.Expect(dialErr).NotTo(HaveOccurred(), "waiting for Dex gRPC server to be ready")
110-
Expect(conn.Close()).To(Succeed())
112+
state := conn.GetState()
113+
g.Expect(state).To(Equal(connectivity.Idle))
114+
result := conn.WaitForStateChange(ctx, connectivity.Ready)
115+
g.Expect(result).To(Equal(true))
116+
err = conn.Invoke(ctx, "/api.Dex/ListPasswords", nil, new(dexapi.ListPasswordResp), grpc.StaticMethod())
117+
g.Expect(err).NotTo(HaveOccurred())
118+
g.Expect(conn.Close()).To(Succeed())
111119
}, 10*time.Second, 1*time.Second).Should(Succeed())
112120

113121
err = dexv1alpha1.AddToScheme(scheme.Scheme)
@@ -133,7 +141,6 @@ var _ = AfterSuite(func() {
133141
_, err := dexCmd.Process.Wait()
134142
if err != nil && err.Error() != "signal: killed" {
135143
GinkgoWriter.Println("error waiting for Dex process to exit: %v", err)
136-
137144
}
138145
GinkgoWriter.Println("Dex server stopped.")
139146
}

0 commit comments

Comments
 (0)