feat: startup environment check + self-heal (v1.3)
- env_check(): verifies daemon/pineapd/UCI/monitors/recon DB, auto-fixes fixable issues and re-verifies; forces runtime SSID-pool broadcast off to match the UI (kills the 'pool on but UI shows off' gap) - server.py --env-check CLI; payload.sh runs it verbosely before starting, aborts on core failure - serve() runs the check at every startup (boot + procd respawn) - /api/health exposes env report + pool_runtime; /api/recon/status exposes wlan0_pinned (2.4GHz under-sampling warning); recon page warns when a radio0 AP pins wlan0mon - recon start failures now include the daemon reason in the UI error - shared stabilization refactor (_stabilize_uci/PINEAPD_SAFE_UCI/_raise_monitors) - tests: test_env_check.py (18) + health/recon updates
This commit is contained in:
@@ -0,0 +1,230 @@
|
||||
import importlib
|
||||
import io
|
||||
import os
|
||||
import sqlite3
|
||||
import sys
|
||||
import tempfile
|
||||
import unittest
|
||||
from contextlib import redirect_stdout
|
||||
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'payload', 'user', 'remote_access', 'pager-webui'))
|
||||
import server
|
||||
|
||||
|
||||
def setUpModule():
|
||||
importlib.reload(server)
|
||||
|
||||
|
||||
class EnvCheckTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
fd, self.db = tempfile.mkstemp(suffix='.db')
|
||||
os.close(fd)
|
||||
conn = sqlite3.connect(self.db)
|
||||
conn.execute('CREATE TABLE scan(id INTEGER PRIMARY KEY, time INT, name TEXT)')
|
||||
conn.execute("INSERT INTO scan (time, name) VALUES (1786466531, 'pager')")
|
||||
conn.commit()
|
||||
conn.close()
|
||||
server.RECON_DB = self.db
|
||||
self.runs = []
|
||||
self.ping_ok = True
|
||||
self.daemon_ok = True
|
||||
self.iface_up = {'wlan0mon': True, 'wlan1mon': True}
|
||||
self.uci_state = {}
|
||||
server.ENV_CHECK_STATE.update({'report': None, 'overall': None, 'updated': 0,
|
||||
'pool_runtime': None})
|
||||
self.old_iface_up = server._iface_up
|
||||
server._iface_up = lambda name: self.iface_up.get(name, True)
|
||||
self.old_daemon = server.daemon_sock_call
|
||||
self.old_run = server.device_run
|
||||
server.daemon_sock_call = self.fake_daemon
|
||||
server.device_run = self.fake_run
|
||||
|
||||
def tearDown(self):
|
||||
server._iface_up = self.old_iface_up
|
||||
server.daemon_sock_call = self.old_daemon
|
||||
server.device_run = self.old_run
|
||||
try:
|
||||
os.unlink(self.db)
|
||||
except OSError:
|
||||
pass
|
||||
|
||||
def fake_daemon(self, method, path, body=None, timeout=10):
|
||||
if self.daemon_ok:
|
||||
return 200, {'autossidpool': False}
|
||||
return 0, None
|
||||
|
||||
def fake_run(self, args, timeout=20, input_data=None):
|
||||
self.runs.append(list(args))
|
||||
a = list(args)
|
||||
if a[0] == 'pidof' and a[1] == 'pineapd':
|
||||
return (0, '12345\n', '') if self.ping_ok else (1, '', '')
|
||||
if a[0] == 'uci':
|
||||
if a[1] == 'set':
|
||||
k, _, v = a[2].partition('=')
|
||||
self.uci_state[k] = v
|
||||
return (0, '', '')
|
||||
if a[1] == 'delete':
|
||||
for k in list(self.uci_state):
|
||||
if k == a[2] or k.startswith(a[2] + '.'):
|
||||
del self.uci_state[k]
|
||||
return (0, '', '')
|
||||
if a[1] == 'get':
|
||||
return (0, self.uci_state.get(a[2], '') + '\n', '')
|
||||
if a[1] == 'commit':
|
||||
return (0, '', '')
|
||||
if a[1] == 'show':
|
||||
sec = a[2]
|
||||
return (0, ''.join("%s=%s\n" % (k, v) for k, v in self.uci_state.items()
|
||||
if k.startswith(sec + '.')), '')
|
||||
if a[0] == '_pineap':
|
||||
return (0, '', '')
|
||||
if a[0] in ('ip', '/etc/init.d/pineapd'):
|
||||
return (0, '', '')
|
||||
return (0, '', '')
|
||||
|
||||
def safe_set(self):
|
||||
for key, value in server.PINEAPD_SAFE_UCI.items():
|
||||
self.uci_state[key] = value
|
||||
|
||||
def steps(self, report, needle):
|
||||
return [r for r in report if needle in r['detail']]
|
||||
|
||||
def test_pass_when_state_sane(self):
|
||||
self.safe_set()
|
||||
report = server.env_check()
|
||||
self.assertEqual(server.ENV_CHECK_STATE['overall'], 'pass')
|
||||
self.assertEqual([r['ok'] for r in report],
|
||||
['pass'] * len(report))
|
||||
self.assertEqual(server.ENV_CHECK_STATE['pool_runtime'], 'disabled')
|
||||
self.assertIn(['_pineap', 'SSIDPOOL', 'DISABLE'], self.runs)
|
||||
|
||||
def test_applies_sane_defaults_when_missing(self):
|
||||
report = server.env_check()
|
||||
self.assertEqual(self.steps(report, 'sane-off UCI defaults applied')[0]['ok'], 'fixed')
|
||||
for key, value in server.PINEAPD_SAFE_UCI.items():
|
||||
self.assertEqual(self.uci_state[key], value)
|
||||
|
||||
def test_uci_pass_when_already_set(self):
|
||||
self.safe_set()
|
||||
report = server.env_check()
|
||||
self.assertEqual(self.steps(report, 'sane-off UCI defaults already set')[0]['ok'], 'pass')
|
||||
|
||||
def test_clears_refilled_pool_list(self):
|
||||
self.uci_state['pineapd.@ssidpool[0].ssid'] = 'QmVlcg=='
|
||||
report = server.env_check()
|
||||
actions = ' | '.join((r.get('action') or '') for r in report)
|
||||
self.assertIn('pool-list cleared', actions)
|
||||
self.assertNotIn('pineapd.@ssidpool[0].ssid', self.uci_state)
|
||||
|
||||
def test_restarts_pineapd_when_down(self):
|
||||
self.safe_set()
|
||||
self.ping_ok = False
|
||||
self.pidof_calls = 0
|
||||
real_ping = self.fake_run
|
||||
|
||||
def ping_then_up(args, timeout=20, input_data=None):
|
||||
if args[0] == 'pidof' and args[1] == 'pineapd':
|
||||
self.pidof_calls += 1
|
||||
if self.pidof_calls > 1:
|
||||
return (0, '12345\n', '')
|
||||
return real_ping(args, timeout=timeout, input_data=input_data)
|
||||
|
||||
server.device_run = ping_then_up
|
||||
report = server.env_check()
|
||||
self.assertEqual(self.steps(report, 'pineapd was down')[0]['ok'], 'fixed')
|
||||
self.assertIn(['/etc/init.d/pineapd', 'restart'], self.runs)
|
||||
|
||||
def test_fail_when_pineapd_stays_down(self):
|
||||
self.safe_set()
|
||||
self.ping_ok = False
|
||||
report = server.env_check()
|
||||
self.assertEqual(self.steps(report, 'did not come back')[0]['ok'], 'fail')
|
||||
self.assertEqual(server.ENV_CHECK_STATE['overall'], 'fail')
|
||||
|
||||
def test_fail_when_daemon_unreachable(self):
|
||||
self.daemon_ok = False
|
||||
report = server.env_check()
|
||||
self.assertEqual(self.steps(report, 'daemon unreachable')[0]['ok'], 'fail')
|
||||
self.assertEqual(server.ENV_CHECK_STATE['overall'], 'fail')
|
||||
|
||||
def test_raises_down_monitors(self):
|
||||
self.safe_set()
|
||||
self.iface_up = {'wlan0mon': False, 'wlan1mon': True}
|
||||
report = server.env_check()
|
||||
self.assertEqual(self.steps(report, 'monitor interfaces brought up')[0]['ok'], 'fixed')
|
||||
self.assertIn(['ip', 'link', 'set', 'wlan0mon', 'up'], self.runs)
|
||||
|
||||
def test_monitors_up_pass(self):
|
||||
self.safe_set()
|
||||
report = server.env_check()
|
||||
self.assertEqual(self.steps(report, 'monitors up')[0]['ok'], 'pass')
|
||||
|
||||
def test_recon_db_unreadable_fails(self):
|
||||
self.safe_set()
|
||||
os.unlink(self.db)
|
||||
report = server.env_check()
|
||||
self.assertEqual(self.steps(report, 'recon DB unreadable')[0]['ok'], 'fail')
|
||||
self.assertEqual(server.ENV_CHECK_STATE['overall'], 'fail')
|
||||
|
||||
def test_recon_db_readable_reports_count(self):
|
||||
self.safe_set()
|
||||
report = server.env_check()
|
||||
step = self.steps(report, 'recon DB readable')[0]
|
||||
self.assertEqual(step['ok'], 'pass')
|
||||
self.assertIn('(1 scans)', step['detail'])
|
||||
|
||||
def test_wlan0_pinned_warns(self):
|
||||
self.safe_set()
|
||||
self.uci_state['wireless.wlan0wpa.disabled'] = '0'
|
||||
report = server.env_check()
|
||||
self.assertEqual(self.steps(report, '2.4GHz under-sampled')[0]['ok'], 'warn')
|
||||
|
||||
def test_wlan0_not_pinned_when_absent(self):
|
||||
self.safe_set()
|
||||
report = server.env_check()
|
||||
self.assertEqual(self.steps(report, 'no radio0 AP pins wlan0mon')[0]['ok'], 'pass')
|
||||
|
||||
def test_wlan0_not_pinned_when_disabled(self):
|
||||
self.safe_set()
|
||||
self.uci_state['wireless.wlan0open.disabled'] = '1'
|
||||
self.uci_state['wireless.wlan0wpa.disabled'] = '1'
|
||||
report = server.env_check()
|
||||
self.assertEqual(self.steps(report, 'no radio0 AP pins wlan0mon')[0]['ok'], 'pass')
|
||||
|
||||
def test_cli_exits_zero_on_pass(self):
|
||||
self.safe_set()
|
||||
buf = io.StringIO()
|
||||
with redirect_stdout(buf):
|
||||
code = server.env_check_cli()
|
||||
self.assertEqual(code, 0)
|
||||
self.assertIn('[PASS]', buf.getvalue())
|
||||
self.assertIn('ENVIRONMENT CHECK: PASS', buf.getvalue())
|
||||
|
||||
def test_cli_exits_one_on_fail(self):
|
||||
self.safe_set()
|
||||
self.daemon_ok = False
|
||||
buf = io.StringIO()
|
||||
with redirect_stdout(buf):
|
||||
code = server.env_check_cli()
|
||||
self.assertEqual(code, 1)
|
||||
self.assertIn('[FAIL]', buf.getvalue())
|
||||
|
||||
def test_health_exposes_env_and_pool_runtime(self):
|
||||
self.safe_set()
|
||||
server.env_check()
|
||||
status, payload = server.h_health(type('C', (), {'query': {}})())
|
||||
self.assertEqual(status, 200)
|
||||
self.assertEqual(payload['pool_runtime'], 'disabled')
|
||||
self.assertEqual(payload['env']['overall'], 'pass')
|
||||
self.assertEqual(payload['env']['counts']['pass'], len(payload['env']['steps']))
|
||||
|
||||
def test_recon_status_exposes_wlan0_pinned(self):
|
||||
self.safe_set()
|
||||
self.uci_state['wireless.wlan0open.disabled'] = '0'
|
||||
status, payload = server.h_recon_status(type('C', (), {'query': {}})())
|
||||
self.assertEqual(status, 200)
|
||||
self.assertTrue(payload['wlan0_pinned'])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
+2
-1
@@ -224,7 +224,8 @@ class DaemonSockTest(unittest.TestCase):
|
||||
type('C', (), {'args': (), 'body': {'scan_time': 30}})())
|
||||
self.assertEqual(status, 502)
|
||||
self.assertEqual(data['error'], 'native recon scan failed')
|
||||
self.assertEqual(data['detail'], {'error': 'no radio'})
|
||||
self.assertEqual(data['detail'], 'recon/new: no radio')
|
||||
self.assertEqual(data['daemon'], {'error': 'no radio'})
|
||||
self.assertFalse(server._recon_scan_state['active'])
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user