python+selenium自动化登录测试，设计不同场景进行登录，两种方式断言，截图保存

#   coding : utf-8
#   date :2019/1/7
#   根据不同场景做自动化登录测试
#   正确账号密码、正确账户错误密码、等其他场景


from selenium import webdriver
import unittest
import time

#current_time = time.strftime("%Y-%m-%d-%H_%M_%S", time.localtime(time.time()))
#pic_path = ("E:\\test\\result01\\image\\xxxxxx.png")

class TestLogin(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()

    def login(self,username,passwd):
        self.driver.get('https://******.com/index')
        self.driver.find_element_by_class_name("normal-link").click()
        self.driver.find_element_by_name("mobile").send_keys(username)
        self.driver.find_element_by_name("password").send_keys(passwd)
        self.driver.find_element_by_class_name("u-passport-submit-btn").click()

    def test_succeed_login(self):
        self.login("183******44","a1234567****")
        time.sleep(3)

        #断言方法1：
        try:
            text = self.driver.find_element_by_xpath("/html/body/div[1]/div/ul/li[2]/a").text
            print(text)
            Text = str("183******44")
            #assertEqual断言（A=B，否则Fail！！！）
            self.assertEqual(text,Text)
            self.driver.save_screenshot("E:\\test\\result01\\image\\succeed_login.png")
        except Exception :
            print("Fail!!!")
            time.sleep(3)
            self.driver.save_screenshot("E:\\test\\result01\\image\\fail_login.png")

        #断言方法2：
        #text = self.driver.find_element_by_xpath("/html/body/div[1]/div/ul/li[1]").text
        #print(text)
        #Text = str("您好！欢迎******！")
        #self.assertEqual(text,Text,msg="Fail")


    def test_passwd_fail_login(self):
        self.login("183*****33","a12345678")
        time.sleep(3)
        try:
            register = self.driver.find_element_by_xpath("/html/body/div[1]/div/ul/li[3]/a").text
            print(register)
            register01 = "***注册"
            self.assertEqual(register,register01)
            self.driver.save_screenshot("E:\\test\\result01\\image\\fail_passwd_login.png")
        except Exception:
            print("Fail")
            #截图
            self.driver.save_screenshot("E:\\test\\result01\\image\\error_login.png")


    def test_mobile_null_login(self):
        self.login('','a1234567****')
        time.sleep(3)
        self.driver.save_screenshot("E:\\test\\result01\\image\\mobile_null_login.png")

    def test_passwd_null_login(self):
        self.login("183******11",'')
        time.sleep(3)
        self.driver.save_screenshot("E:\\test\\result01\\image\\passwd_null_login.png")


    def tearDown(self):
        time.sleep(3)
        self.driver.quit()



if __name__ == "__main__":
    unittest.main