64

Pythonでたくさんのコンテキストマネージャーを使用していることがわかりました。しかし、私はそれらを使用して多くのことをテストしてきました、そして私はしばしば以下を必要としています：

class MyTestCase(unittest.TestCase):
  def testFirstThing(self):
    with GetResource() as resource:
      u = UnderTest(resource)
      u.doStuff()
      self.assertEqual(u.getSomething(), 'a value')

  def testSecondThing(self):
    with GetResource() as resource:
      u = UnderTest(resource)
      u.doOtherStuff()
      self.assertEqual(u.getSomething(), 'a value')

setUp()これが多くのテストに到達すると、明らかに退屈になるので、SPOT / DRY（信頼できる唯一の情報源/繰り返してはいけない）の精神で、これらのビットをテストとtearDown()メソッドにリファクタリングしたいと思います。

しかし、それを行おうとすると、この醜さにつながります。

  def setUp(self):
    self._resource = GetSlot()
    self._resource.__enter__()

  def tearDown(self):
    self._resource.__exit__(None, None, None)

これを行うためのより良い方法があるはずです。理想的には、各テストメソッドの反復ビットなしのsetUp()/tearDown()で（各メソッドでデコレータを繰り返すことでどのように実行できるかがわかります）。

編集：アンダーテストオブジェクトは内部のものであり、GetResourceオブジェクトはサードパーティのものであると考えてください（これは変更していません）。

ここに名前を変更GetSlotしGetResourceました—これは特定の場合よりも一般的です—コンテキストマネージャーは、オブジェクトがロック状態になり、ロックアウトすることを目的とした方法です。

4

5 に答える 5

45

unittest.TestCase.run()下図のようにオーバーライドしてみてはいかがでしょうか。このアプローチでは、プライベート メソッドを呼び出したり、すべてのメソッドに対して何かを実行したりする必要はありません。これは、質問者が望んでいたことです。

from contextlib import contextmanager
import unittest

@contextmanager
def resource_manager():
    yield 'foo'

class MyTest(unittest.TestCase):

    def run(self, result=None):
        with resource_manager() as resource:
            self.resource = resource
            super(MyTest, self).run(result)

    def test(self):
        self.assertEqual('foo', self.resource)

unittest.main()

TestCaseこのアプローチでは、インスタンスをコンテキスト マネージャーで変更する場合に、インスタンスをコンテキスト マネージャーに渡すこともできTestCaseます。

于 2012-06-24T19:33:49.920 に答える
5

__enter__あなたがしたように andを呼び出すことの問題__exit__は、あなたがそうしたということではありません: それらはwithステートメントの外で呼び出すことができます。__exit__問題は、例外が発生した場合にオブジェクトのメソッドを適切に呼び出すための準備がコードにないことです。

したがって、それを行う方法は、元のメソッドへの呼び出しをwithステートメントでラップするデコレータを用意することです。短いメタクラスは、デコレータをクラス内の test* という名前のすべてのメソッドに透過的に適用できます -

# -*- coding: utf-8 -*-

from functools import wraps

import unittest

def setup_context(method):
    # the 'wraps' decorator preserves the original function name
    # otherwise unittest would not call it, as its name
    # would not start with 'test'
    @wraps(method)
    def test_wrapper(self, *args, **kw):
        with GetSlot() as slot:
            self._slot = slot
            result = method(self, *args, **kw)
            delattr(self, "_slot")
        return result
    return test_wrapper

class MetaContext(type):
    def __new__(mcs, name, bases, dct):
        for key, value in dct.items():
            if key.startswith("test"):
                dct[key] = setup_context(value)
        return type.__new__(mcs, name, bases, dct)


class GetSlot(object):
    def __enter__(self): 
        return self
    def __exit__(self, *args, **kw):
        print "exiting object"
    def doStuff(self):
        print "doing stuff"
    def doOtherStuff(self):
        raise ValueError

    def getSomething(self):
        return "a value"

def UnderTest(*args):
    return args[0]

class MyTestCase(unittest.TestCase):
  __metaclass__ = MetaContext

  def testFirstThing(self):
      u = UnderTest(self._slot)
      u.doStuff()
      self.assertEqual(u.getSomething(), 'a value')

  def testSecondThing(self):
      u = UnderTest(self._slot)
      u.doOtherStuff()
      self.assertEqual(u.getSomething(), 'a value')

unittest.main()

（この回答で提案しているデコレータとメタクラスを自分でテストできるように、「GetSlot」とメソッドと関数のモック実装も例に含めました）

于 2011-12-16T02:33:32.630 に答える
2

コンテキスト マネージャーのテストをスロット クラスのテストから分離する必要があると思います。スロットの初期化/ファイナライズ インターフェイスをシミュレートするモック オブジェクトを使用して、コンテキスト マネージャー オブジェクトをテストしてから、スロット オブジェクトを個別にテストすることもできます。

from unittest import TestCase, main

class MockSlot(object):
    initialized = False
    ok_called = False
    error_called = False

    def initialize(self):
        self.initialized = True

    def finalize_ok(self):
        self.ok_called = True

    def finalize_error(self):
        self.error_called = True

class GetSlot(object):
    def __init__(self, slot_factory=MockSlot):
        self.slot_factory = slot_factory

    def __enter__(self):
        s = self.s = self.slot_factory()
        s.initialize()
        return s

    def __exit__(self, type, value, traceback):
        if type is None:
            self.s.finalize_ok()
        else:
            self.s.finalize_error()


class TestContextManager(TestCase):
    def test_getslot_calls_initialize(self):
        g = GetSlot()
        with g as slot:
            pass
        self.assertTrue(g.s.initialized)

    def test_getslot_calls_finalize_ok_if_operation_successful(self):
        g = GetSlot()
        with g as slot:
            pass
        self.assertTrue(g.s.ok_called)

    def test_getslot_calls_finalize_error_if_operation_unsuccessful(self):
        g = GetSlot()
        try:
            with g as slot:
                raise ValueError
        except:
            pass

        self.assertTrue(g.s.error_called)

if __name__ == "__main__":
    main()

これにより、コードが単純になり、懸念事項の混合が防止され、多くの場所でコーディングすることなくコンテキスト マネージャーを再利用できます。

于 2012-02-09T14:57:56.000 に答える