1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
from datetime import datetime, timedelta, timezone
from unittest.mock import patch
import pytest
from requests_cache.policy.expiration import (
EXPIRE_IMMEDIATELY,
get_expiration_datetime,
get_url_expiration,
)
from tests.conftest import HTTPDATE_DATETIME, HTTPDATE_STR
@patch('requests_cache.expiration.datetime')
def test_get_expiration_datetime__no_expiration(mock_datetime):
assert get_expiration_datetime(None) is None
assert get_expiration_datetime(-1) is None
assert get_expiration_datetime(EXPIRE_IMMEDIATELY) == mock_datetime.utcnow()
@pytest.mark.parametrize(
'expire_after, expected_expiration_delta',
[
(timedelta(seconds=60), timedelta(seconds=60)),
(60, timedelta(seconds=60)),
(33.3, timedelta(seconds=33.3)),
],
)
def test_get_expiration_datetime__relative(expire_after, expected_expiration_delta):
expires = get_expiration_datetime(expire_after)
expected_expiration = datetime.utcnow() + expected_expiration_delta
# Instead of mocking datetime (which adds some complications), check for approximate value
assert abs((expires - expected_expiration).total_seconds()) <= 1
def test_get_expiration_datetime__tzinfo():
tz = timezone(-timedelta(hours=5))
dt = datetime(2021, 2, 1, 7, 0, tzinfo=tz)
assert get_expiration_datetime(dt) == datetime(2021, 2, 1, 12, 0)
def test_get_expiration_datetime__httpdate():
assert get_expiration_datetime(HTTPDATE_STR) == HTTPDATE_DATETIME
assert get_expiration_datetime('P12Y34M56DT78H90M12.345S', ignore_invalid_httpdate=True) is None
with pytest.raises(ValueError):
get_expiration_datetime('P12Y34M56DT78H90M12.345S')
@pytest.mark.parametrize(
'url, expected_expire_after',
[
('img.site_1.com', 60 * 60),
('http://img.site_1.com/base/img.jpg', 60 * 60),
('https://img.site_2.com/base/img.jpg', None),
('site_2.com/resource_1', 60 * 60 * 2),
('http://site_2.com/resource_1/index.html', 60 * 60 * 2),
('http://site_2.com/resource_2/', 60 * 60 * 24),
('http://site_2.com/static/', -1),
('http://site_2.com/static/img.jpg', -1),
('site_2.com', None),
('some_other_site.com', None),
(None, None),
],
)
def test_get_url_expiration(url, expected_expire_after, mock_session):
urls_expire_after = {
'*.site_1.com': 60 * 60,
'site_2.com/resource_1': 60 * 60 * 2,
'site_2.com/resource_2': 60 * 60 * 24,
'site_2.com/static': -1,
}
assert get_url_expiration(url, urls_expire_after) == expected_expire_after
@pytest.mark.parametrize(
'url, expected_expire_after',
[
('https://img.site_1.com/image.jpeg', 60 * 60),
('https://img.site_1.com/resource/1', 60 * 60 * 2),
('https://site_2.com', 1),
('https://any_other_site.com', 1),
],
)
def test_get_url_expiration__evaluation_order(url, expected_expire_after):
"""If there are multiple matches, the first match should be used in the order defined"""
urls_expire_after = {
'*.site_1.com/resource': 60 * 60 * 2,
'*.site_1.com': 60 * 60,
'*': 1,
}
assert get_url_expiration(url, urls_expire_after) == expected_expire_after
|
