[gentoo-commits] proj/portage:master commit in: lib/portage/tests/util/futures/

[Zac Medico]

commit:     98536f208194197c521675e0d0072bdc599e015a
Author:     Zac Medico <zmedico <AT> gentoo <DOT> org>
AuthorDate: Fri Nov 25 21:21:30 2022 +0000
Commit:     Zac Medico <zmedico <AT> gentoo <DOT> org>
CommitDate: Fri Nov 25 23:40:38 2022 +0000
URL:        https://gitweb.gentoo.org/proj/portage.git/commit/?id=98536f20

testOverallTimeoutWithException: handle TimeoutError

Bug: https://bugs.gentoo.org/850127
Signed-off-by: Zac Medico <zmedico <AT> gentoo.org>

 lib/portage/tests/util/futures/test_retry.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/lib/portage/tests/util/futures/test_retry.py b/lib/portage/tests/util/futures/test_retry.py
index 8ea832136..cea3e83f5 100644
--- a/lib/portage/tests/util/futures/test_retry.py
+++ b/lib/portage/tests/util/futures/test_retry.py
@@ -176,8 +176,13 @@ class RetryTestCase(TestCase):
                 asyncio.wait([decorated_func()], loop=loop)
             )
             self.assertEqual(len(done), 1)
+            cause = done.pop().exception().__cause__
             self.assertTrue(
-                isinstance(done.pop().exception().__cause__, SucceedNeverException)
+                isinstance(
+                    cause,
+                    (asyncio.TimeoutError, SucceedNeverException),
+                ),
+                msg=f"Cause was {cause.__class__.__name__}",
             )
 
     def testOverallTimeoutWithTimeoutError(self):