예제 #1
0
    def test_extract_error_message_db_errr(self):
        from pyhive.exc import DatabaseError

        exception = DatabaseError({"message": "Err message"})
        result = PrestoEngineSpec._extract_error_message(exception)
        assert result == "Err message"
예제 #2
0
 def test_extract_error_message_general_exception(self):
     exception = Exception("Err message")
     result = PrestoEngineSpec._extract_error_message(exception)
     assert result == "Err message"