selenium2学习：验证脚本执行结果

被调用的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()