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]