登錄測(cè)試腳本

# -*- coding: utf-8 -*-
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
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re

class CyUnitYjs(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()
        self.driver.implicitly_wait(30)
        self.base_url = "http://127.0.0.1:7070/"
        self.verificationErrors = []
        self.accept_next_alert = True
    
    def test_cy_unit_yjs(self):
        driver = self.driver
        driver.get(self.base_url + "/test/yzm")
        driver.get(self.base_url + "/test/login.jsp")
        time.sleep(2)
        driver.find_element_by_css_selector("input").clear()
        driver.find_element_by_css_selector("input").send_keys(u"劉源")
        time.sleep(2)
        driver.find_element_by_css_selector("input[type=\"password\"]").clear()
        driver.find_element_by_css_selector("input[type=\"password\"]").send_keys("123456")
        time.sleep(2)
        driver.find_element_by_css_selector("div.login-input.yzm-input > input").clear()
        driver.find_element_by_css_selector("div.login-input.yzm-input > input").send_keys("20190125")
        time.sleep(3)
        driver.get_screenshot_as_file("E:\\study-python\\project\\unit\\ly_login")
        time.sleep(3)
        driver.find_element_by_css_selector("button").click()
        time.sleep(3)
        driver.get_screenshot_as_file("E:\\study-python\\project\\unit\\ly_login_success")
        time.sleep(20)
        driver.find_element_by_css_selector(".logout")
    
    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()
                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()