eladkal commented on code in PR #62919:
URL: https://github.com/apache/airflow/pull/62919#discussion_r2891185215


##########
providers/fab/tests/unit/fab/auth_manager/test_fab_auth_manager.py:
##########
@@ -1021,6 +1021,26 @@ def 
test_db_error_propagates_when_session_remove_raises(self, auth_manager_with_
             with pytest.raises(OperationalError):
                 auth_manager_with_appbuilder.deserialize_user({"sub": "99997"})
 
+        assert mock_session.remove.call_count == 2
+
+    def test_db_error_retries_once_and_recovers(self, 
auth_manager_with_appbuilder):
+        """A transient DB disconnect is recovered by removing session and 
retrying once."""
+        user = Mock()
+        user.id = 99996
+        original_exc = OperationalError("connection dropped", None, 
Exception())
+        first_result = Mock()

Review Comment:
   `first_result` is not being used anywhere



-- 
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