python封装和调用_python+selenium-登录封装与调用

class login_go():

u'''登录类封装'''

def __init__(self,driver):

u'''初始化driver参数'''

self.driver = driver

def login(self,username,password):

u'''输入用户名和密码,点击登录'''

self.driver.find_element_by_id("liger-textbox-user").clear()

self.driver.find_element_by_id("liger-textbox-user").send_keys(username)

self.driver.find_element_by_id("liger-textbox-pwd_old").clear()

self.driver.find_element_by_id("liger-textbox-pwd").clear()

self.driver.find_element_by_id("liger-textbox-pwd").send_keys(password)

self.driver.find_element_by_id("go").click()

二、调用

from selenium import webdriver

import unittest

from loginclass import login_go

from time import sleep

class Login(unittest.TestCase):

def setUp(self):

self.driver = webdriver.Chrome()

self.driver.implicitly_wait(10)

self.driver.maximize_window()

self.driver.get(login_url)

def tearDown(self):

self.driver.quit()

def case_login(self):

login_go(self.driver).login("chen","chen")#调用login方法

sleep(3)

result1 = self.driver.find_element_by_xpath(".//*[@id='l-topmenu-r-bottm']/span[2]").text

print(result1)

result2 = "欢迎您"

self.assertIn(result2,result1,msg="失败原因：%s中没有发现%s"%(result1,result2))

sleep(2)

if __name__ == "__main__":

login_url = "http://www.abc.com"

unittest.main()