python中的Selenium webdriver:跨测试用例重用相同的Web浏览器

Question

Python newb在这里.我试图在我的测试用例中重复使用相同的浏览器.但是,我无法弄清楚如何传递全局变量来使其工作.

目前,我有一个看起来像这样的main.py#!C:/Python27/python.exe

import unittest
import unittest, time, re, HTMLTestRunner, cgi
import os, sys, inspect

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException

global DRIVER
DRIVER  = webdriver.Firefox()

# Make all subfolders available for importing
cmd_folder = os.path.realpath(os.path.abspath(os.path.split(inspect.getfile( inspect.currentframe() ))[0]))
if cmd_folder not in sys.path:
    sys.path.insert(0, cmd_folder)

# Import test cases
from setup.testcaseA import *
from setup.testcaseB import *

# serialize the testcases (grouping testcases)
suite = unittest.TestSuite() # setup new test suite
suite.addTest(unittest.makeSuite(testcaseA))
suite.addTest(unittest.makeSuite(testcaseB))

runner = HTMLTestRunner.HTMLTestRunner()
print "Content-Type: text/html\n" # header is required for displaying the website
runner.run(suite)

我在setup /文件夹中有testcaseA.py文件,如下所示:

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
import unittest, time, re, cgi

class testcaseA(unittest.TestCase):

    def setUp(self):
        #get global driver variable <- DOESNT WORK!
        self.driver = DRIVER            

    def testcaseA(self):
        driver = self.driver
        #Bunch of tests

    def tearDown(self):
        #self.driver.quit() <- Commented out, because I want to continue re-using the browser

testcaseB.py与testcaseA.py基本相同

当我运行main.py时,我收到一个错误:ft1.1:Traceback(最近一次调用最后一次):文件"C:\ test\setup\testcaseA.py",第10行,在setUp self.driver = DRIVER #get全局驱动程序变量NameError:未定义全局名称"DRIVER"

有什么建议？

谢谢!

Answer 1

您可以尝试创建另一个模块(我通常使用pkg .__ init__来处理此类事情)并在其中放置一个返回selenium驱动程序的函数.当然,如果已经存在,则返回缓存的缓存.例如.在mypkg/__ init__.py中

from selenium import webdriver

DRIVER = None

def getOrCreateWebdriver():
    global DRIVER
    DRIVER = DRIVER or webdriver.Firefox()
    return DRIVER

并通过以下方式调用您的测试:

import mypkg
...
class testcaseA(unittest.TestCase):

    def setUp(self):
        self.driver = mypkg.getOrCreateWebdriver()