60 lines
3.1 KiB
Python
60 lines
3.1 KiB
Python
"""Exercise saved fast-switch controls through the production shared CLI."""
|
|
from pathlib import Path
|
|
import re
|
|
import shutil
|
|
import subprocess
|
|
import tempfile
|
|
import unittest
|
|
from cpp_source import body
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
class RadioSwitchTuningCLITest(unittest.TestCase):
|
|
def test_five_cli_roles_restore_tuning_before_first_radio_initialization(self):
|
|
for role in ('companion_radio', 'simple_repeater', 'simple_room_server',
|
|
'simple_sensor', 'simple_secure_chat'):
|
|
with self.subTest(role=role):
|
|
source = (ROOT / f'examples/{role}/main.cpp').read_text()
|
|
setup = body(source, '\nvoid setup(')
|
|
setup = re.sub(r'//[^\n]*|/\*.*?\*/', '', setup, flags=re.S)
|
|
hook = 'mesh::RadioProfileCLI::configureBootTuning('
|
|
self.assertEqual(setup.count(hook), 1)
|
|
if role != 'simple_secure_chat':
|
|
self.assertLess(setup.index('beginRegulatorSettings('), setup.index(hook))
|
|
self.assertRegex(setup, r'configureBootTuning\(regulator_prefs_fs,\s*radio_driver\)')
|
|
else:
|
|
self.assertRegex(setup, r'configureBootTuning\(\s*volatile_primary_fs\s*\?\s*nullptr\s*:\s*primary_fs,\s*radio_driver\)')
|
|
for mount in ('beginInternalPrimaryFilesystemSafely(',
|
|
'LittleFS.begin()', 'SPIFFS.begin(true)'):
|
|
self.assertEqual(setup.count(mount), 1)
|
|
self.assertLess(setup.index(mount), setup.index(hook))
|
|
self.assertEqual(setup.count('the_mesh.begin('), 1)
|
|
self.assertLess(setup.index('fast_rng.begin('), setup.index('the_mesh.begin('))
|
|
radio_init = re.search(r'\bradio_init\s*\(', setup)
|
|
self.assertIsNotNone(radio_init)
|
|
self.assertLess(setup.index(hook), radio_init.start())
|
|
if role == 'simple_secure_chat':
|
|
self.assertLess(radio_init.start(), setup.index('fast_rng.begin('))
|
|
|
|
def test_saved_controls_and_failure_transactions(self):
|
|
compiler = shutil.which('g++') or shutil.which('clang++')
|
|
self.assertIsNotNone(compiler)
|
|
with tempfile.TemporaryDirectory() as work:
|
|
for fleet_enabled in (False, True):
|
|
with self.subTest(fleet=fleet_enabled):
|
|
executable = Path(work) / f'switch-{int(fleet_enabled)}'
|
|
subprocess.run([
|
|
compiler, '-std=c++17', '-Wall', '-Wextra',
|
|
f'-DMESH_ENABLE_FLEET_CONTROL={int(fleet_enabled)}',
|
|
'-I', str(ROOT / 'test/fixtures/radio_profiles/mocks'),
|
|
'-I', str(ROOT / 'test/mocks'), '-I', str(ROOT / 'src'),
|
|
str(ROOT / 'src/helpers/RadioProfileCLI.cpp'),
|
|
str(ROOT / 'test/fixtures/radio_profiles/switch_tuning_test.cpp'),
|
|
'-o', str(executable)], check=True, capture_output=True, text=True)
|
|
subprocess.run([str(executable)], check=True)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|