解决Selenium测试WebSocket应用并发执行失败的问题

public class Test {
    WebDriver driver1;
    String path = "path/web.html";
    Path sampleFile;
    Server server;
    JavascriptExecutor js1;

    @BeforeAll
    static void setupClass() {
        WebDriverManager.chromedriver().setup();
    }

    @BeforeEach
    void setup() throws InterruptedException {
        driver1 = new ChromeDriver();
        js1 = (JavascriptExecutor) driver1;
        sampleFile = Paths.get(path);
        server = new Server(8800); // 在每个测试前启动服务器
        server.start();
    }

    @AfterEach
    void teardown() throws InterruptedException {
        driver1.quit(); // 仅关闭浏览器
    }

    @Test
    void row41() throws InterruptedException {
        driver1.get(sampleFile.toUri().toString());
        driver1.findElement(By.id("startButton")).click();
        // ... some other operations
        assertEquals(2, server.getNextPlayer());
    }

    @Test
    void row42() throws InterruptedException {
        driver1.get(sampleFile.toUri().toString());
        driver1.findElement(By.id("startButton")).click();
        // ... some other operations
    }
}

import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.openqa.selenium.*;
import org.openqa.selenium.chrome.ChromeDriver;

import io.github.bonigarcia.wdm.WebDriverManager;
import java.nio.file.Path;
import java.nio.file.Paths;
import static org.junit.jupiter.api.Assertions.*;

public class TestClass { // 建议类名不要直接使用Test，避免与JUnit的@Test注解混淆
    WebDriver driver1;
    String path = "path/web.html"; // 请替换为您的HTML文件实际路径
    Path sampleFile;
    Server server; // 您的WebSocket服务器实例
    JavascriptExecutor js1;

    @BeforeAll
    static void setupClass() {
        WebDriverManager.chromedriver().setup();
    }

    @BeforeEach
    void setup() throws InterruptedException {
        driver1 = new ChromeDriver();
        js1 = (JavascriptExecutor) driver1;
        sampleFile = Paths.get(path);
        // 确保服务器在每个测试开始前启动
        server = new Server(8800);
        server.start();
        // 可以在此处添加一个短暂的等待，确保服务器完全启动并监听
        Thread.sleep(500); 
    }

    @AfterEach
    void teardown() throws InterruptedException {
        if (driver1 != null) {
            driver1.quit(); // 关闭WebDriver实例
        }
        if (server != null) {
            // 优雅地关闭WebSocket服务器，释放端口
            // 可以选择带超时参数的stop方法，例如 server.stop(1000)
            server.stop(); 
            // 可以在此处添加一个短暂的等待，确保服务器完全关闭
            Thread.sleep(500); 
        }
    }

    @Test
    void testRow41() throws InterruptedException {
        driver1.get(sampleFile.toUri().toString());
        // 确保页面加载并WebSocket连接建立
        // 可以使用显式等待等待按钮可见或WebSocket连接状态
        // 例如：WebDriverWait wait = new WebDriverWait(driver1, Duration.ofSeconds(10));
        // wait.until(ExpectedConditions.visibilityOfElementLocated(By.id("startButton")));
        driver1.findElement(By.id("startButton")).click();
        // ... some other operations
        // 假设getNextPlayer()是Server类的一个方法
        // assertEquals(2, server.getNextPlayer()); 
    }

    @Test
    void testRow42() throws InterruptedException {
        driver1.get(sampleFile.toUri().toString());
        // 确保页面加载并WebSocket连接建立
        driver1.findElement(By.id("startButton")).click();
        // ... some other operations
    }
}

import org.java_websocket.WebSocket;
import org.java_websocket.handshake.ClientHandshake;
import org.java_websocket.server.WebSocketServer;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.util.*;

public class Server extends WebSocketServer {
    public Server(int port) {
        super(new InetSocketAddress(port));
    }

    @Override
    public void onStart() {
        System.out.println("Server started on port " + getPort() + "!");
        setConnectionLostTimeout(0);
        setConnectionLostTimeout(500);
    }

    @Override
    public void onOpen(WebSocket conn, ClientHandshake handshake) {
        System.out.println("New connection from " + conn.getRemoteSocketAddress());
        // ...
    }

    @Override
    public void onClose(WebSocket conn, int code, String reason, boolean remote) {
        System.out.println("Closed connection to " + conn.getRemoteSocketAddress() + " with exit code " + code + " reason " + reason);
        // ...
    }

    @Override
    public onMessage(WebSocket conn, String message) {
        System.out.println("Received message from " + conn.getRemoteSocketAddress() + ": " + message);
        // ...
    }

    @Override
    public void onError(WebSocket conn, Exception ex) {
        System.err.println("An error occurred on connection " + conn.getRemoteSocketAddress() + ":" + ex);
        // ...
    }

    // 假设您有类似的方法来获取游戏状态
    public int getNextPlayer() {
        // ... 实现获取下一个玩家逻辑
        return 0; 
    }
}

<html>
<head>
    <meta charset="UTF-8">
    <title>something</title>
</head>
<body>
<div ><button id="startButton" onclick="start()" style="visibility: hidden;">START</button></div>
<script type="text/javascript">
        var webSocket; // 声明为全局变量或确保在函数中可访问
        var startButton = document.getElementById("startButton");

        function initWebSocket() {
            webSocket = new WebSocket("ws://localhost:8800");

            webSocket.onopen = function(event){ wsOpen(event);};
            webSocket.onmessage = function(event){ wsGetMessage(event);};
            webSocket.onclose = function(event){ wsClose(event);};
            webSocket.onerror = function(event){ wsError(event);};
        }

        // 在页面加载完成后初始化WebSocket连接
        window.onload = function() {
            initWebSocket();
        };

        function start(){
            if (webSocket && webSocket.readyState === WebSocket.OPEN) {
                wsSendMessage("start,"+player_id); // 假设player_id已定义
                startButton.innerText = "STARTED";
                startButton.disabled="true";
            } else {
                console.warn("WebSocket is not open. Cannot send message.");
            }
        }
        function wsSendMessage(message){
            webSocket.send(message);
        }

        function wsOpen(event) {
            console.log("WebSocket connection opened.");
            startButton.style.visibility = "visible"; // 连接成功后显示按钮
            startButton.disabled = false; // 启用按钮
        }

        function wsGetMessage(event) {
            console.log("Received message: " + event.data);
            // ...
        }

        function wsClose(event) {
            console.log("WebSocket connection closed.");
            startButton.style.visibility = "hidden"; // 连接关闭后隐藏按钮
            startButton.disabled = true; // 禁用按钮
        }

        function wsError(event) {
            console.error("WebSocket error: ", event);
            startButton.style.visibility = "hidden"; // 错误时隐藏按钮
            startButton.disabled = true; // 禁用按钮
        }
    </script>
</body>
</html>