自动化测试QQ邮箱登录【问题及完整测试代码】

一、问题描述：
使用百度搜索QQ邮箱，点击邮箱官网进入QQ邮箱首页后，切换到iframe中报错（无法找到元素）。

二、代码描述：

from selenium import webdriver
import time
driver=webdriver.Chrome()
driver.get('https://www.baidu.com')
souSuo=driver.find_element('id', 'kw')
souSuo.click()
souSuo.clear()
souSuo.send_keys('QQ邮箱')
driver.find_element('id', 'su').click()
time.sleep(2)
driver.find_element('xpath', '//*[@id="content_left"]/div[1]/div/div[1]/h3/a[1]').click()
time.sleep(2)
# driver.get("https://mail.qq.com/")
iframe=driver.find_element("xpath", "//iframe[@id='login_frame']")
driver.switch_to.frame(iframe)

该代码执行后会报错，而将4-12行改为driver.get("https://mail.qq.com/")
代码不再报错。

原因：当使用百度搜索邮箱，再点击打开QQ邮箱首页之后，我们看到页面切换到QQ邮箱首页，而实际上后续的自动化操作依旧是在百度页面上进行的。因此直接切换到iframe会报错。

三、解决方法：
先h=driver.window_handles获取所有句柄，再通过driver.switch_to.window(h[1])切换到第二个窗口，然后再进行后续操作。

代码如下：

from selenium import webdriver
import time
driver=webdriver.Chrome()
driver.get('https://www.baidu.com')
souSuo=driver.find_element('id', 'kw')
souSuo.click()
souSuo.clear()
souSuo.send_keys('QQ邮箱')
driver.find_element('id', 'su').click()
time.sleep(2)
driver.find_element('xpath', '//*[@id="content_left"]/div[1]/div/div[1]/h3/a[1]').click()
time.sleep(2)
# driver.get("https://mail.qq.com/")
h = driver.window_handles#获取所有句柄
driver.switch_to.window(h[1])#切换到第二个窗口
time.sleep(2)
iframe=driver.find_element("xpath", "//iframe[@id='login_frame']")
driver.switch_to.frame(iframe)

四、完整测试代码

from selenium import webdriver
import time
driver=webdriver.Chrome()
driver.get('https://www.baidu.com')
souSuo=driver.find_element('id', 'kw')
souSuo.click()
souSuo.clear()
souSuo.send_keys('QQ邮箱')
driver.find_element('id', 'su').click()
time.sleep(2)
driver.find_element('xpath', '//*[@id="content_left"]/div[1]/div/div[1]/h3/a[1]').click()
time.sleep(2)
# driver.get("https://mail.qq.com/")
h = driver.window_handles#获取所有句柄
driver.switch_to.window(h[1])#切换到第二个窗口
time.sleep(2)
iframe=driver.find_element("xpath", "//iframe[@id='login_frame']")
driver.switch_to.frame(iframe)


driver.find_element('id', 'u').click()
driver.find_element('id', 'u').send_keys('user')
driver.find_element('id', 'p').click()
driver.find_element('id', 'p').send_keys('password')
driver.find_element('id', 'login_button').click()