解决Selenium并发测试WebSocket应用中的端口占用问题

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 MyWebSocketTest { // Renamed Test to MyWebSocketTest to avoid conflict
    WebDriver driver1;
    String path = "path/web.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);

        // 启动WebSocket服务器
        server = new Server(8800);
        server.start();
        // 确保服务器有足够时间启动并监听端口
        Thread.sleep(500); // 可以根据实际情况调整或使用更智能的等待机制
    }

    @AfterEach
    void teardown() throws InterruptedException {
        // 关闭Selenium WebDriver
        if (driver1 != null) {
            driver1.quit();
        }
        // 停止WebSocket服务器并释放端口
        if (server != null) {
            try {
                server.stop();
                // 给予操作系统一些时间来释放端口
                Thread.sleep(500);
            } catch (IOException | InterruptedException e) {
                System.err.println("Error stopping WebSocket server: " + e.getMessage());
                Thread.currentThread().interrupt(); // 重新中断当前线程
            }
        }
    }

    @Test
    void testCaseOne() throws InterruptedException {
        driver1.get(sampleFile.toUri().toString());
        // 确保页面加载完成，并且WebSocket连接已建立
        // 可以添加显式等待，例如等待某个元素可见或JavaScript变量就绪
        WebDriverWait wait = new WebDriverWait(driver1, Duration.ofSeconds(10));
        wait.until(ExpectedConditions.visibilityOfElementLocated(By.id("startButton")));

        driver1.findElement(By.id("startButton")).click();

        // 模拟其他操作
        // assertEquals(2, server.getNextPlayer()); // 假设getNextPlayer方法存在
    }

    @Test
    void testCaseTwo() throws InterruptedException {
        driver1.get(sampleFile.toUri().toString());
        // 确保页面加载完成，并且WebSocket连接已建立
        WebDriverWait wait = new WebDriverWait(driver1, Duration.ofSeconds(10));
        wait.until(ExpectedConditions.visibilityOfElementLocated(By.id("startButton")));

        driver1.findElement(By.id("startButton")).click();
        // 模拟其他操作
    }
}