# coding=utf-8
#!usr/bin/python
# 上边的这两个是必须要有的。 要不然的话运行有错误提示
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
import unittest, time, re
import HTMLTestRunner # 需要导入的内容。
class Login(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Firefox()
self.driver.implicitly_wait(30)
self.base_url = "http://mail.163.com/"
self.verificationErrors = []
self.accept_next_alert = True
def test_login(self):
driver = self.driver
driver.get(self.base_url + "/")
driver.find_element_by_id("idInput").clear()
driver.find_element_by_id("idInput").send_keys("XXXXXX2002")
driver.find_element_by_id("pwdInput").clear()
driver.find_element_by_id("pwdInput").send_keys("XXXXXX")
driver.find_element_by_id("loginBtn").click()
for i in range(60):
try:
if u"退出" == driver.find_element_by_link_text(u"退出").text: break
except: pass
time.sleep(1)
else: self.fail("time out")
driver.find_element_by_link_text(u"退出").click()
def is_element_present(self, how, what):
try: self.driver.find_element(by=how, value=what)
except NoSuchElementException, e: return False
return True
def is_alert_present(self):
try: self.driver.switch_to_alert()
except NoAlertPresentException, e: return False
return True
def close_alert_and_get_its_text(self):
try:
alert = self.driver.switch_to_alert()
alert_text = alert.text
if self.accept_next_alert:
alert.accept()
else:
alert.dismiss()
return alert_text
finally: self.accept_next_alert = True
def tearDown(self):
self.driver.quit()
self.assertEqual([], self.verificationErrors)
if __name__ == "__main__":
unittest.main()
# unittest.main()这个是必须要有的 要不testreport 中是没有内容的。
testsuite=unittest.TestSuite()
testsuite.addTest(Login("test_login"))
filename="e:\\result.html" # report所在的位置。
fp=file(filename,'wb')
runner=HTMLTestRunner.HTMLTestRunner(stream=fp,title='Result',description='Test_Report')
runner.run(testsuite)