mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-18 21:43:51 +08:00
AngusBell97
65 lines
3.1 KiB
Python
65 lines
3.1 KiB
Python
"""Exercise the real Galaxy settings handler with an in-memory Params store."""
|
|
import ast
|
|
from pathlib import Path
|
|
|
|
from flask import Flask, jsonify, request
|
|
import pytest
|
|
|
|
from test_screen_settings import Params
|
|
|
|
|
|
def client_for(params):
|
|
source = Path(__file__).resolve().parents[3] / 'starpilot/system/the_galaxy/the_galaxy.py'
|
|
setup = next(n for n in ast.parse(source.read_text()).body if isinstance(n, ast.FunctionDef) and n.name == 'setup')
|
|
route = next(n for n in setup.body if isinstance(n, ast.FunctionDef) and n.name == 'get_param')
|
|
app = Flask(__name__)
|
|
env = dict(app=app, request=request, jsonify=jsonify, params=params, update_starpilot_toggles=lambda: None,
|
|
LONGITUDINAL_MODE_KEYS=set(), PERSONALITY_PROFILES_PARAM='', PERSONALITY_PARKED_PARAM_KEYS=set(),
|
|
PERSONALITY_PROFILE_ENABLE_PARAM_KEYS=set(), FAVORITE_SLOTS_PARAM='', MODEL_SMOOTHING_KEYS=set(),
|
|
PERSONALITY_ADVANCED_PARAM_KEYS=set(), PERSONALITY_FOLLOW_PARAM_KEYS=set(),
|
|
_get_param_type_info=lambda: (set(), {}))
|
|
exec(compile(ast.Module(body=[route], type_ignores=[]), str(source), 'exec'), env)
|
|
return app.test_client()
|
|
|
|
|
|
def test_mode_write_returns_remembered_manual_value():
|
|
params = Params({'ScreenBrightnessOnroad': 32})
|
|
response = client_for(params).put('/api/params', json={'key': 'ScreenBrightnessOnroad', 'value': 101})
|
|
assert response.status_code == 200
|
|
assert response.json['updated'] == {'ScreenBrightnessOnroad': 101, 'ScreenBrightnessOnroadManual': 32}
|
|
|
|
|
|
@pytest.mark.parametrize('key,value,expected', [('ScreenBrightnessOffset', -25, -25), ('ScreenBrightnessOnroad', 0, 0),
|
|
('StandbyWakeCriticalAlert', False, False)])
|
|
def test_screen_values_are_validated_and_return_typed_readback(key, value, expected):
|
|
params = Params()
|
|
response = client_for(params).put('/api/params', json={'key': key, 'value': value})
|
|
assert response.status_code == 200
|
|
assert response.json['updated'][key] == expected
|
|
assert params.values[key] == expected
|
|
|
|
|
|
@pytest.mark.parametrize('key,value', [
|
|
('ScreenBrightnessOffset', 31), ('ScreenBrightnessOnroadOffset', -31), ('StandbyWakeEngage', 'false'), ('StandbyWakeFake', True),
|
|
])
|
|
def test_invalid_screen_requests_return_400_without_writes(key, value):
|
|
params = Params()
|
|
response = client_for(params).put('/api/params', json={'key': key, 'value': value})
|
|
assert response.status_code == 400
|
|
assert not params.values
|
|
|
|
|
|
def test_native_write_failure_is_reported_instead_of_success():
|
|
params = Params()
|
|
params.fail_key = 'ScreenBrightnessOffset'
|
|
response = client_for(params).put('/api/params', json={'key': 'ScreenBrightnessOffset', 'value': 10})
|
|
assert response.status_code == 503
|
|
|
|
|
|
def test_galaxy_failed_wake_write_reports_error_and_restores_selection():
|
|
from test_screen_settings_transactions import SilentOnceParams, galaxy_params
|
|
params = SilentOnceParams({'StandbyWakeCriticalAlert': True})
|
|
response = client_for(galaxy_params(params)).put('/api/params', json={'key': 'StandbyWakeCriticalAlert', 'value': False})
|
|
assert response.status_code == 503
|
|
assert params.values == {'StandbyWakeCriticalAlert': True}
|