【selenium2】【unittest】

#栗子 

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest ,time,re

class BaiduTest(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(30)
        self.base_url = 'http://www.baidu.com/'
        self.verificationErrors = []
        self.accept_next_alert = True

    def test_baidu(self):
        driver = self.driver
        driver.get(self.base_url,'/')
        driver.find_element(By.ID,'kw').clear()
        driver.find_element(By.ID,'kw').send_keys('selenium ide')
        driver.find_element(By.ID,'su').click()
    def is_element_present(self,how,what):
        try:
            self.driver.find_element(by=how,value=what)
        except NoSuchElementException as e:
            return False
        return True

    def is_alert_present(self):
        try:
            self.driver.switch_to_alert()
        except NoAlertPresentException as 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:
            #表示是否繼續接受下一個警告，初始化狀態爲 Ture
            self.accept_next_alert = True

    def tearDown(self):
        self.driver.quit()
        self.assertEqual([],self.verificationErrors)
if __name__ == '__main__':
    unittest.main()



#栗子 

#用於判斷素數
def is_prime(n):
    if n<=1:
        return False
    for i in range(2,n):
        if n%i == 0:
            return False
    return True


import unittest

class Test(unittest.TestCase):
    def setUp(self):
        pass

    def test_case(self):
        self.prime = is_prime(4)
        self.assertTrue(self.prime,msg='Is not prime!!')

    def tearDown(self):
        pass

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

#結果  AssertionError: False is not true : Is not prime!!

#栗子 登陸126 

from selenium import webdriverfrom selenium.webdriver.common.by import Byimport timeimport unittestclass TestLogin(unittest.TestCase):    def setUp(self):        self.driver = webdriver.Chrome()        self.driver.implicitly_wait(5)        self.base_url = 'http://mail.qq.com'        self.verificationErrors = []        self.accept_next_alert = True    def test_login(self):        driver = self.driver        driver.get(self.base_url)        #登陸        if driver.find_element(By.CSS_SELECTOR, 'iframe#login_frame').is_displayed():            driver.switch_to.default_content()            driver.switch_to.frame(driver.find_element(By.CSS_SELECTOR, 'iframe#login_frame'))            driver.find_element(By.LINK_TEXT, '賬號密碼登陸').click()            driver.find_element(By.XPATH, ".//*[@id='u']").clear()            driver.find_element(By.XPATH, ".//*[@id='u']").send_keys('981094301@qq.com')            driver.find_element(By.XPATH, ".//*[@id='p']").clear()            driver.find_element(By.XPATH, ".//*[@id='p']").send_keys('9(zndxcpvsWWXX')            driver.find_element(By.CSS_SELECTOR, 'input#login_button').click()            # 獲取斷言信息進行斷言            text = driver.find_element(By.ID, 'useraddr').text            self.assertEqual(text, '981094301@qq.com')            # 退出            time.sleep(2)            driver.find_element(By.LINK_TEXT, '退出').click()    def tearDown(self):        self.driver.quit()        self.assertEqual([],self.verificationErrors)if __name__ == '__main__':    unittest.main()【備註】必定要在主函數裏執行，我最開始右鍵的地方就是test_函數，而後控制檯就報錯說找不到元素。因此，在哪裏右鍵，就單獨執行哪裏#栗子