Пример #1
0
 def test_connection_errors(self):
     conn = Mock(name='connection')
     self.c.connection = conn
     conn.connection_errors = (KeyError,)
     assert self.c.connection_errors == conn.connection_errors
     conn.channel_errors = (ValueError,)
     assert self.c.channel_errors == conn.channel_errors
Пример #2
0
def test_ignore_errors():
    connection = Mock()
    connection.channel_errors = (KeyError,)
    connection.connection_errors = (KeyError,)

    with ignore_errors(connection):
        raise KeyError()

    def raising():
        raise KeyError()

    ignore_errors(connection, raising)

    connection.channel_errors = connection.connection_errors = ()

    with pytest.raises(KeyError):
        with ignore_errors(connection):
            raise KeyError()