【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 webdriver
from selenium.webdriver.common.by import By
import time
import unittest

class 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_函数，然后控制台就报错说找不到元素。所以，在哪里右键，就单独执行哪里

#栗子