1

テストで次のコンテキストマネージャーを使用しようとしています:

class APITest(TestCase):
    def create_scenario(self, user):
        # @pytest.yield_fixture
        @contextmanager
        def scenario():
            if user is not None:
                self.client.login(username=user.username, password='password')
            yield
            if user is not None:
                self.client.logout()
        return scenario

    def setUp(self):
        self.user = UserFactory.create()
        self.non_auth_scenario = self.create_scenario(None)
        self.auth_scenario = self.create_scenario(self.user)

    def test_foo_get(self):
        with self.non_auth_scenario:
            assert self.client.get('/api/foo/', format='json').status_code == 401
        with self.auth_scenario:
            assert self.client.get('/api/foo/', format='json').status_code == 200

次の結果が得られます。

    def test_widget_get(self):
\>       with self.non_auth_scenario:
E       AttributeError: __exit__

私は見ていたpytest.yield_fixtureが、機会がなかった。洞察をお願いします。

4

1 に答える 1

1

コンテキスト マネージャーを使用する場合は、次のように呼び出す必要があります。

self.non_auth_scenario():

一般的なケース:

>>> from contextlib import contextmanager
>>> @contextmanager
... def user(name):
...    print 'hello', name
...    yield
... 
>>> with user('world'):
...    pass
... 
hello world

そして、これはあなたが持っていたのと同じエラーを引き起こします:

>>> with user:
...    print 'well'
... 
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: __exit__
于 2015-11-26T16:12:16.193 に答える