java - Selenium 多浏览器测试。只有最后一个浏览器会收到 cookie。测试NG

Question

我一直在进行自动化测试，并尝试通过登录一次、将 cookie 保存在文本文件中，然后最终在新的浏览器实例上读取和添加文件来跳过登录页面。

当我一次只创建一个浏览器时，它工作得很好，但如果我并行创建多个浏览器，则只有一个页面接收 cookie(从而按预期跳过登录页面)。

这是代码:

我的设置()。这很完美。我只是登录，将 cookie 数据保存在文本文件中并关闭浏览器。

@BeforeClass
public void cookie_setup(){
    //Firefox
    System.setProperty("webdriver.gecko.driver", "C:\\SeleniumGecko/geckodriver.exe");
    webdriver = new FirefoxDriver();
File file = new File("Cookies.data");
    try
    {
        file.delete();
        file.createNewFile();
        FileWriter fileWrite = new FileWriter(file);
        BufferedWriter  Bwrite = new BufferedWriter(fileWrite);

        for ( Cookie ck: webdriver.manage().getCookies())
        {
            Bwrite.write((ck.getName()+";"+ck.getValue()+";"+ck.getDomain()+";"+ck.getPath()+";"+ck.getExpiry()+";"+ck.isSecure()));

            Bwrite.newLine();
        }
        Bwrite.close();
        fileWrite.close();
    }catch (Exception ex)
    {
        ex.printStackTrace();
    }
    cookies = webdriver.manage().getCookies();
    webdriver.quit();
}

• 然后我创建一个新的 Firefox 浏览器
• 读取文本文件中的 cookie 数据

• 通过以下方式将 cookie 添加到页面:`webdriver.manage().addCookie(ck);

  @Test(threadPoolSize = 4,  invocationCount = 4)
  public void search() throws InterruptedException {
       System.setProperty("webdriver.gecko.driver", 
                          "C:\\SeleniumGecko/geckodriver.exe");
  webdriver.navigate().to("http://pagethatrequirelogin/");
  
  try {
      File file = new File("Cookies.data");
      FileReader fileReader = new FileReader(file);
      BufferedReader Buffreader = new BufferedReader(fileReader);
      String strline;
      while ((strline = Buffreader.readLine()) != null) {
          StringTokenizer token = new StringTokenizer(strline, ";");
          while (token.hasMoreTokens()) {
              String name = token.nextToken();
              String value = token.nextToken();
              String domain = token.nextToken();
              String path = token.nextToken();
              Date expiry = null;
  
              String val;
              if (!(val = token.nextToken()).equals("null")) {
                  expiry = new Date(val);
              }
              Boolean isSecure = Boolean.valueOf(token.nextToken());
              Cookie ck = new Cookie(name, value, domain, path, expiry, isSecure);
              System.out.println(ck);
              webdriver.manage().addCookie(ck);
  
          }
      }
  }
  catch(Exception ex){
  }
  

  }

我使用 TestNG 并使用参数:

@Test(threadPoolSize = 4,  invocationCount = 4)

我同时创建了4个firefox浏览器，但只有一个浏览器收到cookie并跳过登录页面。

如果我一次只创建一个浏览器多次，效果非常好。

Answer 1

问题是因为您的 @Test 方法可能会覆盖 webdriver 实例。下面的示例展示了它的线程安全版本的外观:

import org.openqa.selenium.Cookie;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.text.DateFormat;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.StringTokenizer;

public class CookiesSample {
    private static final ThreadLocal<WebDriver> drivers = new ThreadLocal<>();
    private List<Cookie> cookies = new ArrayList<>();

    @BeforeClass
    public void beforeClass() throws IOException {
        String text = "Cookies.data";
        List<String> lines = Files.readAllLines(Paths.get(text));
        lines.forEach(line -> {
            StringTokenizer token = new StringTokenizer(line, ";");
            while (token.hasMoreTokens()) {
                String name = token.nextToken();
                String value = token.nextToken();
                String domain = token.nextToken();
                String path = token.nextToken();
                Date expiry = null;

                String val = token.nextToken();
                if (!val.equals("null")) {
                    expiry = asDate(val);
                }
                Boolean isSecure = Boolean.valueOf(token.nextToken());
                Cookie ck = new Cookie(name, value, domain, path, expiry, isSecure);
                cookies.add(ck);
            }
        });
    }

    @BeforeMethod
    public void beforeMethod() {
        WebDriver driver = new FirefoxDriver();
        cookies.forEach(driver.manage()::addCookie);
        drivers.set(driver);
    }

    @AfterMethod
    public void afterMethod() {
        drivers.get().quit();
        drivers.remove();
    }

    @Test(threadPoolSize = 4, invocationCount = 4)
    public void testMethod() {
        drivers.get().get("http://pagethatrequirelogin/");
    }

    private static Date asDate(String text) {
        try {
            return DateFormat.getInstance().parse(text);
        } catch (ParseException e) {
            return new Date();
        }
    }
}