Selenium实现右键保存图片（Java）

1.代码

public class SaveImage extends TestCase {
    private WebDriver driver;
    private Actions action;
    private String baseUrl;

    @Before
    public void setUp() throws Exception {
        baseUrl = "http://baike.pcbaby.com.cn/recipe/";
        System.setProperty("webdriver.chrome.driver",
                "C:\Users\test\AppData\Local\Google\Chrome\Application\chromedriver.exe");
        driver = new ChromeDriver();
        driver.manage().timeouts().pageLoadTimeout(10, TimeUnit.SECONDS);
        action = new Actions(driver);
    }

    @Test
    public void testDownload() throws Exception {
        driver.get(baseUrl);
        WebElement info = new WebDriverWait(driver, 10)
                .until(ExpectedConditions.presenceOfElementLocated(By.xpath(".//a[2]/em")));
        assertEquals("食谱", info.getText());
        driver.findElement(By.linkText("1")).click();
        WebElement image = driver.findElement(By.xpath(".//*[@class='pic']/a/img"));
        action.contextClick(image).build().perform();
        try {
            /*
             * 使用Actions没办法拿到右键菜单 action.sendKeys(Keys.ARROW_DOWN).perform();
             * action.sendKeys(Keys.ARROW_DOWN).perform();
             * action.sendKeys(Keys.ENTER).perform();
             * 使用Java原生键盘操作类Robot
             */
            Robot robot = new Robot();
            robot.keyPress(KeyEvent.VK_DOWN);
            Thread.sleep(1000);
            robot.keyPress(KeyEvent.VK_DOWN);
            Thread.sleep(1000);
            robot.keyPress(KeyEvent.VK_DOWN);
            Thread.sleep(1000);
            robot.keyPress(KeyEvent.VK_DOWN);
            Thread.sleep(1000);
            robot.keyPress(KeyEvent.VK_DOWN);
            Thread.sleep(1000);
            robot.keyPress(KeyEvent.VK_DOWN);
            Thread.sleep(1000);
            robot.keyPress(KeyEvent.VK_DOWN);
            Thread.sleep(1000);

            robot.keyRelease(KeyEvent.VK_DOWN);
            Thread.sleep(1000);
            robot.keyPress(KeyEvent.VK_ENTER);
        } catch (Exception e) {
            e.printStackTrace();
        }

        try {
            Thread.sleep(2000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        try {
            //使用autoit实现保存
            Runtime.getRuntime().exec("E:\storage\selenium\download\test.exe");
            //等待保存完成
            Thread.sleep(10000);
        } catch (Exception e) {
            e.printStackTrace();
        }

    }

    @After
    public void tearDown() throws Exception {
        driver.quit();
    }
}

2.使用autoit实现保存

使用autoit拖动Finder Tool到需要获取界面信息的输入框及按钮。

3.编写autoit实现保存代码

ControlFocus("另存为", "","Edit1")
;ControlFocus("title","text",controlID) Edit1=Edit instance 1
WinWait("[CLASS:#32770]","",10)
$text = ControlGetText("另存为", "","Edit1")
$filename = "E:storageseleniumdownload" + $text
Sleep(1000)
ControlSetText("另存为", "", "Edit1", $filename)
Sleep(5000)
;ControlFocus("另存为", "","Button1")
ControlClick("另存为","保存(&S)","Button1")

4.使用autoit的Compile Script to .exe工具将该au3文件转为.exe可执行文件