unitest自动化测试

#!/usr/bin/env python# -- coding: utf-8 --# @Author : Ailie# @File : mailLogin.py# @Software: PyCharmimport timeimport unittestfrom selenium import webdriverfrom selenium.webdriver.support import expected_conditions as ECclass mailLogin(unittest.TestCase):

import unittest
from selenium import webdriver
from selenium.webdriver.support import expected_conditions as EC
import time

class mailLogin(unittest.TestCase):
def setUp(self):
url = ‘https://mail.yeah.net/’
self.browser = webdriver.Firefox()
self.browser.get(url)
time.sleep(2)

def test_login_01(self):
    '''
    用户名、密码为空
    '''
    self.browser.switch_to.frame("x-URS-iframe")
    self.browser.find_element_by_name('email').send_keys('')
    self.browser.find_element_by_name('password').send_keys('')
    self.browser.find_element_by_id('dologin').click()
    self.browser.switch_to.default_content()
    time.sleep(3)
    name = self.browser.find_element_by_id('spnUid')
    if name == 'sanzang520@yeah.net':
        print('登录成功')
    else:
        print('登陆失败')

def test_login_02(self):
    '''
    用户名正确、密码为错误
    '''
    self.browser.switch_to.frame("x-URS-iframe")
    self.browser.find_element_by_name('email').send_keys('sanzang520')
    self.browser.find_element_by_name('password').send_keys('xxx')
    self.browser.find_element_by_id('dologin').click()
    self.browser.switch_to.default_content()
    time.sleep(3)
    name = self.browser.find_element_by_id('spnUid')
    if name == 'sanzang520@yeah.net':
        print('登录成功')
    else:
        print('登陆失败')

def test_login_03(self):
    '''
    用户名、密码正确
    '''
    self.browser.switch_to.frame("x-URS-iframe")
    self.browser.find_element_by_name('email').send_keys('sanzang520')
    self.browser.find_element_by_name('password').send_keys('xxx')
    self.browser.find_element_by_id('dologin').click()
    self.browser.switch_to.default_content()
    time.sleep(3)
    name = self.browser.find_element_by_id('spnUid')
    if name == 'sanzang520@yeah.net':
        print('登录成功')
    else:
        print('登陆失败')

def tearDown(self):
    self.browser.quit()

if name == “main”:
unittest.main()