test_session.py 文件源码

def test_error_on_close_socket(caplog, session):
    def close_which_raises_error():
        raise ValueError('a problem occurred')

    session._sock = FakeSocket()
    session._sock.close = close_which_raises_error

    session._close_socket()

    import logging

    assert caplog.record_tuples[-1] == (
        'lomond',
        logging.WARNING,
        'error closing socket (a problem occurred)'
    )