如何在Selenium Java+Cucumber框架中处理SSO？

一、前置准备 #

• 明确SSO认证类型：先确认系统采用的是OAuth2、SAML还是基于Session/Cookie的SSO方案，不同类型的处理逻辑差异较大
• 配置测试环境：确保测试环境的SSO服务可正常访问，提前准备好测试账号、密码，若为OAuth2还需获取客户端ID/密钥等配置信息

二、核心处理流程 #

方式1：复用已认证的Cookie/Token（高效推荐） #

这种方式跳过重复登录环节，直接复用之前的认证状态，适合频繁执行的自动化测试场景：

• 首次执行登录流程：
  1. 用Selenium启动浏览器，打开SSO登录页面
  2. 定位用户名、密码输入框，填入测试账号信息并提交登录
  3. 等待页面跳转至目标应用，确认认证完成
  4. 提取当前浏览器的认证Cookie（如JSESSIONID、SSO专属Cookie）或前端存储的Token（如localStorage中的OAuth2 Token）
  5. 将Cookie/Token保存到本地文件、缓存或Cucumber上下文变量中
• 后续测试场景：
  1. 打开目标应用的受保护页面
  2. 将保存的Cookie添加到当前Selenium会话（通过driver.manage().addCookie(cookie)），或把Token注入请求头/本地存储
  3. 刷新页面，直接进入已登录状态

Java代码示例：

// 保存认证Cookie到本地文件
public void saveAuthCookies(WebDriver driver) throws IOException {
    Set<Cookie> cookies = driver.manage().getCookies();
    ObjectMapper mapper = new ObjectMapper();
    mapper.writeValue(new File("auth_cookies.json"), cookies);
}

// 加载Cookie到当前会话
public void loadAuthCookies(WebDriver driver) throws IOException {
    ObjectMapper mapper = new ObjectMapper();
    Set<Cookie> cookies = mapper.readValue(new File("auth_cookies.json"), 
        new TypeReference<Set<Cookie>>(){});
    for (Cookie cookie : cookies) {
        driver.manage().addCookie(cookie);
    }
    driver.navigate().refresh();
}

方式2：模拟完整SSO跳转流程（适合链路验证场景） #

当需要测试完整的SSO认证链路时，模拟用户真实操作流程：

1. 在Cucumber的Given步骤中，打开目标应用的受保护页面，等待自动跳转到SSO认证中心
2. 在When步骤中，填充SSO登录表单的用户名、密码并提交
3. 在Then步骤中，验证是否成功跳转回目标应用，且页面处于已登录状态

三、Cucumber框架集成 #

封装可复用的Step Definition #

将SSO认证逻辑封装为通用步骤，便于在多个Feature场景中调用：

public class SsoSteps {
    private WebDriver driver;
    private CookieManager cookieManager;

    public SsoSteps() {
        driver = DriverFactory.getDriver();
        cookieManager = new CookieManager();
    }

    @Given("用户已通过SSO认证")
    public void userIsAuthenticatedViaSso() throws IOException {
        driver.get("https://target-app.com/dashboard");
        if (!isUserLoggedIn()) {
            if (cookieManager.hasSavedCookies()) {
                cookieManager.loadAuthCookies(driver);
            } else {
                performFullSsoLogin();
                cookieManager.saveAuthCookies(driver);
            }
        }
    }

    private void performFullSsoLogin() {
        driver.findElement(By.id("username")).sendKeys("test_user");
        driver.findElement(By.id("password")).sendKeys("test_pass");
        driver.findElement(By.id("login-btn")).click();
        // 显式等待跳转至目标应用
        new WebDriverWait(driver, Duration.ofSeconds(10))
            .until(ExpectedConditions.titleContains("Dashboard"));
    }

    private boolean isUserLoggedIn() {
        // 通过页面元素判断登录状态，比如检查用户头像是否存在
        return driver.findElements(By.id("user-avatar")).size() > 0;
    }
}

Feature场景示例 #

Feature: 目标应用核心功能测试
  Scenario: 访问受保护的仪表盘页面
    Given 用户已通过SSO认证
    When 用户点击仪表盘菜单
    Then 应显示数据统计模块

四、关键注意事项 #

• 显式等待处理跳转：使用WebDriverWait替代硬编码等待时间，等待跳转后的页面标题、目标元素出现，避免因网络延迟导致的失败
• 多环境适配：将SSO地址、账号等配置抽离到配置文件，适配测试、预发等不同环境
• 会话隔离：并行测试时，确保每个线程的Cookie/Token独立存储，避免测试用例互相干扰
• 过期处理：定期清理过期的Cookie/Token，或在检测到认证失效时自动触发重新登录

内容的提问来源于stack exchange，提问作者Techie_Taks