训练第一天作业2-Diana

import time

from selenium import webdriver
from selenium.webdriver.common.by import By

class TestLagou():
def setup_method(self, method):
self.driver = webdriver.Chrome()
self.vars = {}

def teardown_menthod(self, method):
    self.driver.quit()

def test_lagou(self):
    #打开拉钩网页
    self.driver.get("https://www.lagou.com/")
    #设置窗口大小   
    self.driver.set_window_size(1546,922)
    time.sleep(2)

    #叉掉弹窗
    self.driver.find_element(By.ID, "cboxClose").click()
    time.sleep(2)
    self.driver.find_element(By.ID, "search_input").send_keys("字节跳动")
    time.sleep(2)

    self.driver.find_element(By.ID, "search_button").click()
    time.sleep(2)
    res = self.driver.find_element(By.CSS_SELECTOR, '.company-car_18XKp h3')
    print(res.text)
    assert "字节跳动" == res.text