from selenium import webdriver
import unittest,time,traceback
from selenium.common.exceptions import TimeoutException,NoSuchElementException


class TestDemo(unittest.TestCase):
 def setUp(self):
 self.driver=webdriver.Firefox(executable_path='f:\\geckodriver')


 def test_sendqqemail(self):
 try:
  url = 'https://mail.qq.com/'
  self.driver.get(url)
  self.driver.maximize_window()
  self.driver.switch_to_frame("login_frame")
  self.driver.find_element_by_id("u").send_keys("用戶名")
  self.driver.find_element_by_id("p").send_keys("密碼")
  self.driver.find_element_by_id("login_button").click()
  time.sleep(5)
  self.driver.switch_to.default_content() #退出iframe,如果沒(méi)這句話后面的元素定位都會(huì)出錯(cuò)
  self.driver.find_element_by_id('composebtn').click()
  #寫(xiě)信模塊都在iframe=mainFrame頁(yè)面上
  self.driver.switch_to_frame("mainFrame")
  time.sleep(3)
  self.driver.find_element_by_xpath("http://*[@id='toAreaCtrl']/div[2]/input").send_keys("收件人")
  #定位主題
  self.driver.find_element_by_xpath('//input[@id="subject"]').send_keys("主題")
  #正文編寫(xiě)iframe中
  self.driver.switch_to_frame(self.driver.find_element_by_class_name("qmEditorIfrmEditArea"))
  #self.driver.find_element_by_xpath("/html/body").send_keys('你好，再見(jiàn)')
  #定位正文
  content=self.driver.find_element_by_tag_name("body")
  #必須click(),不然會(huì)寫(xiě)到主題里面
  content.click()
  content.send_keys("正文內(nèi)容")
  time.sleep(3)
  self.driver.switch_to.default_content()
  self.driver.switch_to_frame("mainFrame")
  #定位發(fā)送按鈕
  self.driver.find_element_by_xpath('//a[.="發(fā)送"]').click()
 except TimeoutException:
  print("顯示等待頁(yè)面元素超時(shí)")
 except NoSuchElementException:
  print("尋找頁(yè)面元素不存在",traceback.print_exc())
 except Exception:
  print(traceback.print_exc())

 def tearDown(self):
 self.driver.quit()


 if __name__=="__main__":
 unittest.main()