Fixed flacky TokenManager test

Author: vladvildanov

tests/test_auth/test_token_manager.py

@@ -19,8 +19,8 @@ class TestTokenManager:
     @pytest.mark.parametrize(
         "exp_refresh_ratio,tokens_refreshed",
         [
-            (0.9, 2),
-            (0.28, 4),
+            (0.9, [2, 3]),
+            (0.28, [4, 5]),
         ],
         ids=[
             "Refresh ratio = 0.9,  2 tokens in 0,1 second",
@@ -39,14 +39,14 @@ def test_success_token_renewal(self, exp_refresh_ratio, tokens_refreshed):
             ),
             SimpleToken(
                 "value",
-                (datetime.now(timezone.utc).timestamp() * 1000) + 130,
-                (datetime.now(timezone.utc).timestamp() * 1000) + 30,
+                (datetime.now(timezone.utc).timestamp() * 1000) + 150,
+                (datetime.now(timezone.utc).timestamp() * 1000) + 50,
                 {"oid": "test"},
             ),
             SimpleToken(
                 "value",
-                (datetime.now(timezone.utc).timestamp() * 1000) + 160,
-                (datetime.now(timezone.utc).timestamp() * 1000) + 60,
+                (datetime.now(timezone.utc).timestamp() * 1000) + 170,
+                (datetime.now(timezone.utc).timestamp() * 1000) + 70,
                 {"oid": "test"},
             ),
             SimpleToken(
@@ -70,7 +70,7 @@ def on_next(token):
         mgr.start(mock_listener)
         sleep(0.1)
 
-        assert len(tokens) == tokens_refreshed
+        assert len(tokens) in tokens_refreshed
 
     @pytest.mark.parametrize(
         "exp_refresh_ratio,tokens_refreshed",