fix: adding on success to dialer

getlantern · Nov 19, 2024 · 04bb1c2 · 04bb1c2
1 parent b82c397
commit 04bb1c2
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/client/client_test.go b/client/client_test.go
@@ -72,13 +72,22 @@ func newTestUserConfig() *common.UserConfigData {
 }
 
 func resetDialers(client *Client, dial func(network, addr string) (net.Conn, error)) {
+	rdyC := make(chan struct{}, 1)
 	d := dialer.New(&dialer.Options{
 		Dialers: []dialer.ProxyDialer{&testDialer{
 			name: "test-dialer",
 			dial: dial,
 		}},
+		OnSuccess: func(dialer dialer.ProxyDialer) {
+			select {
+			case rdyC <- struct{}{}:
+			default:
+			}
+		},
 	})
 	client.dialer.set(d)
+	// Wait for the dialer to be ready
+	<-rdyC
 }
 
 func newClient() *Client {