unit/test/test_njs_modules.py
2023-05-10 13:02:52 +01:00

99 lines
3.3 KiB
Python

from unit.applications.proto import TestApplicationProto
from unit.option import option
class TestNJSModules(TestApplicationProto):
prerequisites = {'modules': {'njs': 'any'}}
def njs_script_load(self, module, name=None, expect='success'):
if name is None:
name = module
with open(f'{option.test_dir}/njs/{module}/script.js', 'rb') as s:
assert expect in self.conf(s.read(), f'/js_modules/{name}')
def test_njs_modules(self):
self.njs_script_load('next')
assert 'export' in self.conf_get('/js_modules/next')
assert 'error' in self.conf_post('"blah"', '/js_modules/next')
assert 'success' in self.conf(
{
"settings": {"js_module": "next"},
"listeners": {"*:7080": {"pass": "routes/first"}},
"routes": {
"first": [{"action": {"pass": "`routes/${next.route()}`"}}],
"next": [{"action": {"return": 200}}],
},
}
)
assert self.get()['status'] == 200, 'string'
assert 'success' in self.conf({"js_module": ["next"]}, 'settings')
assert self.get()['status'] == 200, 'array'
# add one more value to array
assert len(self.conf_get('/js_modules').keys()) == 1
self.njs_script_load('next', 'next_2')
assert len(self.conf_get('/js_modules').keys()) == 2
assert 'success' in self.conf_post('"next_2"', 'settings/js_module')
assert self.get()['status'] == 200, 'array len 2'
assert 'success' in self.conf(
'"`routes/${next_2.route()}`"', 'routes/first/0/action/pass'
)
assert self.get()['status'] == 200, 'array new'
# can't update exsisting script
self.njs_script_load('global_this', 'next', expect='error')
# delete modules
assert 'error' in self.conf_delete('/js_modules/next_2')
assert 'success' in self.conf_delete('settings/js_module')
assert 'success' in self.conf_delete('/js_modules/next_2')
def test_njs_modules_import(self):
self.njs_script_load('import_from')
assert 'success' in self.conf(
{
"settings": {"js_module": "import_from"},
"listeners": {"*:7080": {"pass": "routes/first"}},
"routes": {
"first": [
{"action": {"pass": "`routes/${import_from.num()}`"}}
],
"number": [{"action": {"return": 200}}],
},
}
)
assert self.get()['status'] == 200
def test_njs_modules_this(self):
self.njs_script_load('global_this')
assert 'success' in self.conf(
{
"settings": {"js_module": "global_this"},
"listeners": {"*:7080": {"pass": "routes/first"}},
"routes": {
"first": [
{"action": {"pass": "`routes/${global_this.str()}`"}}
],
"string": [{"action": {"return": 200}}],
},
}
)
assert self.get()['status'] == 200
def test_njs_modules_invalid(self, skip_alert):
skip_alert(r'.*JS compile module.*failed.*')
self.njs_script_load('invalid', expect='error')