Cookie 复用

同学们大家好，接下来我们来学习一下 cookie 复用，浏览网站时，服务器会往浏览器发一些数据，叫做 Cookie。它是一种认证数据，存储在电脑浏览器上，用于后续访问时身份验证和记录登录信息。通常，服务器会把一些关键信息如用户 ID，会话 ID，密码等存在 Cookie 里。Cookie 相当于个人信息，浏览器请求页面时携带 Cookie，这样服务器就能通过 Cookie 判断用户是否有效。通过获取和管理 Cookie，我们能模拟用户登录状态、记录用户信息，实现多个页面间的数据共享等。需要注意的是，一旦 Web 服务器把页面发给浏览器后，在连接关闭后，服务端不再保留用户的信息。

---

为什么要使用 Cookie 自动化登录

在进行自动化测试时，通常需要登录网站进行操作。复用浏览器仍然在每次用例开始都需要人为介入，只要关闭浏览器后就需要在重新人为介入。会导致测试效率将非常低下，甚至无法进行大规模的测试。使用 Cookie 自动化登录可以很好地解决这个问题。 通过 Cookie，可以在不需要手动登录的情况下访问网站，从而避免了重复输入用户名和密码的麻烦。同时，由于大部分 Cookie 的时效性都很长，只需要扫描一次 Cookie，就可以在多个测试用例中使用。

---

使用 cookie 的思路

class TestCookieLogin:
    def setup_class(self):
        self.drvier = webdriver.Chrome()

    def test_get_cookies(self):
        # 1. 访问企业微信主页/登录页面
        self.drvier.get("https://work.weixin.qq.com/wework_admin/frame#contacts")
        # 2. 等待20s，人工扫码操作
        time.sleep(20)
        # 3. 等成功登陆之后，再去获取cookie信息
        cookie = self.drvier.get_cookies()
        # 4. 将cookie存入一个可持久存储的地方，文件
        # 打开文件的时候添加写入权限
        with open("cookie.yaml", "w") as f:
            # 第一个参数是要写入的数据
            yaml.safe_dump(cookie, f)

    def test_add_cookie(self):
        # 1. 访问企业微信主页面
        self.drvier.get("https://work.weixin.qq.com/wework_admin/frame#contacts")
        # 2. 定义cookie，cookie信息从已经写入的cookie文件中获取
        cookie = yaml.safe_load(open("cookie.yaml"))
        # 3. 植入cookie
        for c in cookie:
            self.drvier.add_cookie(c)
        time.sleep(3)
        # 4.再次访问企业微信页面，发现无需扫码自动登录，而且可以多次使用
        self.drvier.get("https://work.weixin.qq.com/wework_admin/frame#contacts")

@Slf4j
public class LoginTest {
    static ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
    static WebDriver driver;

    @BeforeAll
    static void setUp(){
        ChromeOptions options = new ChromeOptions();
        //防止连接报错
        options.addArguments("--remote-allow-origins=*");
        driver = new ChromeDriver(options);
    }

    @Test
    void dumpCookie() throws  IOException {
        // 保存cookie
        ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
        String url = "https://work.weixin.qq.com/wework_admin/frame";
        // 1. 访问企业微信主页
        driver.get(url);
        // 2. 判断url是否登录成功
        WebDriverWait wait = new WebDriverWait(driver,
                Duration.ofSeconds(20),
                Duration.ofSeconds(1));
        wait.until(webDriver1 -> StringUtils.contains(webDriver1.getCurrentUrl(), "wework_admin/frame"));
        // 3. 等成功登陆之后，再去获取cookie信息
        Set<Cookie> cookies = driver.manage().getCookies();
        log.info("登录cookies:",cookies);
        mapper.writeValue(new File("cookies.yaml"), cookies);

    }

    @Test
    void loadCookie() throws IOException {
        String url = "https://work.weixin.qq.com/wework_admin/frame";
        // 访问企业微信主页
        driver.get(url);
        // 使用 cookie 登录
        // 1.声明一个泛型
        TypeReference<List<HashMap<String, Object>>> typeReference =
                new TypeReference<>() {};
        // 2. 从yaml中获取cookies
        List<HashMap<String, Object>> loadCookies =
                mapper.readValue(new File("cookies.yaml"),
                        typeReference);
        loadCookies.stream()
                // 找到domain包含work.weixin.qq.com
                .filter(cookie -> cookie.get("domain").toString().
                        contains("work.weixin.qq.com"))
                .forEach(cookie -> {
                    driver.manage().addCookie(
                            new Cookie(cookie.get("name").toString(),
                                    cookie.get("value").toString()));
                });
        // 4. 刷新页面，登录成功
        driver.navigate().refresh();
    }
}

具体实现方法包括以下几个步骤第一步打开浏览器并登录，确保登录成功后获取 Cookie 信息。第二步将获取到的 Cookie 信息存储到本地文件中，以便于下次使用。通过文件读写操作，可以将 Cookie 信息保存到一个文本文件中。第三步检查本地文件中是否已经成功获取了 Cookie 信息。第四步再次打开浏览器并植入 Cookie 信息进入主页。

---

常见问题

1. 企业微信等具有互踢机制
2. 获取cookie时为登录成功状态。
3. 复用 cookie 之后的验证问题

第一对于企业微信等具有互踢机制的系统，需要注意在获取 cookie 成功之后，不要再进行扫码登录操作，否则会导致之前获取的 cookie 失效。第二在获取 cookie 时，务必确保已经成功完成登录操作，否则获取的 cookie 将变得无效。第三获取 cookie 之后，应进入登录页面并刷新，以验证是否能够自动登录成功。若自动登陆失败，需检查获取 cookie 的代码是否存在问题或者 Cookie 是否已经国企。这一过程有助于确保所获取的 Cookie 在后续操作中能够有效使用。

---

总结

在 web 自动化测试中，掌握 Cookie 复用技巧至关重要。这不仅仅是关于复用 Cookie，更涉及到复用登录凭证，以减少繁琐的重复登录步骤，从而提高测试效率。这一技巧的核心在于有效地利用已有的登录状态，为测试过程增添便捷性和高效性。

在 web 自动化测试中，掌握 Cookie 复用技巧至关重要。这不仅仅是关于复用 Cookie，更涉及到复用登录凭证，以减少繁琐的重复登录步骤，从而提高测试效率。这一技巧的核心在于有效地利用已有的登录状态，为测试过程增添便捷性和高效性。