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()