From 9aff9f47af6cfea7944e6f55b414247c26b04f0e Mon Sep 17 00:00:00 2001
From: Vladimir Kotal <vlada@kotalovi.cz>
Date: Tue, 14 Feb 2023 22:10:30 +0100
Subject: [PATCH] add basic back-off test

---
 .gitignore            |  3 +++
 tests/backoff_test.py | 59 +++++++++++++++++++++++++++++++++++++++++++
 tox.ini               | 11 ++++++++
 3 files changed, 73 insertions(+)
 create mode 100644 tests/backoff_test.py
 create mode 100644 tox.ini

diff --git a/.gitignore b/.gitignore
index db3d5387..342b4999 100644
--- a/.gitignore
+++ b/.gitignore
@@ -46,3 +46,6 @@ _build
 .idea
 .vscode
 *~
+
+# tox local cache
+.tox
diff --git a/tests/backoff_test.py b/tests/backoff_test.py
new file mode 100644
index 00000000..ed66dd5f
--- /dev/null
+++ b/tests/backoff_test.py
@@ -0,0 +1,59 @@
+# SPDX-FileCopyrightText: 2023 Vladimír Kotal
+#
+# SPDX-License-Identifier: Unlicense
+
+"""exponential back-off tests"""
+
+import socket
+import ssl
+import time
+from unittest import TestCase, main
+from unittest.mock import call, patch
+
+import adafruit_minimqtt.adafruit_minimqtt as MQTT
+
+
+class ExpBackOff(TestCase):
+    """basic exponential back-off test"""
+
+    connect_times = []
+
+    # pylint: disable=unused-argument
+    def fake_connect(self, arg):
+        """connect() replacement that records the call times and always raises OSError"""
+        self.connect_times.append(time.monotonic())
+        raise OSError("this connect failed")
+
+    def test_failing_connect(self) -> None:
+        """test that exponential back-off is used when connect() always raises OSError"""
+        # use RFC 1918 address to avoid dealing with IPv6 in the call list below
+        host = "172.40.0.3"
+        port = 1883
+
+        with patch.object(socket.socket, "connect") as mock_method:
+            mock_method.side_effect = self.fake_connect
+
+            connect_retries = 3
+            mqtt_client = MQTT.MQTT(
+                broker=host,
+                port=port,
+                socket_pool=socket,
+                ssl_context=ssl.create_default_context(),
+                connect_retries=connect_retries,
+            )
+            print("connecting")
+            with self.assertRaises(MQTT.MMQTTException) as context:
+                mqtt_client.connect()
+                self.assertTrue("Repeated connect failures" in str(context.exception))
+
+            mock_method.assert_called()
+            calls = [call((host, port)) for _ in range(0, connect_retries)]
+            mock_method.assert_has_calls(calls)
+
+            print(f"connect() call times: {self.connect_times}")
+            for i in range(1, connect_retries):
+                assert self.connect_times[i] >= 2**i
+
+
+if __name__ == "__main__":
+    main()
diff --git a/tox.ini b/tox.ini
new file mode 100644
index 00000000..04f4809a
--- /dev/null
+++ b/tox.ini
@@ -0,0 +1,11 @@
+# SPDX-FileCopyrightText: 2023 Vladimír Kotal
+#
+# SPDX-License-Identifier: MIT
+
+[tox]
+envlist = py39
+
+[testenv]
+changedir = {toxinidir}/tests
+deps = pytest==6.2.5
+commands = pytest