From 214eb1349dbb473b495fd1481593190531d4f401 Mon Sep 17 00:00:00 2001
From: Christian Hopps <chopps@labn.net>
Date: Mon, 9 Sep 2024 09:01:09 -0400
Subject: [PATCH] tests: add retry tests

Signed-off-by: Christian Hopps <chopps@labn.net>
---
 tests/testing/test_testing.py | 56 ++++++++++++++++++++++++++++++++++-
 1 file changed, 55 insertions(+), 1 deletion(-)

diff --git a/tests/testing/test_testing.py b/tests/testing/test_testing.py
index ec286c3..506caf2 100644
--- a/tests/testing/test_testing.py
+++ b/tests/testing/test_testing.py
@@ -6,10 +6,13 @@
 # Copyright (c) 2022, LabN Consulting, L.L.C.
 #
 "Test the testing fucntionality that has been imported into conftest.py"
+import pytest
+
+from munet.testing.util import retry
 
 
 def test_import_util():
-    from munet.testing.util import pause_test # pylint: disable=C0415,W0611
+    from munet.testing.util import pause_test  # pylint: disable=C0415,W0611
 
 
 def test_addopts(pytestconfig):
@@ -17,6 +20,57 @@ def test_addopts(pytestconfig):
     assert hasattr(pytestconfig.option, "pause")
     assert hasattr(pytestconfig.option, "pause_on_error")
 
+
+# Need to check the log for this test.
 def test_stepfunction(stepf):
     stepf("the first step")
     stepf("the second step")
+
+
+@retry(retry_timeout=2, retry_sleep=0.1)
+def retrying_test_assert():
+    if not hasattr(retrying_test_assert, "count"):
+        retrying_test_assert.count = 0
+    else:
+        retrying_test_assert.count += 1
+    assert retrying_test_assert.count == 2, "count not 2"
+    retrying_test_assert.count = 0
+
+
+@retry(retry_timeout=2, retry_sleep=0.1)
+def retrying_test_string():
+    if not hasattr(retrying_test_string, "count"):
+        retrying_test_string.count = 0
+    else:
+        retrying_test_string.count += 1
+    if retrying_test_string.count != 2:
+        return "count not 2"
+    return None
+
+
+def test_retry_string(caplog):
+    retrying_test_string()
+    assert caplog.text.count("Sleeping") == 2
+
+
+def test_retry_fail(caplog):
+    @retry(retry_timeout=1, retry_sleep=0.1)
+    def retrying_fail():
+        return "Fail"
+
+    retrying_fail(expected=False)
+    assert caplog.text.count("Sleeping") == 0
+
+
+def test_retry_assert_fail(caplog):
+    @retry(retry_timeout=1, retry_sleep=0.2)
+    def retrying_assert():
+        assert False, "Fail"
+
+    try:
+        # Expected does not consider assert an expected failure
+        retrying_assert(expected=False)
+    except AssertionError:
+        assert caplog.text.count("Sleeping") == 5
+    else:
+        assert False, "Failed b/c succeeded"