# coding=utf-8
'''
Created on 2016-8-13
@author: Jennifer
Project:基礎(chǔ)類BasePage，封裝所有頁面都公用的方法，
定義open函數(shù)，重定義find_element，switch_frame，send_keys等函數(shù)。
在初始化方法中定義驅(qū)動driver，基本url，title
WebDriverWait提供了顯式等待方式。
'''
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

class BasePage(object):
  """
  BasePage封裝所有頁面都公用的方法，例如driver, url ,FindElement等
  """
  #初始化driver、url、pagetitle等
  #實例化BasePage類時，最先執(zhí)行的就是__init__方法，該方法的入?yún)?，其實就是BasePage類的入?yún)ⅰ?
  #__init__方法不能有返回值，只能返回None
  #self只實例本身，相較于類Page而言。
  def __init__(self, selenium_driver, base_url, pagetitle):
    self.driver = selenium_driver
    self.base_url = base_url
    self.pagetitle = pagetitle
     
  #通過title斷言進入的頁面是否正確。
  #使用title獲取當前窗口title，檢查輸入的title是否在當前title中，返回比較結(jié)果（True 或 False）
  def on_page(self, pagetitle):
    return pagetitle in self.driver.title
  
  #打開頁面，并校驗頁面鏈接是否加載正確
  #以單下劃線_開頭的方法，在使用import *時，該方法不會被導(dǎo)入，保證該方法為類私有的。
  def _open(self, url, pagetitle):
    #使用get打開訪問鏈接地址
    self.driver.get(url)
    self.driver.maximize_window()
    #使用assert進行校驗，打開的窗口title是否與配置的title一致。調(diào)用on_page()方法
    assert self.on_page(pagetitle), u"打開開頁面失敗 %s"%url
  
  #定義open方法，調(diào)用_open()進行打開鏈接
  def open(self):
    self._open(self.base_url, self.pagetitle)
  
  #重寫元素定位方法
  def find_element(self,*loc):
#    return self.driver.find_element(*loc)
    try:
      #確保元素是可見的。
      #注意：以下入?yún)樵M的元素，需要加*。Python存在這種特性，就是將入?yún)⒎旁谠M里。
#      WebDriverWait(self.driver,10).until(lambda driver: driver.find_element(*loc).is_displayed())
      #注意：以下入?yún)⒈旧硎窃M，不需要加*
      WebDriverWait(self.driver,10).until(EC.visibility_of_element_located(loc))
      return self.driver.find_element(*loc)
    except:
      print u"%s 頁面中未能找到 %s 元素"%(self, loc)
  
  #重寫switch_frame方法
  def switch_frame(self, loc):
    return self.driver.switch_to_frame(loc)
  
  #定義script方法，用于執(zhí)行js腳本，范圍執(zhí)行結(jié)果
  def script(self, src):
    self.driver.execute_script(src)
  
  #重寫定義send_keys方法
  def send_keys(self, loc, vaule, clear_first=True, click_first=True):
    try:
      loc = getattr(self,"_%s"% loc) #getattr相當于實現(xiàn)self.loc
      if click_first:
        self.find_element(*loc).click()
      if clear_first:
        self.find_element(*loc).clear()
        self.find_element(*loc).send_keys(vaule)
    except AttributeError:
      print u"%s 頁面中未能找到 %s 元素"%(self, loc)

# coding=utf-8
'''
Created on 2016-8-13
@author: Jennifer
Project:頁面基本操作方法：如open，input_username，input_password，click_submit
'''
from selenium.webdriver.common.by import By
from test_8_3_2_BasePage import BasePage

#繼承BasePage類
class LoginPage(BasePage):
  #定位器，通過元素屬性定位元素對象
  username_loc =(By.NAME,'email')
  password_loc =(By.NAME,'password')
  submit_loc =(By.ID,'dologin')
  span_loc =(By.CSS_SELECTOR,"div.error-tt>p")
  dynpw_loc =(By.ID,"lbDynPw")
  userid_loc =(By.ID,"spnUid")
  
  #操作
  #通過繼承覆蓋（Overriding）方法：如果子類和父類的方法名相同，優(yōu)先用子類自己的方法。
  #打開網(wǎng)頁
  def open(self):
  #調(diào)用page中的_open打開連接
    self._open(self.base_url, self.pagetitle)
  
  #輸入用戶名：調(diào)用send_keys對象，輸入用戶名
  def input_username(self, username):
#    self.find_element(*self.username_loc).clear()
    self.find_element(*self.username_loc).send_keys(username)
  
  #輸入密碼：調(diào)用send_keys對象，輸入密碼
  def input_password(self, password):
#    self.find_element(*self.password_loc).clear()
    self.find_element(*self.password_loc).send_keys(password)
    
  #點擊登錄：調(diào)用send_keys對象，點擊登錄
  def click_submit(self):
    self.find_element(*self.submit_loc).click()
  
  #用戶名或密碼不合理是Tip框內(nèi)容展示
  def show_span(self):
    return self.find_element(*self.span_loc).text
  
  #切換登錄模式為動態(tài)密碼登錄（IE下有效）
  def swich_DynPw(self):
    self.find_element(*self.dynpw_loc).click()
  
  #登錄成功頁面中的用戶ID查找
  def show_userid(self):
    return self.find_element(*self.userid_loc).text

# coding=utf-8
'''
Created on 2016-8-13
@author: Jennifer
Project:使用unittest框架編寫測試用例。
'''
import unittest 
from test_8_3_2_LoginPage import LoginPage
from selenium import webdriver

class Caselogin126mail(unittest.TestCase):
  """
     登錄126郵箱的case
  """
  def setUp(self):
    self.driver = webdriver.Firefox()
    self.driver.implicitly_wait(30)
    self.url ="http://mail.126.com"
    self.username ="XXX"
    self.password ="XXX"
  
  #用例執(zhí)行體
  def test_login_mail(self):
    #聲明LoginPage類對象
    login_page = LoginPage(self.driver, self.url, u"網(wǎng)易")
    #調(diào)用打開頁面組件
    login_page.open()
    #切換到登錄框Frame
    login_page.switch_frame('x-URS-iframe')
    #調(diào)用用戶名輸入組件
    login_page.input_username(self.username)  
    #調(diào)用密碼輸入組件
    login_page.input_password(self.password)    
    #調(diào)用點擊登錄按鈕組件
    login_page.click_submit()

  def tearDown(self):
    self.driver.quit()
    
if __name__ == "__main__":
  unittest.main()