Skip to content

Commit 50a88e8

Browse files
fixed bug in integration test
1 parent 4c2c56a commit 50a88e8

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

tests/it/utils.go

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,6 @@ import (
2727
"google.golang.org/grpc"
2828
"google.golang.org/grpc/credentials/insecure"
2929
"google.golang.org/grpc/resolver"
30-
31-
"k8s.io/apimachinery/pkg/util/wait"
3230
"k8s.io/klog/v2"
3331

3432
"sigs.k8s.io/ibm-powervs-block-csi-driver/pkg/driver"
@@ -121,15 +119,13 @@ func newCSIClient() (*CSIClient, error) {
121119
return nil, err
122120
}
123121
var conn net.Conn
124-
err = wait.PollUntilContextTimeout(context.Background(), 10*time.Second, 3*time.Minute, true, func(context.Context) (bool, error) {
122+
Eventually(func() error {
125123
conn, err = net.Dial(scheme, addr)
126124
if err != nil {
127125
klog.Warningf("Client failed to dial endpoint %v", endpoint)
128-
return false, err
129126
}
130-
klog.Infof("Client succeeded to dial endpoint %v", endpoint)
131-
return true, nil
132-
})
127+
return err
128+
}, 3*time.Minute, 10*time.Second).Should(Succeed())
133129
if err != nil || conn == nil {
134130
return nil, fmt.Errorf("failed to get client connection: %v", err)
135131
}

0 commit comments

Comments
 (0)