arjav1528 commented on code in PR #60161:
URL: https://github.com/apache/airflow/pull/60161#discussion_r2669531085


##########
providers/snowflake/tests/unit/snowflake/hooks/test_snowflake_sql_api.py:
##########
@@ -1122,8 +1123,8 @@ def test_wait_for_query_timeout_error(self, sleep_mock, 
time_mock):
         # Simulate a query that keeps running and never finishes
         hook.get_sql_api_query_status = mock.MagicMock(return_value={"status": 
"running"})
 
-        # More side effects to ensure we hit the timeout and avoid 
StopIteration error
-        time_mock.side_effect = list(range(10))
+        # Use an infinite generator to avoid StopIteration error

Review Comment:
   I have reverted this file, you can review the final changes and merge the PR 
at your availbilility



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]

Reply via email to