test: fix flaky test

Helper method that performs initial assigning of master/replica roles
and is widely used in ConnectionPool tests was adjusted to wait for
the roles to be applied successfully.
Prior to this patch it doesn't, so sometimes subsequent test code might
work unexpectedly (the problem was caught with
TestConnectionHandlerOpenUpdateClose)

Closes #452

Author: elhimov

test_helpers/pool_helper.go

@@ -2,8 +2,10 @@ package test_helpers
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"reflect"
+	"sync"
 	"time"
 
 	"github.com/tarantool/go-tarantool/v2"
@@ -206,6 +208,45 @@ func SetInstanceRO(ctx context.Context, dialer tarantool.Dialer, connOpts tarant
 		return err
 	}
 
+	// Wait for the role to be applied.
+	for {
+		time.Sleep(10 * time.Millisecond)
+
+		var reason string
+
+		data, err := conn.Do(tarantool.NewCallRequest("box.info")).Get()
+		switch {
+		case err != nil:
+			reason = fmt.Sprintf("failed to get box.info: %s", err)
+		case len(data) < 1:
+			reason = "box.info is empty"
+		default:
+			status, statusFound := data[0].(map[interface{}]interface{})["status"]
+			readonly, readonlyFound := data[0].(map[interface{}]interface{})["ro"]
+			switch {
+			case !statusFound:
+				reason = "box.info.status is missing"
+			case status != "running":
+				reason = fmt.Sprintf("box.info.status='%s' (waiting for 'running')", status)
+			case !readonlyFound:
+				reason = "box.info.ro is missing"
+			case readonly != isReplica:
+				reason = fmt.Sprintf("box.info.ro='%v' (waiting for '%v')", readonly, isReplica)
+			}
+		}
+
+		if len(reason) == 0 {
+			break
+		}
+
+		select {
+		case <-ctx.Done():
+			return fmt.Errorf("%w: failed to apply role, the last reason: %s", ctx.Err(), reason)
+		default:
+			continue
+		}
+	}
+
 	return nil
 }
 
@@ -215,16 +256,23 @@ func SetClusterRO(dialers []tarantool.Dialer, connOpts tarantool.Opts,
 		return fmt.Errorf("number of servers should be equal to number of roles")
 	}
 
+	ctx, cancel := GetConnectContext()
+	defer cancel()
+
+	// Apply roles in parallel.
+	errs := make([]error, len(dialers))
+	var wg sync.WaitGroup
 	for i, dialer := range dialers {
-		ctx, cancel := GetConnectContext()
-		err := SetInstanceRO(ctx, dialer, connOpts, roles[i])
-		cancel()
-		if err != nil {
-			return err
-		}
+		wg.Add(1)
+		// Pass loop variables to avoid its closure.
+		go func(i int, dialer tarantool.Dialer) {
+			defer wg.Done()
+			errs[i] = SetInstanceRO(ctx, dialer, connOpts, roles[i])
+		}(i, dialer)
 	}
+	wg.Wait()
 
-	return nil
+	return errors.Join(errs...)
 }
 
 func StartTarantoolInstances(instsOpts []StartOpts) ([]*TarantoolInstance, error) {