如何在 Python 中获取依赖注入对象的每个测试用例范围？

Question

我正在使用 python-inject，python 2.6（带有unittest2）。

我们有使用注入进行测试的类，以及也使用相同值的测试用例。我们目前使用inject.appscope来“单一化”这些值。否则，它们将按用户实例化。

import inject

class A(object):
    '''shared data between many parts'''
    ...

class Foo(object):
    '''uses a to do stuff'''

    a = inject.attr('a', A)

    def bar(self):
        self.a.doStuff()

class TestFoo(TestCase):
    a = inject.attr('a', A)

    def setUp(self):
        self.foo = Foo()

    def testBar(self):
        self.foo.bar()
        self.assertTrue(self.a.stuffWasDone())
        self.assertFalse(self.a.stuffWasDoneTwice())

    def testBarTwice(self):
        self.foo.bar()
        self.foo.bar()
        self.assertTrue(self.a.stuffWasDoneTwice())


injector = inject.Injector()
injector.bind(A, scope=inject.appscope)
inject.register(injector)

runTests()

理想情况下，我想在每次测试运行后重置 A，以避免交叉污染。 （目前在tearDown（）中执行类似A.reset（）的操作。）

inject.reqscope支持类似的操作（一组本地范围的实例），但我真的不知道在哪里调用注册() & unregister()（重置注入对象缓存）。在 setUp() 和 tearDown() 中为时已晚，因为 Foo.a 可能已经在 Foo.__init__() 中为每个他们。

关于如何做到这一点的任何想法，或者我应该使用不同的方法？

Answer 1

您可以使用 setUp 和 tearDown 来注册/取消注册范围，将每个测试方法调用视为一个新的“请求”。

通过以下更改，您的示例单元测试将通过：

class TestFoo(unittest2.TestCase):

    a = inject.attr('a', A)

    def __init__(self, *n, **kw):
        unittest2.TestCase.__init__(self, *n, **kw)
        self.unit_scope = inject.reqscope

    def setUp(self):
        self.foo = Foo()
        self.unit_scope.register()

    def tearDown(self):
        self.unit_scope.unregister()

    ...

并使用 inject.reqscope 绑定您的 A 实例：

injector = inject.Injector()
injector.bind(A, scope=inject.reqscope)