0b85fe29f7
After the launch of the project, the testing infrastructure was shared with nginx project in some cases. To avoid port overlap, a decision was made to shift the port range for Unit tests. This problem was resolved a long time ago and is no longer relevant, so it is now safe to use port 8XXX range as the default, as it is more appropriate for testing purposes.
102 lines
2.6 KiB
Python
102 lines
2.6 KiB
Python
import os
|
|
|
|
import pytest
|
|
from unit.applications.proto import ApplicationProto
|
|
from unit.option import option
|
|
from unit.utils import waitforfiles
|
|
|
|
prerequisites = {'modules': {'njs': 'any'}}
|
|
|
|
client = ApplicationProto()
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup_method_fixture(temp_dir):
|
|
assert 'success' in client.conf(
|
|
{
|
|
"listeners": {"*:8080": {"pass": "routes"}},
|
|
"routes": [{"action": {"share": f"{temp_dir}/assets$uri"}}],
|
|
}
|
|
)
|
|
|
|
|
|
def create_files(*files):
|
|
assets_dir = f'{option.temp_dir}/assets/'
|
|
os.makedirs(assets_dir)
|
|
|
|
[open(assets_dir + f, 'a') for f in files]
|
|
waitforfiles(*[assets_dir + f for f in files])
|
|
|
|
|
|
def set_share(share):
|
|
assert 'success' in client.conf(share, 'routes/0/action/share')
|
|
|
|
|
|
def check_expression(expression, url='/'):
|
|
set_share(f'"`{option.temp_dir}/assets{expression}`"')
|
|
assert client.get(url=url)['status'] == 200
|
|
|
|
|
|
def test_njs_template_string(temp_dir):
|
|
create_files('str', '`string`', '`backtick', 'l1\nl2')
|
|
|
|
check_expression('/str')
|
|
check_expression('/\\\\`backtick')
|
|
check_expression('/l1\\nl2')
|
|
|
|
set_share(f'"{temp_dir}/assets/`string`"')
|
|
assert client.get()['status'] == 200
|
|
|
|
|
|
def test_njs_template_expression():
|
|
create_files('str', 'localhost')
|
|
|
|
check_expression('${uri}', '/str')
|
|
check_expression('${uri}${host}')
|
|
check_expression('${uri + host}')
|
|
check_expression('${uri + `${host}`}')
|
|
|
|
|
|
def test_njs_iteration():
|
|
create_files('Connection,Host', 'close,localhost')
|
|
|
|
check_expression('/${Object.keys(headers).sort().join()}')
|
|
check_expression('/${Object.values(headers).sort().join()}')
|
|
|
|
|
|
def test_njs_variables(temp_dir):
|
|
create_files('str', 'localhost', '127.0.0.1')
|
|
|
|
check_expression('/${host}')
|
|
check_expression('/${remoteAddr}')
|
|
check_expression('/${headers.Host}')
|
|
|
|
set_share(f'"`{temp_dir}/assets/${{cookies.foo}}`"')
|
|
assert (
|
|
client.get(headers={'Cookie': 'foo=str', 'Connection': 'close'})[
|
|
'status'
|
|
]
|
|
== 200
|
|
), 'cookies'
|
|
|
|
set_share(f'"`{temp_dir}/assets/${{args.foo}}`"')
|
|
assert client.get(url='/?foo=str')['status'] == 200, 'args'
|
|
|
|
|
|
def test_njs_invalid(skip_alert):
|
|
skip_alert(r'js exception:')
|
|
|
|
def check_invalid(template):
|
|
assert 'error' in client.conf(template, 'routes/0/action/share')
|
|
|
|
check_invalid('"`a"')
|
|
check_invalid('"`a``"')
|
|
check_invalid('"`a`/"')
|
|
|
|
def check_invalid_resolve(template):
|
|
assert 'success' in client.conf(template, 'routes/0/action/share')
|
|
assert client.get()['status'] == 500
|
|
|
|
check_invalid_resolve('"`${a}`"')
|
|
check_invalid_resolve('"`${uri.a.a}`"')
|