selenium自动化测试框架03

from selenium.webdriver.common.by import By

class LoginPage():
    def __init__(self, driver):
        self.driver = driver

    def login(self, username, passwordfield):

        email = self.driver.find_element(By.ID, "form_email")
        email.send_keys(username)

        password = self.driver.find_element(By.ID, "form_password")
        password.send_keys(passwordfield)

        loginbnt = self.driver.find_element(By.CLASS_NAME, "bn-submit")
        loginbnt.click()

from selenium import webdriver
from selenium.webdriver.common.by import By
from pages.home.login_page import LoginPage
import unittest

class LoginTests(unittest.TestCase):
     def test_vaildLogin(self):
         baseURL = "https://www.douban.com/"
         driver = webdriver.Firefox()
         driver.maximize_window()
         driver.implicitly_wait(3)
         driver.get(baseURL)

         lp = LoginPage(driver)
         lp.login("", "")

         username = driver.find_element(By.XPATH, "/html/body/div[1]/div/div[1]/ul/li[2]/a/span[1]")
         if username is not None:
             print("login successful")
             driver.quit()
         else:
             print("login failed")
             driver.quit()