被调用的login.py
#0508001:类:qq邮箱登录&退出 #coding=utf-8 from selenium import webdriver from time import sleep class Login(): #登录 def user_login(self,driver,username,password): driver.switch_to.frame('login_frame') driver.find_element_by_id("u").clear() driver.find_element_by_id("u").send_keys(username) driver.find_element_by_id("p").clear() driver.find_element_by_id("p").send_keys(password) driver.find_element_by_id("login_button").click() sleep(1) #退出 def user_logout(self,driver): driver.find_element_by_link_text("[退出]").click() sleep(1) driver.quit()
执行脚本QQmail-login.py:
#0517003:QQmail_login:assert from selenium import webdriver from time import sleep from login import * class LoginTest(): #初始化函数:__init__(self) def __init__(self): self.driver = webdriver.Chrome() sleep(1) url = "https://mail.qq.com/" self.driver.get(url) def test_admin_login(self): username = '492254' password = '000128' Login().user_login(self.driver,username,password) sleep(2) try: mail = self.driver.find_element_by_id('useraddr').text assert('492254'in mail), '邮箱登录失败了!——邮箱错误' except AssertionError as msg2: print (msg2) else: print ('登录成功啦:邮箱正确~') try: user = self.driver.find_element_by_id('useralias').text assert(user =='yayaya'), '邮箱登录失败了!——邮箱名错误' except AssertionError as msg1: print (msg1) else: print ('登录成功啦:邮箱名正确~') self.driver.quit() LoginTest().test_admin_login()