def __init__(self, exception): if ( exception.__class__.__name__ == r.ReqlNonExistenceError("").__class__.__name__ ): error_message = "The table is empty." elif exception.__class__.__name__ == r.ReqlOpFailedError("").__class__.__name__: error_message = "The table is not initialized." elif exception.__class__.__name__ == r.ReqlDriverError("").__class__.__name__: error_message = "Could not connect to RethinkDB." else: error_message = exception super().__init__(error_message) self.message = error_message
def mock_raise(*args, **kwargs): raise rethinkdb.ReqlOpFailedError('')