fix tests

Author: daniel-sanche

tests/asyncio/retry/test_retry_streaming_async.py

@@ -36,7 +36,7 @@ async def test_retry_streaming_target_bad_sleep_generator():
     from google.api_core.retry.retry_streaming_async import retry_target_stream
 
     with pytest.raises(ValueError, match="Sleep generator"):
-        await retry_target_stream(None, None, [], None).__anext__()
+        await retry_target_stream(None, lambda x: True, [], None).__anext__()
 
 
 class TestAsyncStreamingRetry(Test_BaseRetry):

tests/asyncio/retry/test_retry_unary_async.py

@@ -137,7 +137,7 @@ async def test_retry_target_timeout_exceeded(monotonic, sleep, use_deadline_arg)
 async def test_retry_target_bad_sleep_generator():
     with pytest.raises(ValueError, match="Sleep generator"):
         await retry_async.retry_target(
-            mock.sentinel.target, mock.sentinel.predicate, [], None
+            mock.sentinel.target, lambda x: True, [], None
         )
 
 

tests/unit/retry/test_retry_streaming.py

@@ -33,7 +33,7 @@ def test_retry_streaming_target_bad_sleep_generator():
     with pytest.raises(
         ValueError, match="Sleep generator stopped yielding sleep values"
     ):
-        next(retry_streaming.retry_target_stream(None, None, [], None))
+        next(retry_streaming.retry_target_stream(None, lambda x: True, [], None))
 
 
 class TestStreamingRetry(Test_BaseRetry):

tests/unit/retry/test_retry_unary.py

@@ -146,7 +146,7 @@ def test_retry_target_timeout_exceeded(monotonic, sleep, use_deadline_arg):
 
 def test_retry_target_bad_sleep_generator():
     with pytest.raises(ValueError, match="Sleep generator"):
-        retry.retry_target(mock.sentinel.target, mock.sentinel.predicate, [], None)
+        retry.retry_target(mock.sentinel.target, lambda x: True, [], None)
 
 
 class TestRetry(Test_BaseRetry):