From 5eeb96fa96d6352e6548bd4388cd8eb286eee62a Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Wed, 17 Apr 2024 20:57:18 +0200
Subject: [PATCH] Add test for measure_time

---
 tests/utils/test_measure_time.py | 34 ++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 tests/utils/test_measure_time.py

diff --git a/tests/utils/test_measure_time.py b/tests/utils/test_measure_time.py
new file mode 100644
index 000000000..dac509907
--- /dev/null
+++ b/tests/utils/test_measure_time.py
@@ -0,0 +1,34 @@
+from unittest.mock import MagicMock
+
+import time_machine
+
+from freqtrade.util import MeasureTime
+
+
+def test_measure_time():
+
+    callback = MagicMock()
+    with time_machine.travel("2021-09-01 05:00:00 +00:00", tick=False) as t:
+
+        measure = MeasureTime(callback, 5, ttl=60)
+        with measure:
+            pass
+
+        assert callback.call_count == 0
+
+        with measure:
+            t.shift(10)
+
+        assert callback.call_count == 1
+        callback.reset_mock()
+        with measure:
+            t.shift(10)
+        assert callback.call_count == 0
+
+        callback.reset_mock()
+        # Shift past the ttl
+        t.shift(45)
+
+        with measure:
+            t.shift(10)
+        assert callback.call_count == 1