|
1 | 1 | import asyncio |
2 | 2 | from datetime import datetime, timezone |
3 | 3 | from time import sleep |
4 | | -from unittest.mock import AsyncMock, Mock |
| 4 | +from unittest.mock import Mock |
5 | 5 |
|
6 | 6 | import pytest |
7 | 7 | from redis.auth.err import RequestTokenErr, TokenRenewalErr |
@@ -29,48 +29,38 @@ class TestTokenManager: |
29 | 29 | ) |
30 | 30 | def test_success_token_renewal(self, exp_refresh_ratio): |
31 | 31 | tokens = [] |
| 32 | + errors = [] |
| 33 | + |
| 34 | + # Use a function to generate fresh tokens at request time |
| 35 | + # to avoid timing issues on slow CI runners |
| 36 | + def generate_token(): |
| 37 | + now = datetime.now(timezone.utc).timestamp() * 1000 |
| 38 | + return SimpleToken("value", now + 10000, now, {"oid": "test"}) |
| 39 | + |
32 | 40 | mock_provider = Mock(spec=IdentityProviderInterface) |
33 | | - mock_provider.request_token.side_effect = [ |
34 | | - SimpleToken( |
35 | | - "value", |
36 | | - (datetime.now(timezone.utc).timestamp() * 1000) + 100, |
37 | | - (datetime.now(timezone.utc).timestamp() * 1000), |
38 | | - {"oid": "test"}, |
39 | | - ), |
40 | | - SimpleToken( |
41 | | - "value", |
42 | | - (datetime.now(timezone.utc).timestamp() * 1000) + 150, |
43 | | - (datetime.now(timezone.utc).timestamp() * 1000) + 50, |
44 | | - {"oid": "test"}, |
45 | | - ), |
46 | | - SimpleToken( |
47 | | - "value", |
48 | | - (datetime.now(timezone.utc).timestamp() * 1000) + 170, |
49 | | - (datetime.now(timezone.utc).timestamp() * 1000) + 70, |
50 | | - {"oid": "test"}, |
51 | | - ), |
52 | | - SimpleToken( |
53 | | - "value", |
54 | | - (datetime.now(timezone.utc).timestamp() * 1000) + 190, |
55 | | - (datetime.now(timezone.utc).timestamp() * 1000) + 90, |
56 | | - {"oid": "test"}, |
57 | | - ), |
58 | | - ] |
| 41 | + mock_provider.request_token.side_effect = ( |
| 42 | + lambda *args, **kwargs: generate_token() |
| 43 | + ) |
59 | 44 |
|
60 | 45 | def on_next(token): |
61 | 46 | nonlocal tokens |
62 | 47 | tokens.append(token) |
63 | 48 |
|
| 49 | + def on_error(err): |
| 50 | + nonlocal errors |
| 51 | + errors.append(err) |
| 52 | + |
64 | 53 | mock_listener = Mock(spec=CredentialsListener) |
65 | 54 | mock_listener.on_next = on_next |
66 | | - mock_listener.on_error = AsyncMock() # Add this line |
| 55 | + mock_listener.on_error = on_error |
67 | 56 |
|
68 | 57 | retry_policy = RetryPolicy(1, 10) |
69 | 58 | config = TokenManagerConfig(exp_refresh_ratio, 0, 1000, retry_policy) |
70 | 59 | mgr = TokenManager(mock_provider, config) |
71 | 60 | mgr.start(mock_listener) |
72 | 61 | sleep(0.1) |
73 | 62 |
|
| 63 | + assert len(errors) == 0, f"Unexpected errors: {errors}" |
74 | 64 | assert len(tokens) > 0 |
75 | 65 |
|
76 | 66 | @pytest.mark.parametrize( |
|
0 commit comments