Selenium WebSocket测试套件中的端口冲突与解决方案

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!");
        setConnectionLostTimeout(0);
        setConnectionLostTimeout(500);
    }

    // 其他方法...

    @Override
    public void onOpen(WebSocket conn, ClientHandshake handshake) {
        System.out.println(conn.getRemoteSocketAddress().getAddress().getHostAddress() + " entered the room!");
    }

    @Override
    public void onClose(WebSocket conn, int code, String reason, boolean remote) {
        System.out.println(conn + " has left the room!");
    }

    @Override
    public void onMessage(WebSocket conn, String message) {
        System.out.println(conn + ": " + message);
        // 处理消息逻辑
    }

    @Override
    public void onError(WebSocket conn, Exception ex) {
        ex.printStackTrace();
        if (conn != null) {
            // some errors like port binding failed may not be assignable to a specific 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 MySeleniumWebSocketTest { // 建议使用更具描述性的类名
    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, IOException { // stop() 方法可能抛出 IOException
        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 {
        // 关闭浏览器实例
        if (driver1 != null) {
            driver1.quit();
        }
        // 关闭WebSocket服务器实例
        if (server != null) {
            try {
                server.stop();
                // 确保服务器有足够时间关闭并释放端口
                Thread.sleep(500); // 短暂等待，确保端口被释放
            } catch (IOException e) {
                System.err.println("Error stopping WebSocket server: " + e.getMessage());
                e.printStackTrace();
            }
        }
    }

    @Test
    void testCaseOne() throws InterruptedException {
        driver1.get(sampleFile.toUri().toString());
        // 确保WebSocket连接已建立
        // 可以通过JS执行判断WebSocket状态，或等待特定元素出现
        // 例如：等待按钮可见并可点击
        WebElement startButton = driver1.findElement(By.id("startButton"));
        WebDriverWait wait = new WebDriverWait(driver1, Duration.ofSeconds(10));
        wait.until(ExpectedConditions.elementToBeClickable(startButton));

        startButton.click();
        // 其他操作
        // assertEquals(2, server.getNextPlayer()); // 假设 server 有 getNextPlayer 方法
    }

    @Test
    void testCaseTwo() throws InterruptedException {
        driver1.get(sampleFile.toUri().toString());
        // 确保WebSocket连接已建立
        WebElement startButton = driver1.findElement(By.id("startButton"));
        WebDriverWait wait = new WebDriverWait(driver1, Duration.ofSeconds(10));
        wait.until(ExpectedConditions.elementToBeClickable(startButton));

        startButton.click();
        // 其他操作
    }
}