Fix integration tests

Author: lukasz-antoniak

.github/workflows/main.yml

@@ -51,8 +51,30 @@ jobs:
           key: ${{ runner.os }}-go-${{ hashFiles('go.mod') }}
           restore-keys: |
             ${{ runner.os }}-go-
+      - name: Install Java
+        run: |
+          curl -s "https://get.sdkman.io" | bash
+          source "$HOME/.sdkman/bin/sdkman-init.sh"
+          echo "sdkman_auto_answer=true" >> ~/.sdkman/etc/config
+          # sdk list java
+
+          sdk install java 11.0.24-zulu
+          echo "JAVA11_HOME=$JAVA_HOME_11_X64" >> $GITHUB_ENV
+
+          sdk install java 17.0.12-zulu
+          echo "JAVA17_HOME=$JAVA_HOME_17_X64" >> $GITHUB_ENV
+
+          # by default use JDK 11
+          sdk default java 11.0.24-zulu
+          sdk use java 11.0.24-zulu
+          echo "JAVA_HOME=$JAVA_HOME_11_X64" >> $GITHUB_ENV
+          echo "PATH=$PATH" >> $GITHUB_ENV
       - name: Install CCM
-        run: pip install "git+https://github.com/riptano/ccm.git@${CCM_VERSION}"
+        run: |
+          python3 -m venv ~/venv
+          ~/venv/bin/pip install setuptools
+          ~/venv/bin/pip install "git+https://github.com/riptano/ccm.git@${CCM_VERSION}"
+          echo "PATH=~/venv/bin:$PATH" >> $GITHUB_ENV
       - name: Start cassandra nodes
         run: |
           VERSION=${{ matrix.cassandra_version }}
@@ -135,8 +157,30 @@ jobs:
       - uses: actions/setup-go@v4
         with:
           go-version: ${{ matrix.go }}
+      - name: Install Java
+        run: |
+          curl -s "https://get.sdkman.io" | bash
+          source "$HOME/.sdkman/bin/sdkman-init.sh"
+          echo "sdkman_auto_answer=true" >> ~/.sdkman/etc/config
+          # sdk list java
+
+          sdk install java 11.0.24-zulu
+          echo "JAVA11_HOME=$JAVA_HOME_11_X64" >> $GITHUB_ENV
+
+          sdk install java 17.0.12-zulu
+          echo "JAVA17_HOME=$JAVA_HOME_17_X64" >> $GITHUB_ENV
+
+          # by default use JDK 11
+          sdk default java 11.0.24-zulu
+          sdk use java 11.0.24-zulu
+          echo "JAVA_HOME=$JAVA_HOME_11_X64" >> $GITHUB_ENV
+          echo "PATH=$PATH" >> $GITHUB_ENV
       - name: Install CCM
-        run: pip install "git+https://github.com/riptano/ccm.git@${CCM_VERSION}"
+        run: |
+          python3 -m venv ~/venv
+          ~/venv/bin/pip install setuptools
+          ~/venv/bin/pip install "git+https://github.com/riptano/ccm.git@${CCM_VERSION}"
+          echo "PATH=~/venv/bin:$PATH" >> $GITHUB_ENV
       - name: Start cassandra nodes
         run: |
           VERSION=${{ matrix.cassandra_version }}

control_ccm_test.go

@@ -55,7 +55,7 @@ func (f *TestHostFilter) SetAllowedHosts(hosts map[string]ccm.Host) {
 }
 
 func TestControlConn_ReconnectRefreshesRing(t *testing.T) {
-	if err := ccm.AllUp(); err != nil {
+	if err := ccm.AllUpWait(); err != nil {
 		t.Fatal(err)
 	}
 

internal/ccm/ccm.go

@@ -35,6 +35,7 @@ import (
 	"os/exec"
 	"runtime"
 	"strings"
+	"time"
 )
 
 func execCmd(args ...string) (*bytes.Buffer, error) {
@@ -71,6 +72,19 @@ func AllUp() error {
 	return nil
 }
 
+func AllUpWait() error {
+	deadline := time.Now().Add(60 * time.Second)
+	observedErr := errors.New("placeholder")
+	for {
+		observedErr = AllUp()
+		if observedErr == nil || time.Now().After(deadline) {
+			return observedErr
+		}
+		time.Sleep(500 * time.Millisecond)
+	}
+	return observedErr
+}
+
 func NodeUp(node string) error {
 	args := []string{node, "start", "--wait-for-binary-proto"}
 	if runtime.GOOS == "windows" {