pytest在测试方法

时间:2018-05-16 14:44:13

标签: python tdd pytest

我有pytest的以下测试类:

class TestConnection(AsyncTestCase):
      '''Integration test'''

      @gen_test
      def test_connecting_to_server(self):
          '''Connecting to the TCPserver'''
          client = server = None
          try:
              sock, port = bind_unused_port()
              with NullContext():
                  server = EchoServer()
                  server.add_socket(sock)
              client = IOStream(socket.socket())

              #### HERE I WANT TO HAVE THE caplog FIXTURE

              with ExpectLog(app_log, '.*decode.*'):
                  yield client.connect(('localhost', port))
                  yield client.write(b'hello\n')
                  # yield client.read_until(b'\n')
                  yield gen.moment
                  assert False
          finally:
              if server is not None:
                  server.stop()
              if client is not None:
                  client.close()

在这个类中,显然ExpectLog没有工作,所以在pytest的文档中进行了一天的挖掘之后,我发现有一个可以在你的方法中插入的caplog fixture,以便访问捕获的日志。如果我有一个测试函数,我可以添加caplog参数,但是如何在上面的测试类的方法中使caplog fixture可用?

1 个答案:

答案 0 :(得分:5)

虽然您无法将灯具作为参数传递给unittest测试方法,但您可以将它们作为实例属性注入。例如:

# spam.py
import logging

def eggs():
    logging.getLogger().info('bacon')

测试spam.eggs()

# test_spam.py
import logging
import unittest
import pytest
import spam


class SpamTest(unittest.TestCase):

    @pytest.fixture(autouse=True)
    def inject_fixtures(self, caplog):
        self._caplog = caplog

    def test_eggs(self):
        with self._caplog.at_level(logging.INFO):
            spam.eggs()
            assert self._caplog.records[0].message == 'bacon'