java + selenium测试框架（之下载页面上的图片） 版本演化七

java + selenium测试框架（之下载） 版本演化七

准备攻克验证码验证，知识积累中。

下载感兴趣的图片

从网上找了个程序，挺好用的，直接放这里了。

import java.io.BufferedInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;

public class SaveFileUtil {

    public final static boolean DEBUG = true;//调试用

    private static int BUFFER_SIZE = 8096;

    /**
       * 将HTTP资源另存为文件
       *
       * @param destUrl String
       * @param fileName String
       * @throws Exception
       */
      public static void saveToFile(String destUrl, String fileName) {
        FileOutputStream fos = null;
        BufferedInputStream bis = null;
        HttpURLConnection httpUrl = null;
        URL url = null;
        byte[] buf = new byte[BUFFER_SIZE];
        int size = 0;

        //建立链接
        try {
            url = new URL(destUrl);
            httpUrl = (HttpURLConnection) url.openConnection();
            //连接指定的资源
            httpUrl.connect();
            //获取网络输入流
            bis = new BufferedInputStream(httpUrl.getInputStream());
            //建立文件
            fos = new FileOutputStream(fileName);

            if (DEBUG) 
                System.out.println("正在获取链接[" + destUrl + "]的内容..\n将其保存为文件[" + fileName + "]");

            //保存文件
            while ( (size = bis.read(buf)) != -1) 
              fos.write(buf, 0, size);

        } catch (MalformedURLException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                if(fos!=null)fos.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
            try {
                if(bis!=null)bis.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
            httpUrl.disconnect();
        }
      }
}

Demos

需求：从126.com的网站上，下载个图片，就是那个二维码的小图片。

public class Login126Page extends Page{

    public Login126Page(WebDriver driver) {
        super(driver);
    }

    @FindBy(id="yxCode")
    protected WebElement imgSrc;

    protected String getPicInfo() {
        sleep(2);
        return imgSrc.getAttribute("src");
    }
}

测试用例：

public class Login126Test extends TestBase{

    Login126Page page = null;

    String path = "http://www.126.com/";

    @BeforeMethod
    public void initPage() {
        page = new Login126Page(driver);
        page.openAndWait(path);
        PageFactory.initElements(driver, page);
    }

    @Test
    public void testLogin() {
        String url = page.getPicInfo();
        SaveFileUtil.saveToFile(url, "1.jpg");
        System.out.println(page.getCurrentUrl());
    }
}

写在最后，为了增加点击。需要TestBase，Page的兄弟请看之前的博客。