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('')