Add caller happy path tests

hynek · Jan 29, 2024 · 23204f7 · 23204f7
1 parent 542ea48
commit 23204f7
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/tests/test_async.py b/tests/test_async.py
@@ -212,6 +212,17 @@ async def test_retry_blocks_can_be_disabled():
 
 
 class TestAsyncRetryingCaller:
+ async def test_ok(self):
+ """
+ No error, no problem.
+ """
+ arc = stamina.AsyncRetryingCaller().on(BaseException)
+
+ async def f():
+ return 42
+
+ assert 42 == await arc(f)
+
  async def test_retries(self):
  """
  Retries if the specific error is raised. Arguments are passed through.

diff --git a/tests/test_sync.py b/tests/test_sync.py
@@ -175,6 +175,17 @@ def test_never(self):
 
 
 class TestRetryingCaller:
+ def test_ok(self):
+ """
+ No error, no problem.
+ """
+ rc = stamina.RetryingCaller().on(BaseException)
+
+ def f():
+ return 42
+
+ assert 42 == rc(f)
+
  def test_retries(self):
  """
  Retries if the specific error is raised. Arguments are passed through.