Added test for `get_duration_from_expiry()`

1 files changed, 17 insertions, 0 deletions

diff --git a/tests/utils/test_time.py b/tests/utils/test_time.py
index 29aca5cfe..0afffd9b1 100644
--- a/tests/utils/test_time.py
+++ b/tests/utils/test_time.py
@@ -77,3 +77,20 @@ def test_wait_until(sleep_patch):
 )
 def test_get_duration(date_from: datetime, date_to: datetime, expected: str):
     assert time.get_duration(date_from, date_to) == expected
+
+
[email protected](
+    ('expiry', 'date_from', 'expected'),
+    (
+        ('2019-12-12T00:01:00Z', datetime(2019, 12, 12, 12, 0, 5), '11 hours, 59 minutes'),
+        ('2019-12-12T00:00:00Z', datetime(2019, 12, 11, 23, 59), '1 minute'),
+        ('2019-11-23T20:09:00Z', datetime(2019, 11, 30, 20, 15), '1 week, 6 minutes'),
+        ('2019-11-23T20:09:00Z', datetime(2019, 4, 25, 20, 15), '7 months, 2 weeks'),
+        ('2019-11-23T20:58:00Z', datetime(2019, 11, 23, 21, 3), '5 minutes'),
+        ('2019-11-23T23:59:00Z', datetime(2019, 11, 24, 0, 0), '1 minute'),
+        ('2019-11-23T23:59:00Z', datetime(2022, 11, 23, 23, 0), '3 years, 3 months'),
+        ('2019-11-23T23:59:00Z', datetime(2019, 11, 23, 23, 49, 5), '9 minutes, 55 seconds'),
+    )
+)
+def test_get_duration_from_expiry(expiry: str, date_from: datetime, expected: str):
+    assert time.get_duration_from_expiry(expiry, date_from) == expected