200 lines
8.2 KiB
Python
200 lines
8.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Run the production pairing UI methods and OLED selection with USB present."""
|
|
from pathlib import Path
|
|
import subprocess
|
|
import tempfile
|
|
import unittest
|
|
|
|
from test_replay_reset_integration import extract_braced
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
HARNESS = r'''
|
|
#include <Arduino.h>
|
|
#include <helpers/ui/DisplayDriver.h>
|
|
#include <helpers/ui/BluetoothPairingUiPolicy.h>
|
|
#include <helpers/ui/CompanionHomeLayout.h>
|
|
#include <cassert>
|
|
#include <string>
|
|
#include <vector>
|
|
ColorVal UIColor::window_bkg = 0, UIColor::title_bkg = 0, UIColor::title_txt = 1;
|
|
ColorVal UIColor::primary_txt = 1, UIColor::secondary_txt = 1, UIColor::warning_txt = 1;
|
|
ColorVal UIColor::popup_bkg = 0, UIColor::popup_txt = 1, UIColor::corp_blue = 1;
|
|
constexpr unsigned long BLE_PAIRING_DISPLAY_MILLIS = 120000;
|
|
struct TestDisplay : DisplayDriver {
|
|
bool on = false;
|
|
std::vector<std::string> text;
|
|
TestDisplay() : DisplayDriver(128, 64) {}
|
|
bool isOn() override { return on; }
|
|
void turnOn() override { on = true; }
|
|
void turnOff() override { on = false; }
|
|
void clear() override {}
|
|
void startFrame(ColorVal) override { text.clear(); }
|
|
void setTextSize(int) override {}
|
|
void setColor(ColorVal) override {}
|
|
void setCursor(int, int) override {}
|
|
void print(const char* value) override { text.emplace_back(value); }
|
|
void fillRect(int, int, int, int) override {}
|
|
void drawRect(int, int, int, int) override {}
|
|
void drawXbm(int, int, const uint8_t*, int, int) override {}
|
|
uint16_t getTextWidth(const char* value) override { return strlen(value) * 6; }
|
|
void endFrame() override {}
|
|
};
|
|
struct Board {
|
|
bool usb = true;
|
|
bool isExternalPowered() const { return usb; }
|
|
bool isUsbHostConnected() const { return usb; }
|
|
};
|
|
struct Interfaces {
|
|
bool pending = false, enabled = true, usb = true, ble = false;
|
|
bool takePairingRequest() { bool result = pending; pending = false; return result; }
|
|
};
|
|
struct UIScreen {};
|
|
struct HomeScreen : UIScreen { void showFirstPage() {} };
|
|
struct MsgPreviewScreen : UIScreen { bool hasMessages() const { return false; } };
|
|
struct Mesh {
|
|
uint32_t pin = 246810;
|
|
uint32_t getBLEPin() const { return pin; }
|
|
void notificationButton() {}
|
|
} the_mesh;
|
|
struct UITask {
|
|
TestDisplay* _display;
|
|
Board* _board;
|
|
Interfaces* _interfaceManager;
|
|
HomeScreen home_screen;
|
|
MsgPreviewScreen preview_screen;
|
|
UIScreen* home = &home_screen;
|
|
UIScreen* msg_preview = &preview_screen;
|
|
UIScreen* curr = home;
|
|
unsigned long _pairing_screen_until = 0, _next_refresh = 0;
|
|
bool _pairing_from_button = false, _deferred_msg_preview = false;
|
|
int _msgcount = 0;
|
|
UITask(TestDisplay* display, Board* board, Interfaces* interfaces)
|
|
: _display(display), _board(board), _interfaceManager(interfaces) {}
|
|
bool isBluetoothEnabled() const { return _interfaceManager->enabled; }
|
|
bool hasBluetoothConnection() const { return _interfaceManager->ble; }
|
|
bool hasConnection() const { return _interfaceManager->ble || _interfaceManager->usb; }
|
|
const char* connectedClientLabel() const {
|
|
return _interfaceManager->ble ? "BLUETOOTH" : _interfaceManager->usb ? "USB" : nullptr;
|
|
}
|
|
void setCurrScreen(UIScreen* screen) { curr = screen; _next_refresh = 100; }
|
|
void gotoHomeScreen() { setCurrScreen(home); }
|
|
bool isPairingPromptActive() const { return isPairingScreenActive(); }
|
|
bool isPairingScreenActive() const;
|
|
void renderPairingBanner();
|
|
void showPairingPin(bool from_button = false);
|
|
void finishPairingScreen(bool timed_out);
|
|
void servicePairingState();
|
|
char checkDisplayOn(char c);
|
|
void navigation(char c);
|
|
};
|
|
@METHODS@
|
|
static void homePairing(UITask* _task, TestDisplay& display) {
|
|
const int home_content_top = 20;
|
|
@HOME_SELECTION@
|
|
}
|
|
int main() {
|
|
using namespace mesh::ui;
|
|
g_mock_millis = 100;
|
|
displayPowerPrefs() = DisplayPowerPrefs();
|
|
displayPowerPrefs().usb.mode = DisplayMode::ButtonPairing;
|
|
Board board;
|
|
Interfaces interfaces;
|
|
TestDisplay display;
|
|
UITask ui{&display, &board, &interfaces};
|
|
ui.servicePairingState();
|
|
assert(!display.isOn() && !ui.isPairingScreenActive());
|
|
interfaces.pending = true; // Real PIN request, while USB remains connected.
|
|
ui.servicePairingState();
|
|
assert(display.isOn() && ui.isPairingScreenActive());
|
|
assert(!ui._pairing_from_button && interfaces.usb && !interfaces.ble);
|
|
ui.renderPairingBanner();
|
|
assert(display.text.back() == "PAIR PIN 246810");
|
|
ui.navigation('n'); // Navigation may not steal a real phone pairing prompt.
|
|
assert(ui.isPairingScreenActive());
|
|
g_mock_millis += 120000;
|
|
ui.servicePairingState();
|
|
assert(!ui.isPairingScreenActive() && !display.isOn());
|
|
|
|
// Manual button wake must expose the PIN despite USB and still allow paging.
|
|
assert(ui.checkDisplayOn('n') == 0);
|
|
assert(display.isOn() && ui.isPairingScreenActive() && ui._pairing_from_button);
|
|
assert(ui._pairing_screen_until - millis() == 15000);
|
|
ui.renderPairingBanner();
|
|
assert(display.text.back() == "PAIR PIN 246810");
|
|
ui.navigation('n');
|
|
assert(!ui.isPairingScreenActive() && !ui._pairing_from_button);
|
|
display.text.clear();
|
|
homePairing(&ui, display);
|
|
assert(display.text.back() == "246810"); // USB must not replace the idle PIN.
|
|
|
|
interfaces.pending = true;
|
|
ui.servicePairingState();
|
|
interfaces.ble = true; // Successful BLE pairing, not merely a USB connection.
|
|
ui.servicePairingState();
|
|
assert(!ui.isPairingScreenActive());
|
|
display.text.clear();
|
|
homePairing(&ui, display);
|
|
assert(display.text.back() == "CONNECTED");
|
|
|
|
// Explicit off must remain off; disabling BLE must not reveal a stale PIN.
|
|
interfaces.ble = false;
|
|
displayPowerPrefs().usb.mode = DisplayMode::Off;
|
|
ui.servicePairingState();
|
|
interfaces.pending = true;
|
|
ui.servicePairingState();
|
|
assert(!display.isOn());
|
|
ui.finishPairingScreen(false);
|
|
ui.checkDisplayOn('n');
|
|
assert(!display.isOn() && !ui._pairing_from_button);
|
|
displayPowerPrefs().usb.mode = DisplayMode::ButtonPairing;
|
|
interfaces.enabled = false;
|
|
ui.servicePairingState();
|
|
ui.checkDisplayOn('n');
|
|
assert(display.isOn() && !ui.isPairingScreenActive());
|
|
display.text.clear();
|
|
homePairing(&ui, display);
|
|
assert(display.text.back() == "CONNECTED"); // Only USB status remains.
|
|
}
|
|
'''
|
|
|
|
|
|
class CompanionPairingDisplayTests(unittest.TestCase):
|
|
def test_real_pairing_methods_and_usb_oled_selection(self):
|
|
source = (ROOT / "examples/companion_radio/ui-new/UITask.cpp").read_text()
|
|
signatures = (
|
|
"bool UITask::isPairingScreenActive() const",
|
|
"void UITask::renderPairingBanner()",
|
|
"void UITask::showPairingPin(bool from_button)",
|
|
"void UITask::finishPairingScreen(bool timed_out)",
|
|
"void UITask::servicePairingState()",
|
|
"char UITask::checkDisplayOn(char c)",
|
|
)
|
|
methods = "\n".join(extract_braced(source, signature) + "\n"
|
|
for signature in signatures)
|
|
navigation = extract_braced(source, "if (_pairing_from_button && c != 0)")
|
|
methods += "void UITask::navigation(char c) {" + navigation + "}\n"
|
|
marker = "#else\n const bool bluetooth_enabled = _task->isBluetoothEnabled();"
|
|
compact = source[source.index(marker) + len("#else\n"):]
|
|
condition = "if (client_label != nullptr || show_bluetooth_pin)"
|
|
selection = compact[:compact.index(condition)] + extract_braced(compact, condition) + "\n"
|
|
program = HARNESS.replace("@METHODS@", methods).replace("@HOME_SELECTION@", selection)
|
|
with tempfile.TemporaryDirectory(prefix="mesh-pairing-display-") as temp:
|
|
cpp, binary = Path(temp) / "test.cpp", Path(temp) / "test"
|
|
cpp.write_text(program)
|
|
result = subprocess.run([
|
|
"c++", "-std=c++17", "-Wall", "-Wextra", "-Werror",
|
|
"-Wno-unused-parameter",
|
|
"-fsanitize=address,undefined", "-fno-pie", "-no-pie",
|
|
"-isystem", str(ROOT / "test/mocks"), "-I", str(ROOT / "src"),
|
|
str(cpp), str(ROOT / "src/helpers/ui/DisplayDriver.cpp"),
|
|
"-o", str(binary),
|
|
], text=True, capture_output=True)
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
result = subprocess.run([str(binary)], text=True, capture_output=True)
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|