章节十四、5- web页面的截图

 1 package basicweb;
 2 
 3 import java.io.File;
 4 import java.util.concurrent.TimeUnit;
 5 
 6 import org.apache.commons.io.FileUtils;
 7 import org.junit.jupiter.api.AfterEach;
 8 import org.junit.jupiter.api.BeforeEach;
 9 import org.junit.jupiter.api.Test;
10 import org.openqa.selenium.By;
11 import org.openqa.selenium.Keys;
12 import org.openqa.selenium.OutputType;
13 import org.openqa.selenium.TakesScreenshot;
14 import org.openqa.selenium.WebDriver;
15 import org.openqa.selenium.WebElement;
16 import org.openqa.selenium.chrome.ChromeDriver;
17 
18 
19 
20 class Screenshots {
21     
22     WebDriver driver;
23     String url;
24     @BeforeEach
25     void setUp() throws Exception {
26         driver = new ChromeDriver();
27         url = "https://login.yahoo.com/config/login?.src=fpctx&.intl=us&.lang=en-US&.done=https%3A%2F%2Fwww.yahoo.com";
28         driver.manage().window().maximize();
29         driver.manage().timeouts().implicitlyWait(2000,TimeUnit.SECONDS);
30     }
31 
32     @Test
33     void test() throws InterruptedException {
34         driver.get(url);
35 //        查找到登录文本框，输入test
36         driver.findElement(By.xpath("//input[@name='username']")).sendKeys("test");
37 //        点击登录按钮
38         driver.findElement(By.xpath("//input[@id='login-signin']")).sendKeys(Keys.ENTER);
39     }
40     
41 //    创建一个获取随机字符串的方法用来给截图的内容图片赋值
42 //    在调用该方法时，列如length赋值是5，那么文件名的长度就为5，由5个字符组成
43     public static String getRandomString(int length) {
44 //        StringBuilder 是一个可变的字符序列
45         StringBuilder sb = new StringBuilder();
46         String characters = "qwertyuiopasdfghjklzxcvbnmQWERTYUIOPAZSXDCFVGBHNJMKL1234567890";
47 //        for循环从定义的characters字符串中随机取出指定length长度的字符串
48         for(int i = 0; i<length;i++) {
49 //            Math.random()取0到1之间的一个随机数然后乘以characters字符串的长度，得到一个double类型的数据，因此要强制转换成int类型得到一个整数，成为characters字符串的索引
50             int index = (int)(Math.random() * characters.length());
51 //            通过charAt获得字符串上索引的值，来作为append方法的参数
52 //            append方法是为了把获取到的字符全部连接到一起，先取一个字符放到bulider缓存区中，然后再循环追加取剩下的值，直到满足length长度
53             sb.append(characters.charAt(index));
54             }
55 //        然后将取到的字符串值返回
56         return sb.toString();
57     }
58     
59     @AfterEach
60     void tearDown() throws Exception {
61     Thread.sleep(2000);
62 //    截图首先需要一个文件名,调用上面定义的随机生成文件名的方法,.png表示文件的扩展名
63     String fileName = getRandomString(10)+".png";
64 //    存放图片的路径
65     String directory = "C:\Users\acer\Desktop\";
66 //    以文件的形式输出
67 //    把driver转换为TakesScreenshot类型
68     File sourceFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
69 //    复制文件
70     FileUtils.copyFile(sourceFile,new File(directory+fileName));
71     driver.quit();
72     }
73 }