SpringBoot+Selenium模拟用户操作浏览器
Selenium
Selenium是一个用于Web应用程序自动化测试的开源工具套件。它主要用于以下目的：
- 浏览器自动化：Selenium能够模拟真实用户在不同浏览器（如Chrome、Firefox、IE/Edge等）中的交互行为，通过编程方式控制浏览器执行一系列操作，例如点击按钮、填写表单、导航页面等。
- 兼容性测试：通过编写脚本在多种浏览器和操作系统环境下运行测试，确保Web应用在不同组合下都能正确工作。
- 功能测试：创建回归测试用例来验证软件功能是否按预期工作，以及用户需求是否得到满足。
- 端到端测试：进行从用户界面到后端服务的整体流程测试，检查系统组件间的集成效果。
- 支持多种语言：Selenium WebDriver API 可以使用Java、Python、C#、Ruby等多种编程语言实现测试脚本。
- 与DevTools集成：在较新的Selenium版本中，提供了对Chromium内嵌的Chrome DevTools的支持，允许开发者执行更深层次的浏览器操作和调试任务。
总的来说，Selenium帮助开发团队提高Web应用的质量保障效率，通过自动化测试减少手动测试的重复性和复杂性，并有助于持续集成和持续部署（CI/CD）流程的实施。
下载Selenium
Selenium的下载通常分为两个步骤：首先，你需要下载并安装适用于你开发环境的Selenium WebDriver库（针对不同编程语言有不同的库）；其次，根据你的测试需求下载对应浏览器的驱动程序。
以下是一些常见编程语言中Selenium WebDriver库的下载和安装方式：
Python：
使用pip工具进行安装：
pip install selenium
Java：
在Maven项目中添加如下依赖到pom.xml文件：
<dependency>
<groupId>org.seleniumhq.selenium</groupId>
<artifactId>selenium-java</artifactId>
<version>最新版本号</version>
</dependency>
或在Gradle项目中添加：
implementation 'org.seleniumhq.selenium:selenium-java:最新版本号'
如果不使用构建工具，可以直接从Maven仓库下载selenium-java.jar包。
C# ：
对于.NET环境，可以通过NuGet包管理器在Visual Studio中安装Selenium.WebDriver NuGet包。
下载浏览器驱动
安装完WebDriver库后，接下来需要下载浏览器驱动，例如：
- ChromeDriver：从ChromeDriver官网（https://sites.google.com/a/chromium.org/chromedriver/downloads）下载与你的Chrome浏览器版本相匹配的驱动。
- GeckoDriver（Firefox）：访问GeckoDriver GitHub发布页面（https://github.com/mozilla/geckodriver/releases）下载适合你Firefox版本的驱动。
- EdgeDriver 或 IEDriverServer：分别在Microsoft WebDriver下载页面和Selenium官方IEdriver下载页获取相应驱动。
将下载的浏览器驱动放在系统的PATH路径下，或者在代码中明确指定驱动路径，即可配合WebDriver开始编写自动化测试脚本了。
获取最新版本的chrome和chromeDriver https://sites.google.com/a/chromium.org/chromedriver/downloads
Java代码
yml配置文件
system:
#驱动信息
driver:
mode: 1 # 驱动模式（0：本地 1：远程）
headless: 0 #是否显示浏览器（0：显示 1：不显示）
name: webdriver.chrome.driver
path: D:\drivers\chromedriver-win64\chromedriver.exe
# 远程driver驱动地址
remoteUrl: http://127.0.0.1:9515
systemUrl: http://目标服务器地址
工具类：包含打开、关闭、刷新浏览器功能
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.openqa.selenium.Alert;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.env.Environment;
import org.springframework.stereotype.Component;
import javax.annotation.PostConstruct;
import java.net.ConnectException;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.Set;
import java.util.concurrent.TimeUnit;
@Slf4j
@Component
public class WebDriverUtils {
@Autowired
private Environment env;
/**
* 是否显示浏览器（0：否 1：是）
*/
private static String headless;
/**
* 驱动远程地址
*/
private static String driverRemoteUrl;
private static String systemUrl;
@PostConstruct
public void init() {
headless = env.getProperty("system.driver.headless");
driverRemoteUrl = env.getProperty("system.driver.remoteUrl");
systemUrl = env.getProperty("system.driver.systemUrl");
}
/**
* 等待时间
*/
public final static int timeSeconds = 10;
private static String getRemoteUrl(String driverRemoteUrl) {
String remoteUrl = null;
if (StringUtils.isNotBlank(driverRemoteUrl)) {
if (driverRemoteUrl.contains(",")) {
String[] split = driverRemoteUrl.split(",");
remoteUrl = split[(new Random()).nextInt(split.length)];
} else {
remoteUrl = driverRemoteUrl;
}
}
return remoteUrl;
}
/**
* 获取驱动
*
* @return
* @throws ConnectException
*/
public static WebDriver getDriver() {
log.info("进入启动浏览器");
ChromeOptions chromeOptions = new ChromeOptions();
// 设置浏览器是否可见
if ("1".equals(headless)) {
chromeOptions.addArguments("--headless");
chromeOptions.addArguments("--disable-gpu");
// headless模式下必须设置这两个参数，否则会报连接超时
chromeOptions.addArguments("--proxy-server='direct://'");
chromeOptions.addArguments("--proxy-bypass-list=*");
} else {
chromeOptions.addArguments("--no-sandbox");
}
WebDriver driver = null;
log.info(String.format("driverRemoteUrl:%s", driverRemoteUrl));
DesiredCapabilities dc = DesiredCapabilities.chrome();
dc.setCapability(ChromeOptions.CAPABILITY, chromeOptions);
try {
driver = new RemoteWebDriver(new URL(getRemoteUrl(driverRemoteUrl)), dc);
} catch (Exception e) {
e.printStackTrace();
throw new ApplicationException(-1, "RemoteWebDriver连接远程驱动失败");
}
driver.manage().window().maximize();
driver.manage().timeouts().implicitlyWait(timeSeconds, TimeUnit.SECONDS);
driver.manage().timeouts().pageLoadTimeout(timeSeconds, TimeUnit.SECONDS);
driver.manage().timeouts().setScriptTimeout(timeSeconds, TimeUnit.SECONDS);
return driver;
}
/**
* 关闭浏览器驱动
*
* @param driver
*/
public static void quitBrowser(WebDriver driver) {
try {
if (driver != null) {
driver.close();
driver.quit();
log.info("浏览器关闭成功");
}
} catch (Exception e) {
log.error("浏览器关闭异常 - {}", e.getMessage());
throw new ApplicationException(-1, e.toString());
}
}
public static synchronized boolean openBrowser(WebDriver driver) {
log.info("即将打开浏览器");
try {
driver.get(systemUrl);
return true;
} catch (Exception e) {
log.error("打开浏览器异常,{}",e);
return false;
}
}
/**
* 刷新浏览器
* @param driver
* @return
*/
public static synchronized void refreshBrowser(WebDriver driver) {
driver.navigate().refresh();
}
private static boolean alertExists(float timeout, WebDriver driver) {
long start = System.currentTimeMillis();
while ((System.currentTimeMillis() - start) < timeout * 1000) {
try {
driver.switchTo().alert();
return true;
} catch (Exception e) {
}
}
return false;
}
/**
* 关闭登录消息提醒弹窗
*
* @param timeout
* @param driver
*/
private static void closePrompt(float timeout, WebDriver driver) {
long start = System.currentTimeMillis();
while ((System.currentTimeMillis() - start) < timeout * 1000) {
try {
close_ymPrompt(driver);
} catch (Exception e) {
log.error("无法找到登陆消息提醒弹窗");
} finally {
sleep("500");
}
}
}
public static void close_ymPrompt(WebDriver driver) {
WebElement closePromptEle = driver.findElement(By.className("ymPrompt_close"));
closePromptEle.click();
}
/**
* 点击弹窗
*
* @param driver
*/
private static void clickConfirmEle(WebDriver driver) {
try {
long start = System.currentTimeMillis();
// 点击确定
while (System.currentTimeMillis() - start < 2 * 1000) {
WebElement saveconfirm = driver.findElement(By.id("ymPrompt_btn_0"));
saveconfirm.click();
sleep("500");
}
} catch (Exception e) {
log.debug("没有确定弹窗");
}
}
/**
* 获取属性值
*
* @param eleName
* @param attributeName
* @return
*/
public static String getAttributeByEle(WebElement webElement, String eleName, String attributeName) {
WebElement webEle = webElement.findElement(By.name(eleName));
return webEle.getAttribute(attributeName);
}
/**
* 休眠
*
* @param millis
*/
public static void sleep(String millis) {
try {
long waittingTime1 = Long.parseLong(millis);
Thread.sleep(waittingTime1);
} catch (Exception e) {
e.printStackTrace();
}
}
public static String findWindow(WebDriver driver, String title) {// 找到属于该标题的窗口句柄
Boolean find = false;
String findhandle = null;
long start = System.currentTimeMillis();
while (!find && (System.currentTimeMillis() - start) < 10 * 1000) {
Set<String> handles = driver.getWindowHandles();
for (String handle : handles) {
if (driver.switchTo().window(handle).getTitle().equals(title)) {
findhandle = handle;
find = true;
break;
}
}
}
return findhandle;
}
private static List<WebElement> findWebElements(WebDriverWait wait, String ele) {
try{
return wait.until(ExpectedConditions.presenceOfAllElementsLocatedBy(By.name(ele)));
}catch (Exception e){
log.error("未查询到元素");
}
return null;
}
/**
*
* @return 当前打开窗口的最后一个句柄
*/
public static String getLastHandle(WebDriver driver) {
Set<String> Allhandles = driver.getWindowHandles();//获取当前打开窗口的所有句柄
List<String> lst = new ArrayList<String>(Allhandles);
return lst.get(lst.size()-1);
}
}
模拟用户浏览器操作
我把这段代码也放到工具类里了
/**
* 模拟登录某个系统
* @param driver
* @param wait
* @param username
* @param password
* @return
*/
public static synchronized boolean loginSystem(WebDriver driver, WebDriverWait wait, String username,
String password) {
log.info("即将登录，页面地址 - [{}]", systemUrl);
try {
log.info("即将登录，登录信息 - [账号: {}]", username);
sleep("1000");
WebElement usernameElement = driver.findElement(By.id("name"));
// 输入用户名
usernameElement.sendKeys(username);
WebElement passwordElement = driver.findElement(By.id("pwd"));
// 输入密码
passwordElement.sendKeys(password);
// 点击登录
log.info("点击登录");
WebElement loginElement = driver.findElement(By.id("login"));
loginElement.click();
// 等待2秒，防止目标系统反应过慢
sleep("2000");
//获取弹框值判断是否登录成功
Alert alert = driver.switchTo().alert();
String text = alert.getText();
alert.accept();
log.info("登录成功");
return true;
} catch (Exception e) {
log.error("登录异常,{}",e);
return false;
}
}
/**
* 模拟填充表单内容并保存
* @param driver
* @param param
* @return
*/
public static synchronized boolean sendGzjs(WebDriver driver, MyParam param) {
try {
log.info("开始填充表单：{}", param.toString());
WebElement ndElement = driver.findElement(By.id("nd"));
// 输入年度
ndElement.sendKeys(param.getNd());
WebElement yfElement = driver.findElement(By.id("yf"));
// 输入月份
yfElement.sendKeys(param.getYf());
WebElement contentElement = driver.findElement(By.id("content"));
// 输入内容
contentElement.sendKeys(param.getContent());
log.info("点击保存");
WebElement loginElement = driver.findElement(By.id("save"));
loginElement.click();
// 等待2秒，防止目标系统反应过慢
sleep("2000");
//获取弹框值判断是否保存成功
Alert alert = driver.switchTo().alert();
String text = alert.getText();
alert.accept();
if ("目标系统返回的错误信息!".equals(text)) {
log.info("发送失败");
return false;
}
log.info("发送成功");
return true;
} catch (Exception e) {
log.error("发送异常,{}",e);
return false;
}
}
