#!/usr/bin/env python3
"""Execute the real heap-free diagnostics collector and bounded CLI formatter."""

from pathlib import Path
import os
import shlex
import subprocess
import tempfile
import unittest

ROOT = Path(__file__).resolve().parents[1]


class PowerSavingDiagnosticsTests(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.temp = tempfile.TemporaryDirectory(prefix="meshcore-power-diag-test-")
        cls.addClassCleanup(cls.temp.cleanup)
        cls.binary = Path(cls.temp.name) / "diagnostics"
        result = subprocess.run(
            shlex.split(os.environ.get("CXX", "g++"))
            + ["-std=c++11", "-Wall", "-Wextra", "-Werror", "-pedantic",
               "-I", str(ROOT / "src"),
               str(ROOT / "test/fixtures/power_saving_diagnostics.cpp"),
               str(ROOT / "test/fixtures/power_saving_diagnostics_other.cpp"),
               "-o", str(cls.binary)],
            capture_output=True, text=True,
        )
        if result.returncode:
            raise RuntimeError(result.stderr)

    def run_case(self, name):
        result = subprocess.run([str(self.binary), name], capture_output=True, text=True)
        self.assertEqual(result.returncode, 0, result.stdout + result.stderr)

    def test_disabled_has_no_statistics_or_policy_side_effects(self):
        self.run_case("disabled")

    def test_overlapping_reasons_count_union_once(self):
        self.run_case("overlap")

    def test_off_freezes_and_next_on_starts_fresh(self):
        self.run_case("freeze")

    def test_clear_preserves_mode_but_clears_diagnostics(self):
        self.run_case("reset")

    def test_millis_and_micros_wrap_and_short_return_boundary(self):
        self.run_case("wrap")

    def test_veto_cancellation_and_sdk_returns_are_distinct_from_success(self):
        self.run_case("veto_error")

    def test_elapsed_counters_and_transition_counts_saturate(self):
        self.run_case("saturation")

    def test_cli_commands_and_reason_queries(self):
        self.run_case("commands")

    def test_invalid_commands_do_not_change_collector(self):
        self.run_case("strict")

    def test_stale_snapshot_age_and_no_residency_claim(self):
        self.run_case("format")

    def test_all_pages_fit_prefixed_160_byte_protocol_and_tiny_buffers(self):
        self.run_case("capacity")

    def test_header_state_is_shared_across_translation_units(self):
        self.run_case("shared")

    def test_unknown_mask_bits_are_not_exposed_as_real_reasons(self):
        self.run_case("unknown_mask")

    def test_serial_reason_survives_board_alias_include_order_without_changing_protocol(self):
        # WM1110 maps Serial to Serial1 in its board header, which may be seen
        # before or after this shared diagnostic header in different sources.
        header = '#include <helpers/PowerSavingDiagnostics.h>\n'
        alias = '#define Serial Serial1\n'
        code = r"""
#include <assert.h>
#include <string.h>
@CONTEXT@
using namespace mesh::powerdiag;
static_assert(static_cast<unsigned>(Reason::SerialIO) == 15,
              "Serial diagnostics retain their published bit position");
static_assert(mask(Reason::SerialIO) == 0x008000UL, "Serial mask changed");
static_assert(static_cast<unsigned>(Reason::Bridge) == 16, "Following reasons moved");
static_assert(static_cast<unsigned>(Reason::NoWake) == 23, "Final reason moved");
static_assert(ReasonCount == 24 && ValidMask == 0xffffffUL, "Reason inventory changed");
int main() {
#ifdef EXPECT_SERIAL_ALIAS
  // The diagnostic header must not undefine or retarget the actual UART.
  int Serial1 = 7;
  assert(Serial == 7);
#endif
  assert(!strcmp(reasonName(Reason::SerialIO), "serial"));
  setEnabled(true, 100);
  decision(mask(Reason::SerialIO), 101);
  const Snapshot value = snapshot(111);
  assert(value.current == 0x008000UL && value.seen == 0x008000UL);
  assert(value.blocked_ms[15] == 10 && value.transitions[15] == 1);
  char reply[160];
  assert(handleCommand("get powersaving.diag serial", reply, sizeof(reply), 111));
  assert(!strcmp(reply,
      "> serial active=1 seen=1 transitions=1 blocked_ms=10 age_ms=10"));
  assert(handleCommand("get powersaving.diag state", reply, sizeof(reply), 111));
  assert(strstr(reply, "why=serial mask=008000 seen=008000"));
  assert(handleCommand("get powersaving.diag 6", reply, sizeof(reply), 111));
  assert(strstr(reply, " serial=1/10"));
}
"""
        for name, context in (
                ("normal", header),
                ("board_before_diagnostics", alias + header),
                ("board_after_diagnostics", header + alias)):
            with self.subTest(include_order=name):
                source = Path(self.temp.name) / (name + '.cpp')
                binary = source.with_suffix('')
                source.write_text(code.replace('@CONTEXT@', context), encoding='ascii')
                result = subprocess.run(
                    shlex.split(os.environ.get('CXX', 'g++'))
                    + ['-std=c++11', '-Wall', '-Wextra', '-Werror', '-pedantic',
                       *(['-DEXPECT_SERIAL_ALIAS=1'] if name != 'normal' else []),
                       '-I', str(ROOT / 'src'), str(source), '-o', str(binary)],
                    capture_output=True, text=True, timeout=30)
                self.assertEqual(result.returncode, 0, result.stdout + result.stderr)
                result = subprocess.run([str(binary)], capture_output=True,
                                        text=True, timeout=10)
                self.assertEqual(result.returncode, 0, result.stdout + result.stderr)

    def test_common_cli_routes_without_persisting_preferences(self):
        source = (ROOT / "src/helpers/CommonCLI.cpp").read_text(encoding="ascii")
        self.assertIn('#include "PowerSavingDiagnostics.h"', source)
        self.assertEqual(source.count(
            "if (mesh::powerdiag::handleCommand(command, reply, 157, millis())) return;"), 1)
        handler = (ROOT / "src/helpers/PowerSavingDiagnostics.h").read_text(encoding="ascii")
        self.assertNotIn("savePrefs", handler)
        self.assertNotIn("#include <Arduino", handler)


if __name__ == "__main__":
    unittest.main()