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