xref: /unit/test/test_static_variables.py (revision 2616:ab2896c980ab)
1import os
2from pathlib import Path
3
4import pytest
5
6from unit.applications.proto import ApplicationProto
7
8client = ApplicationProto()
9
10
11@pytest.fixture(autouse=True)
12def setup_method_fixture(temp_dir):
13    os.makedirs(f'{temp_dir}/assets/dir')
14    os.makedirs(f'{temp_dir}/assets/d$r')
15    Path(f'{temp_dir}/assets/index.html').write_text(
16        '0123456789', encoding='utf-8'
17    )
18    Path(f'{temp_dir}/assets/dir/file').write_text('file', encoding='utf-8')
19    Path(f'{temp_dir}/assets/d$r/file').write_text('d$r', encoding='utf-8')
20
21    assert 'success' in client.conf(
22        {
23            "listeners": {"*:8080": {"pass": "routes"}},
24            "routes": [{"action": {"share": f'{temp_dir}/assets$uri'}}],
25        }
26    )
27
28
29def update_share(share):
30    if isinstance(share, list):
31        return client.conf(share, 'routes/0/action/share')
32
33    return client.conf(f'"{share}"', 'routes/0/action/share')
34
35
36def test_static_variables(temp_dir):
37    assert client.get(url='/index.html')['status'] == 200
38    assert client.get(url='/d$r/file')['status'] == 200
39
40    assert 'success' in update_share('$uri')
41    assert client.get(url=f'{temp_dir}/assets/index.html')['status'] == 200
42
43    assert 'success' in update_share(f'{temp_dir}/assets${{uri}}')
44    assert client.get(url='/index.html')['status'] == 200
45
46
47def test_static_variables_array(temp_dir):
48    assert 'success' in update_share([f'{temp_dir}/assets$uri', '$uri'])
49
50    assert client.get(url='/dir/file')['status'] == 200
51    assert client.get(url=f'{temp_dir}/assets/index.html')['status'] == 200
52    assert client.get(url='/blah')['status'] == 404
53
54    assert 'success' in client.conf(
55        {
56            "share": [f'{temp_dir}/assets$uri', '$uri'],
57            "fallback": {"return": 201},
58        },
59        'routes/0/action',
60    )
61
62    assert client.get(url='/dir/file')['status'] == 200
63    assert client.get(url=f'{temp_dir}/assets/index.html')['status'] == 200
64    assert client.get(url='/dir/blah')['status'] == 201
65
66
67def test_static_variables_buildin_start(temp_dir):
68    assert 'success' in update_share('$uri/assets/index.html')
69    assert client.get(url=temp_dir)['status'] == 200
70
71
72def test_static_variables_buildin_mid(temp_dir):
73    assert 'success' in update_share(f'{temp_dir}$uri/index.html')
74    assert client.get(url='/assets')['status'] == 200
75
76
77def test_static_variables_buildin_end():
78    assert client.get(url='/index.html')['status'] == 200
79
80
81def test_static_variables_invalid(temp_dir):
82    assert 'error' in update_share(f'{temp_dir}/assets/d$r$uri')
83    assert 'error' in update_share(f'{temp_dir}/assets/$$uri')
84    assert 'error' in update_share(
85        [f'{temp_dir}/assets$uri', f'{temp_dir}/assets/dir', '$$uri']
86    )
87