SNOW-80651 Fix typo in SnowflakeConnection.get_query_status_throw_if_error (snowflakedb#551)

RomeoDespres · web-flow · commit f9ecaedcc679 · 2020-12-07T11:32:17.000-08:00
diff --git a/src/snowflake/connector/connection.py b/src/snowflake/connector/connection.py
@@ -1206,7 +1206,7 @@ def get_query_status_throw_if_error(self, sf_qid: str) -> QueryStatus:
                                         'errno': int(code),
                                         'sqlstate': sql_state,
                                         'sfqid': sf_qid})
-            return status
+        return status
 
     @staticmethod
     def is_still_running(status: QueryStatus) -> bool:
diff --git a/test/integ/test_async.py b/test/integ/test_async.py
@@ -36,6 +36,8 @@ def test_async_exec(conn_cnx):
         with con.cursor() as cur:
             status = con.get_query_status(q_id)
             assert status == QueryStatus.SUCCESS
+            status = con.get_query_status_throw_if_error(q_id)
+            assert status == QueryStatus.SUCCESS
             cur.get_results_from_sfqid(q_id)
             assert len(cur.fetchall()) == 1
 
