如何在selenium中使用Junit单击模式窗口中的关闭图像

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;

public class CloseImageTest {
    private WebDriver driver;

    @Before
    public void setUp() {
        // 设置ChromeDriver路径
        System.setProperty("webdriver.chrome.driver", "path/to/chromedriver");

        // 创建ChromeDriver实例
        driver = new ChromeDriver();
    }

    @Test
    public void testCloseImage() {
        // 打开模式窗口
        driver.get("http://example.com");

        // 切换到模式窗口的句柄
        String mainWindowHandle = driver.getWindowHandle();
        // ...

        // 定位关闭图像元素
        WebElement closeImage = driver.findElement(By.id("close-image"));

        // 模拟点击关闭图像
        closeImage.click();

        // 切换回主窗口
        driver.switchTo().window(mainWindowHandle);
    }

    @After
    public void tearDown() {
        // 关闭浏览器
        driver.quit();
    }
}