如何在unittest中使用命令行参数

Question

由于某些原因，我在单元测试中使用命令行参数时遇到问题。

简单地说，我只想在所有测试中使用env。我在这里做错了什么？

# -*- coding: utf-8 -*-
import argparse
from selenium import webdriver
from selenium.webdriver import ActionChains
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.keys import Keys
import unittest, time, re
class mySmokeTest(unittest.TestCase):
    def __init__(self, args):
        self.env = args.env
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.base_url = "http://google.com"
        self.verificationErrors = []
        self.accept_next_alert = True
    def test_add_to_cart(self, env):
        # open base page
        driver = self.driver                    
        driver.get("{0}".format(env))
    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)
if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('env', default='environment to test against')
    args = parser.parse_args()
    env = args.env
    runner = unittest.TextTestRunner()
    itersuite = unittest.TestLoader().loadTestsFromTestCase(mySmokeTest)
    runner.run(itersuite)

错误：

self.env = args.env
AttributeError: 'str' object has no attribute 'env'

Answer 1

我真的不知道为什么。。。但将主要内容更改为此项工作：

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('env', default='url to test against')
    args = parser.parse_args()
    env=args.env
    mySmokeTest.env = env
    runner = unittest.TextTestRunner()
    itersuite = unittest.TestLoader().loadTestsFromTestCase(mySmokeTest)
    runner.run(itersuite)

注意：

    mySmokeTest.env = args.env does not work