如何覆写nosetest的teardown函数及解决自定义测试输出插件无效问题

嘿，我来帮你搞定这几个nose测试的问题——我之前也踩过类似的坑，给你梳理清楚解决方案：

nose提供了几种灵活的覆写方式，看你需要的范围：

• 单个测试方法的后置清理：直接在测试类里定义teardown(self)方法，nose会自动在每个测试方法执行后调用它：

class TestSuite(object):
    def teardown(self):
        # 自定义清理逻辑，比如关闭连接、删除临时文件
        print("Running custom teardown for this test method")
    
    def test1(self):
        assert_true(True)

• 整个模块的全局清理：如果要在当前模块所有测试执行完毕后运行一次清理，定义模块级的teardown()函数即可：

def teardown():
    # 模块级全局清理
    print("Running global teardown for the entire test module")

class TestSuite(object):
    def test1(self):
        assert_true(True)

• 全局所有测试的统一覆写：如果要给所有测试用例加统一的清理逻辑，写个插件实现afterTest方法：

from nose.plugins import Plugin

class GlobalTeardownPlugin(Plugin):
    def afterTest(self, test):
        # 对所有测试生效的自定义清理
        print(f"Custom teardown for test: {test}")

你想替换nose默认的./F/E输出，确实可以通过插件的addSuccess/addError/addFailure方法实现，但你之前的问题出在插件没被正确激活，导致这些方法根本没被调用。后面我会讲怎么正确注册激活插件，解决这个问题。

另外要注意：nose默认会捕获stdout输出，所以即使插件生效，print的内容也可能被吃掉——记得用-s参数关闭输出捕获，或者直接写入sys.stdout并刷新。

你之前的两次尝试都差了关键一步，给你两种可行的实现方式：

方案一：通过命令行加载并激活插件 #

1. 先给你的插件类加一个name属性（nose靠这个识别插件），并在configure方法里启用它：

# plugin1.py
import os
import sys
from nose.plugins import Plugin

class Plugin1(Plugin):
    name = 'custom-output'  # 必须指定插件名称
    
    def addSuccess(self, test):
        sys.stdout.write("Test successful\n")
        sys.stdout.flush()  # 绕过nose的输出捕获
    def addError(self, test, err):
        sys.stdout.write(f"Had error: {err}\n")
        sys.stdout.flush()
    def addFailure(self, test, err):
        sys.stdout.write(f"Had failure: {err}\n")
        sys.stdout.flush()
    
    def options(self, parser, env=os.environ):
        super(Plugin1, self).options(parser, env=env)
        # 可选：添加自定义命令行参数来开关插件
        parser.add_option("--enable-custom-output", action="store_true", default=False, help="Turn on custom test output")
    
    def configure(self, options, conf):
        super(Plugin1, self).configure(options, conf)
        # 根据命令行参数启用插件，或者直接设为True强制启用
        self.enabled = options.enable_custom_output

2. 运行测试时，指定启用插件：

nosetests --nologcapture -s test1.py --enable-custom-output

如果不想用命令行参数开关，直接把self.enabled = True写死在configure里，这样只要插件被加载就会生效。

方案二：通过代码调用nose.main加载插件 #

你的代码里的问题是插件的enabled属性默认是False，所以即使添加到addplugins里也不会生效。修改插件的configure方法强制启用：

# plugin1.py
import sys
from nose.plugins import Plugin

class Plugin1(Plugin):
    def addSuccess(self, test):
        sys.stdout.write("Test successful\n")
        sys.stdout.flush()
    def addError(self, test, err):
        sys.stdout.write(f"Had error: {err}\n")
        sys.stdout.flush()
    def addFailure(self, test, err):
        sys.stdout.write(f"Had failure: {err}\n")
        sys.stdout.flush()
    
    def options(self, parser, env=os.environ):
        super(Plugin1, self).options(parser, env=env)
    
    def configure(self, options, conf):
        super(Plugin1, self).configure(options, conf)
        self.enabled = True  # 关键：强制启用插件

然后你的test1.py保持不变，运行时记得加-s参数：

from nose.tools import assert_true
import nose
from plugin1 import Plugin1

class TestSuite(object):
    def test1(self):
        assert_true(True)

if __name__ == '__main__':
    # 运行时加-s关闭输出捕获
    nose.main(addplugins=[Plugin1()], argv=['', '-s', 'test1.py'])

现在运行python test1.py就能看到插件的自定义输出了！

内容的提问来源于stack exchange，提问作者Alex