Python webdriver：重用 setUp 和 tearDown

Question

你好，你能给我一个在另一个 class 中重用 setUp 和 tearDown 的例子吗

base.py

from selenium import webdriver
class Login(object):
  def setUp(self):
  # create a new Firefox session
   self.driver = webdriver.Firefox()
   self.driver.get("google.com")
   self.driver.implicitly_wait(30)
   self.driver.maximize_window()

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

login.py

from selenium import webdriver
import base
import unittest

class LoginPage(unittest.TestCase):
 ...how to call setUp() and tearDown() from base.py...
    def test_logn(self):
        self.driver.find_element_by_id('username')
     ...etc...

你能提供示例运行它在 pc 上，看看它如何 works.Thank 你

Answer 1

使LoginPage成为base.Login的子类，使其继承Login.setUp和Login.tearDown:

base.py

import unittest
from selenium import webdriver

class Login(unittest.TestCase):

    def setUp(self):
        # create a new Firefox session
        self.driver = webdriver.Firefox()
        self.driver.get("http://google.com")
        self.driver.implicitly_wait(30)
        self.driver.maximize_window()

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

login.py

from selenium import webdriver
import base
import unittest

class LoginPage(base.Login):

    def test_logn(self):
        elt = self.driver.find_element_by_name('q')
        elt.send_keys('python')
        elt.submit()
        for result in self.driver.find_elements_by_class_name('g'):
            print(result.text)
            print('-' * 80)

if __name__ == '__main__':
    unittest.main()

Python webdriver：重用 setUp 和 tearDown

Python webdriver: reusing setUp and tearDown

python

oop

automation

webdriver