fix(adb): ADB-off is asked by every device path, not inferred from a missing poller (t23)
The t22 review's O1, and the same bug class as t21's F1 — so it gets the same
structural treatment instead of a note.
WHAT WAS MEASURED. With a poller injected into an entry whose own option said
`adb_enabled: False`, the installed-apps sweep sent 4 real shells (the reviewer;
3 on this test's double, which yields one launchable package) and a label probe
sent 1 more. Both gated on `self._poller is None` — and the poller's absence is a
property of the SETUP code (`_build_poller_factory` returns None when ADB is off),
which is precisely the indirect guarantee the t21 repair had already shown to be
insufficient. Production was unreachable only through that setup path.
THE FIX. `_adb_disabled()` is now asked directly by every path in the manager
that can reach the device, and `_adb_active()` (option ON *and* a poller) is the
shared predicate for the ones that also need a transport. The `available` property
deliberately keeps its transport meaning (diagnostics reads it as "ADB is
configured"); a new docstring says so, because overloading it is how the gate got
lost in the first place.
Gated paths, and what each asks now:
* intent scan — the option (t21, unchanged) AND the private shell site itself
* installed-apps sweep — the option first, then poller+connected
* installed-apps refresh (public entry) — returns the last table, starts nothing
* app-label lookup / drain / the ONE probe site (`_dump_label`) — the option; a
lookup returns None and queues NOTHING (no request, no task, no cache write)
* ADB text entry, media keys, bring-to-foreground, digit and enter keys — `_adb_active()`
* the heartbeat `_refresh` — the option: no wake probe, no media dump, no
running-apps poll, no foreground fallback, no label drain, no logcat ensure
* the watchdog — the option BEFORE the poller factory, so not even a factory can
build a transport; and `_watchdog_cycle` refuses a direct call too (a TCP
connect is device traffic)
* the logcat stream — the option at the ensure gate, the option in
`_open_logcat_stream`, and the ladder exits WITHOUT counting a failure (a local
option must not auto-disable the option for the session)
* the public `adb_poller` accessor — None while the option is off, which closes
the shell paths that live OUTSIDE this module (the `adb_command` service, the
screen-capture image and the intent shell) without reaching into them: all
three already refuse on None with their own message
* `test_connect` (the pairing probe) stays ungated on purpose: it runs BEFORE an
entry exists, and the setup menu only offers that step when the ADB option is
on (config_flow.py). `async_stop`/`_note_failure` disconnect calls are teardown,
not traffic.
PROOF, BY TRAFFIC. tests/test_adb_off_gates_every_device_path.py drives every
audited path — as a parametrized AUDIT table plus one full cycle on one manager —
with `adb_enabled: False` and a double that WOULD answer everything, and asserts
the poller's complete traffic list (shells, polls, commands, connects) is EMPTY,
plus that nothing was cached (no intent answer, no app table, no stamp, no queue,
no store field, no wake reading). It also proves the watchdog never builds a
poller and never connects, that a direct cycle does not connect either, that the
option is a CHECK and not a latch (the same manager resumes the sweep, the scan,
the label queue, the media key, the digit key and the accessor when the option
flips back on), and that ADB-off and no-poller stay told apart.
PRE-FIX PROOF: the same file against `git archive HEAD` FAILS 16 of 23 tests, and
the failures print the measured traffic — the sweep's 3 shells, the label probe's
`pm dump … | grep …`, `input text hello`, `media:play`, `foreground:<pkg>`,
`digit:5`, `enter`, and 4 polls per heartbeat cycle.
HARNESS TRUTHFULNESS. The suites' managers were built with `entry=None`, which
means ADB is OFF (fail closed, matching `__init__.py`). Those harnesses now pass a
realistic entry whose option says ADB is on — they model production instead of the
one combination the gates block. No production behaviour changed for an ADB-on
entry: 923 passed / 5 skipped, up from 900 by this file.
Floors re-derived from the new walk (1661 blocks → 1627, same 2% band): raised,
because that is what the re-measurement says. Drift exit 0. Strings files
byte-identical.
NOT VERIFIED HERE: the device side is a scripted double, so what is proven is that
NOTHING WAS SENT, not what firmware would do; no real Shield was involved.
This commit is contained in:
1 parent
8f064b3247
commit
de1c34777d
8 files changed
+651
-37
No files matched your search
@@ -143,7 +143,7 @@ HA_CALL_OWNERS: dict[str, tuple[str, str]] = {
|
||||
#: Non-vacuity floors — the ONE source of truth for the standalone entry point
|
||||
#: and the pytest wrapper (t39 F1). They sit below the current walk (26 modules,
|
||||
#: 77 imports, 282 relative imports, 65 checked arguments, 6 awaited HA calls,
|
||||
#: 1650 statement blocks) so a broken or mis-targeted walk fails loudly instead of
|
||||
#: 1661 statement blocks) so a broken or mis-targeted walk fails loudly instead of
|
||||
#: printing zeros and exiting 0.
|
||||
#:
|
||||
#: THE MARGIN POLICY (t20/O3 — the t19 review asked for this in writing). A floor
|
||||
@@ -247,7 +247,11 @@ FLOORS: dict[str, int] = {
|
||||
# t21: the walk is 1650 (the ADB-off guard returns before the scan, and the
|
||||
# unterminated-value check is a few blocks more), so the 2% band is re-derived
|
||||
# again: 1617.
|
||||
"statement_blocks": 1617,
|
||||
# t23: the walk is 1661 — the ADB-off gate is asked by every device path now
|
||||
# (sweep, label probe, logcat, heartbeat, watchdog, delegation, the accessor),
|
||||
# which is a dozen more statement blocks. Re-derived at the same 2% band: 1627.
|
||||
# Raised, not lowered, because that is what the re-measured walk says.
|
||||
"statement_blocks": 1627,
|
||||
}
|
||||
|
||||
#: The interpreter floor Home Assistant itself declares: PyPI metadata for
|
||||
|
||||
@@ -0,0 +1,412 @@
|
||||
"""t23/O1 — with ADB switched off, NO path reaches the device.
|
||||
|
||||
The t22 review measured the harm: with a poller injected into an entry whose own
|
||||
option said ``adb_enabled: False``, the installed-apps sweep sent **4** real
|
||||
shells and a label probe sent **1** more. Both gated on the POLLER's absence,
|
||||
which is a property of the setup code (``_build_poller_factory`` returns None when
|
||||
ADB is off) — exactly the indirect guarantee the t21 repair had already proven
|
||||
insufficient for the intent scan.
|
||||
|
||||
This file drives EVERY path in the connection manager that can touch the device
|
||||
and asserts on the TRAFFIC (shells, polls, commands, connects), because the
|
||||
traffic is the harm — a correct-looking status string is not the point. The
|
||||
device side is a scripted double that would answer everything: no real Shield is
|
||||
involved, so what is proven is that nothing was sent, not what firmware would do.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
import contextlib
|
||||
from types import SimpleNamespace
|
||||
|
||||
import pytest
|
||||
from nsh.connection import (
|
||||
FIELD_AWAKE,
|
||||
FIELD_FOREGROUND,
|
||||
FIELD_RUNNING_APPS,
|
||||
FIELD_SESSIONS,
|
||||
ShieldConnectionManager,
|
||||
)
|
||||
from nsh.const import CONF_ADB_ENABLED, CONF_LOGCAT_EVENTS
|
||||
from nsh.ha_adb_common import (
|
||||
INTENT_SCAN_ADB_DISABLED,
|
||||
INTENT_SCAN_ADB_UNAVAILABLE,
|
||||
INTENT_SCAN_FRESH,
|
||||
INTENT_SCAN_FRESH_EMPTY,
|
||||
)
|
||||
|
||||
PKG = "com.netflix.ninja"
|
||||
|
||||
#: A minimal answer that IS a resolver section, so a scan that were allowed to run
|
||||
#: would count as a real answer (nothing here fails for a second reason).
|
||||
RESOLVER_SECTION = (
|
||||
"Activity Resolver Table:\n"
|
||||
" Schemes:\n"
|
||||
" https:\n"
|
||||
" deca7fa com.netflix.ninja/.MainActivity filter fc1\n"
|
||||
' Action: "android.intent.action.VIEW"\n'
|
||||
"Service Resolver Table:\n"
|
||||
)
|
||||
|
||||
|
||||
class WillingPoller:
|
||||
"""A poller that answers EVERY call and records every one (t23/O1).
|
||||
|
||||
Nothing here refuses: if a path is still gated by the wrong fact, this double
|
||||
will happily talk to "the device" and the shell/command count proves it.
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.connected = True
|
||||
self.awake = True
|
||||
self.foreground = "app.a"
|
||||
self.audio_dump = " package=app.a\n"
|
||||
self.shell_calls: list[str] = []
|
||||
self.poll_calls: list[str] = []
|
||||
self.media_calls: list[str] = []
|
||||
self.digit_calls: list[str] = []
|
||||
self.foreground_sends: list[str] = []
|
||||
self.enter_calls = 0
|
||||
self.connect_calls = 0
|
||||
self.disconnect_calls = 0
|
||||
self.sessions = [SimpleNamespace(package="app.a", playing=True)]
|
||||
|
||||
# ── the transport ──
|
||||
async def connect(self) -> bool:
|
||||
self.connect_calls += 1
|
||||
self.connected = True
|
||||
return True
|
||||
|
||||
async def disconnect(self) -> None:
|
||||
self.disconnect_calls += 1
|
||||
self.connected = False
|
||||
|
||||
async def health_check(self) -> bool:
|
||||
return self.connected
|
||||
|
||||
# ── the heartbeat ──
|
||||
async def is_awake(self) -> bool | None:
|
||||
self.poll_calls.append("is_awake")
|
||||
return self.awake
|
||||
|
||||
async def poll_media_sessions(self, foreground_package=None, head_lines=200):
|
||||
self.poll_calls.append("poll_media_sessions")
|
||||
return self.sessions[0] if self.sessions else None
|
||||
|
||||
@property
|
||||
def all_sessions(self):
|
||||
return self.sessions
|
||||
|
||||
@property
|
||||
def last_dump(self):
|
||||
return self.audio_dump
|
||||
|
||||
async def poll_running_apps(self) -> list[dict]:
|
||||
self.poll_calls.append("poll_running_apps")
|
||||
return [{"package": "app.a", "name": "App A", "icon": None}]
|
||||
|
||||
async def poll_foreground(self) -> str | None:
|
||||
self.poll_calls.append("poll_foreground")
|
||||
return self.foreground
|
||||
|
||||
# ── anything a shell can reach ──
|
||||
async def shell(self, command: str):
|
||||
self.shell_calls.append(command)
|
||||
if "Activity Resolver Table" in command:
|
||||
return RESOLVER_SECTION
|
||||
if "pm list packages" in command:
|
||||
return f"package:{PKG}\n"
|
||||
if "resolve-activity" in command:
|
||||
return f"{PKG}/.MainActivity\n"
|
||||
if "pm dump" in command:
|
||||
return 'Application Label: "Netflix"\n'
|
||||
return "ok\n"
|
||||
|
||||
async def send_media_command(self, command: str) -> bool:
|
||||
self.media_calls.append(command)
|
||||
return True
|
||||
|
||||
async def bring_to_foreground(self, package: str) -> bool:
|
||||
self.foreground_sends.append(package)
|
||||
return True
|
||||
|
||||
async def send_digit_keyevent(self, digit: str) -> bool:
|
||||
self.digit_calls.append(digit)
|
||||
return True
|
||||
|
||||
async def send_enter_keyevent(self) -> bool:
|
||||
self.enter_calls += 1
|
||||
return True
|
||||
|
||||
@property
|
||||
def config(self):
|
||||
return {"host": "shield.invalid", "port": 5555, "key_path": "/tmp/does-not-exist"}
|
||||
|
||||
@property
|
||||
def traffic(self) -> list[str]:
|
||||
"""EVERY device-touching call, in one list — the harm in one place (t23)."""
|
||||
return [
|
||||
*(f"shell:{call}" for call in self.shell_calls),
|
||||
*(f"poll:{call}" for call in self.poll_calls),
|
||||
*(f"media:{call}" for call in self.media_calls),
|
||||
*(f"digit:{digit}" for digit in self.digit_calls),
|
||||
*(f"foreground:{pkg}" for pkg in self.foreground_sends),
|
||||
*(["enter"] * self.enter_calls),
|
||||
*(["connect"] * self.connect_calls),
|
||||
]
|
||||
|
||||
|
||||
def _entry(*, adb_enabled: bool, **options):
|
||||
"""The entry's OWN options decide — this is the fact every path must ask."""
|
||||
return SimpleNamespace(
|
||||
options={
|
||||
CONF_ADB_ENABLED: adb_enabled,
|
||||
# ON, so the logcat path is skipped for the OPTION's reason (ADB off)
|
||||
# rather than for its own (a stream nobody asked for).
|
||||
CONF_LOGCAT_EVENTS: True,
|
||||
**options,
|
||||
},
|
||||
data={},
|
||||
title="Shield",
|
||||
entry_id="e1",
|
||||
)
|
||||
|
||||
|
||||
def _manager(poller, *, adb_enabled: bool) -> ShieldConnectionManager:
|
||||
return ShieldConnectionManager(
|
||||
hass=None, entry=_entry(adb_enabled=adb_enabled), poller=poller
|
||||
)
|
||||
|
||||
|
||||
#: THE AUDIT (t23/O1), as data: every path in the manager that can reach the
|
||||
#: device, with the call that drives it. Each is run against a manager whose
|
||||
#: option says ADB is off, and the poller's traffic must stay EMPTY.
|
||||
AUDITED_PATHS = [
|
||||
("intent scan", lambda m: m.async_scan_intents(PKG)),
|
||||
("intent scan (forced)", lambda m: m.async_scan_intents(PKG, force=True)),
|
||||
("installed-apps refresh", lambda m: m.async_refresh_installed_apps(force=True)),
|
||||
("installed-apps refresh (non-forced)", lambda m: m.async_refresh_installed_apps()),
|
||||
("app-label lookup", lambda m: _sync_label(m)),
|
||||
("app-label drain (watchdog side)", lambda m: _drain(m)),
|
||||
("adb text entry", lambda m: m.send_adb_input_text("hello")),
|
||||
("media key send", lambda m: m.send_media_command("play")),
|
||||
("bring to foreground", lambda m: m.bring_to_foreground(PKG)),
|
||||
("digit key send", lambda m: m.send_digit_key("5")),
|
||||
("enter key send", lambda m: m.send_enter_key()),
|
||||
("watchdog cycle", lambda m: m._watchdog_cycle(5.0)),
|
||||
("heartbeat refresh", lambda m: m._refresh()),
|
||||
("logcat ensure", lambda m: _logcat(m)),
|
||||
("watchdog start", lambda m: _start_watchdog(m)),
|
||||
]
|
||||
|
||||
|
||||
async def _maybe_await(value):
|
||||
"""Drive an audited path: some are coroutines, some are plain calls (t23)."""
|
||||
if asyncio.iscoroutine(value):
|
||||
return await value
|
||||
return value
|
||||
|
||||
|
||||
def _sync_label(manager):
|
||||
return manager.app_label(PKG)
|
||||
|
||||
|
||||
async def _settle_label_tasks(manager):
|
||||
"""Await any label fetch the drain started, so its shell WOULD have happened."""
|
||||
for task in list(manager._tasks):
|
||||
if task.get_name().startswith("app_label"):
|
||||
with contextlib.suppress(Exception):
|
||||
await task
|
||||
|
||||
|
||||
async def _drain(manager):
|
||||
"""The render path queues a label fetch, then the watchdog drains the queue."""
|
||||
manager.app_label(PKG)
|
||||
manager._drain_app_label_requests()
|
||||
await _settle_label_tasks(manager)
|
||||
|
||||
|
||||
def _logcat(manager):
|
||||
manager._ensure_logcat_stream()
|
||||
|
||||
|
||||
async def _start_watchdog(manager):
|
||||
await manager.async_start()
|
||||
await asyncio.sleep(0.01) # would be one cycle if the loop had started
|
||||
await manager.async_stop()
|
||||
|
||||
|
||||
class TestEveryAuditedPathSendsNothing:
|
||||
"""ADB off + a poller that WOULD answer = zero traffic on every path."""
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
("name", "drive"), AUDITED_PATHS, ids=[name for name, _ in AUDITED_PATHS]
|
||||
)
|
||||
async def test_the_path_does_not_touch_the_device(self, name, drive):
|
||||
poller = WillingPoller()
|
||||
manager = _manager(poller, adb_enabled=False)
|
||||
|
||||
await _maybe_await(drive(manager))
|
||||
await asyncio.sleep(0) # let any task the path started really run
|
||||
|
||||
assert poller.traffic == [], f"{name} reached the device: {poller.traffic}"
|
||||
|
||||
async def test_a_full_cycle_of_every_path_still_sends_nothing(self):
|
||||
"""One manager, every path in sequence — the whole cycle, not one call."""
|
||||
poller = WillingPoller()
|
||||
manager = _manager(poller, adb_enabled=False)
|
||||
|
||||
# 1. the intent scan (t21's repair, asked again one layer deeper)
|
||||
assert (await manager.async_scan_intents(PKG)).status == INTENT_SCAN_ADB_DISABLED
|
||||
assert (
|
||||
await manager.async_scan_intents(PKG, force=True)
|
||||
).status == INTENT_SCAN_ADB_DISABLED
|
||||
|
||||
# 2. the installed-apps sweep — the 4-shell path the review measured
|
||||
assert await manager.async_refresh_installed_apps(force=True) == []
|
||||
assert manager.installed_apps() == []
|
||||
|
||||
# 3. the app-label probe — the 1-shell path, both entry points
|
||||
assert manager.app_label(PKG) is None
|
||||
manager._drain_app_label_requests()
|
||||
await _settle_label_tasks(manager)
|
||||
|
||||
# 4. text entry and every key/command delegation
|
||||
assert await manager.send_adb_input_text("hello") is False
|
||||
assert await manager.send_media_command("play") is False
|
||||
assert await manager.bring_to_foreground(PKG) is False
|
||||
assert await manager.send_digit_key("5") is False
|
||||
assert await manager.send_enter_key() is False
|
||||
|
||||
# 5. the heartbeat, the watchdog and the logcat stream
|
||||
await manager._watchdog_cycle(5.0)
|
||||
await _start_watchdog(manager)
|
||||
manager._ensure_logcat_stream()
|
||||
await asyncio.sleep(0)
|
||||
|
||||
# 6. the public accessor the OUT-OF-FILE consumers (adb_command service,
|
||||
# screen capture, intent shell) read — it must refuse too
|
||||
assert manager.adb_poller is None
|
||||
|
||||
assert poller.traffic == [], poller.traffic
|
||||
|
||||
async def test_nothing_is_cached_when_the_cycle_is_refused(self):
|
||||
"""No cache write either: an unpermitted scan is not an answer."""
|
||||
poller = WillingPoller()
|
||||
manager = _manager(poller, adb_enabled=False)
|
||||
|
||||
await manager.async_scan_intents(PKG)
|
||||
await manager.async_refresh_installed_apps(force=True)
|
||||
manager.app_label(PKG)
|
||||
manager._drain_app_label_requests()
|
||||
await manager._refresh()
|
||||
|
||||
assert manager.intent_scan_state(PKG) is None, "no intent answer"
|
||||
assert manager.installed_apps() == [], "no app table"
|
||||
assert manager.installed_apps_stale is True, "nothing was stamped fresh"
|
||||
assert manager.app_label_requests == set(), "nothing queued"
|
||||
assert manager.app_label_inflight == set(), "nothing in flight"
|
||||
assert manager.store.get(FIELD_AWAKE) is None, "no wake reading"
|
||||
assert manager.store.get(FIELD_SESSIONS, []) == []
|
||||
assert manager.store.get(FIELD_RUNNING_APPS, []) == []
|
||||
assert manager.store.get(FIELD_FOREGROUND) is None
|
||||
assert manager.awake is None
|
||||
|
||||
|
||||
class TestTheWatchdogNeverConnects:
|
||||
"""A TCP connect is device traffic too — the loop must not even start."""
|
||||
|
||||
async def test_the_watchdog_does_not_build_a_poller_or_connect(self):
|
||||
ours = WillingPoller()
|
||||
built = []
|
||||
|
||||
def factory():
|
||||
built.append(1)
|
||||
return ours
|
||||
|
||||
manager = ShieldConnectionManager(
|
||||
hass=None,
|
||||
entry=_entry(adb_enabled=False),
|
||||
poller_factory=factory,
|
||||
)
|
||||
await manager.async_start()
|
||||
await asyncio.sleep(0.02)
|
||||
await manager.async_stop()
|
||||
|
||||
assert built == [], "the factory was never asked for a poller"
|
||||
assert ours.connect_calls == 0, "and nothing was connected"
|
||||
assert ours.traffic == []
|
||||
|
||||
async def test_a_direct_cycle_does_not_connect_either(self):
|
||||
poller = WillingPoller()
|
||||
poller.connected = False
|
||||
manager = _manager(poller, adb_enabled=False)
|
||||
|
||||
await manager._watchdog_cycle(5.0)
|
||||
|
||||
assert poller.connect_calls == 0
|
||||
assert poller.traffic == []
|
||||
|
||||
|
||||
class TestTheOptionIsACheckNotALatch:
|
||||
"""Turning ADB back on resumes normal work — no remembered decision."""
|
||||
|
||||
async def test_the_same_manager_resumes_when_the_option_flips(self):
|
||||
poller = WillingPoller()
|
||||
manager = _manager(poller, adb_enabled=False)
|
||||
|
||||
assert await manager.async_refresh_installed_apps(force=True) == []
|
||||
assert manager.app_label(PKG) is None
|
||||
assert await manager.send_media_command("play") is False
|
||||
assert poller.traffic == [], "nothing while it is off"
|
||||
|
||||
manager.entry.options[CONF_ADB_ENABLED] = True # the user turns ADB on
|
||||
|
||||
rows = await manager.async_refresh_installed_apps(force=True)
|
||||
assert rows, "the sweep really ran once ADB is on"
|
||||
assert manager.app_label("com.other.app") is None # queued, not sent yet
|
||||
assert manager.app_label_requests == {"com.other.app"}
|
||||
assert await manager.send_media_command("play") is True
|
||||
assert await manager.send_digit_key("5") is True
|
||||
assert manager.adb_poller is poller
|
||||
assert poller.shell_calls, "the device was reached again"
|
||||
assert poller.media_calls == ["play"]
|
||||
assert poller.digit_calls == ["5"]
|
||||
|
||||
async def test_the_intent_scan_resumes_too(self):
|
||||
poller = WillingPoller()
|
||||
manager = _manager(poller, adb_enabled=False)
|
||||
assert (await manager.async_scan_intents(PKG)).status == INTENT_SCAN_ADB_DISABLED
|
||||
|
||||
manager.entry.options[CONF_ADB_ENABLED] = True
|
||||
|
||||
result = await manager.async_scan_intents(PKG)
|
||||
assert result.status in (INTENT_SCAN_FRESH, INTENT_SCAN_FRESH_EMPTY)
|
||||
assert result.answered is True
|
||||
assert manager.intent_scan_state(PKG) is not None, "the answer is cached now"
|
||||
|
||||
|
||||
class TestTheOtherStatusesAreUntouched:
|
||||
"""ADB ON with no poller is 'unavailable', never 'disabled' (t21's split)."""
|
||||
|
||||
async def test_no_poller_reads_as_unavailable(self):
|
||||
manager = _manager(None, adb_enabled=True)
|
||||
|
||||
assert (await manager.async_scan_intents(PKG)).status == (
|
||||
INTENT_SCAN_ADB_UNAVAILABLE
|
||||
)
|
||||
assert await manager.async_refresh_installed_apps(force=True) == []
|
||||
assert manager.installed_apps_status == "unknown"
|
||||
assert await manager.send_media_command("play") is False
|
||||
|
||||
async def test_adb_off_and_no_poller_are_told_apart(self):
|
||||
disabled = _manager(WillingPoller(), adb_enabled=False)
|
||||
unbuilt = _manager(None, adb_enabled=True)
|
||||
|
||||
assert (await disabled.async_scan_intents(PKG)).status == (
|
||||
INTENT_SCAN_ADB_DISABLED
|
||||
)
|
||||
assert (await unbuilt.async_scan_intents(PKG)).status == (
|
||||
INTENT_SCAN_ADB_UNAVAILABLE
|
||||
)
|
||||
@@ -18,6 +18,7 @@ from entity_stublib import install, load_entity_modules
|
||||
from nsh.connection import ShieldConnectionManager
|
||||
from nsh.const import (
|
||||
APP_LABEL_EXIT_MARKER,
|
||||
CONF_ADB_ENABLED,
|
||||
CONF_APP_NAME_MAP,
|
||||
CONF_COMMAND_SCRIPTS,
|
||||
DEFAULT_APP_DIRECTORY,
|
||||
@@ -93,10 +94,21 @@ class FakeAdbPoller:
|
||||
return True
|
||||
|
||||
|
||||
def _adb_entry():
|
||||
"""A realistic entry: its own option says ADB is ON (t23/O1).
|
||||
|
||||
Production ALWAYS builds the manager with an entry; passing None modelled an
|
||||
ADB-off entry (fail closed), which the device paths now honour as the gate.
|
||||
"""
|
||||
return SimpleNamespace(
|
||||
options={CONF_ADB_ENABLED: True}, data={}, title="Shield", entry_id="e1"
|
||||
)
|
||||
|
||||
|
||||
def make_manager(store=None, poller=None, protocol=None):
|
||||
return ShieldConnectionManager(
|
||||
hass=None,
|
||||
entry=None,
|
||||
entry=_adb_entry(),
|
||||
poller=poller or FakeAdbPoller(),
|
||||
protocol=protocol or FakeProtocolClient(),
|
||||
store=store,
|
||||
@@ -115,7 +127,12 @@ def make_player(modules, manager=None, jellyfin_entity=None, options=None):
|
||||
mp_module = modules["media_player"]
|
||||
if manager is None:
|
||||
manager = make_manager()
|
||||
entry = SimpleNamespace(entry_id="e1", options={}, data={}, title="mdarcy")
|
||||
entry = SimpleNamespace(
|
||||
entry_id="e1",
|
||||
options={CONF_ADB_ENABLED: True},
|
||||
data={},
|
||||
title="mdarcy",
|
||||
)
|
||||
manager.entry = entry
|
||||
|
||||
def _set_option(key, value):
|
||||
@@ -2185,7 +2202,7 @@ class DownAdbPoller(FakeAdbPoller):
|
||||
def make_manager_with(*, poller=None, protocol=None):
|
||||
"""A manager with exactly these transports (None = not configured)."""
|
||||
return ShieldConnectionManager(
|
||||
hass=None, entry=None, poller=poller, protocol=protocol, store=None
|
||||
hass=None, entry=_adb_entry(), poller=poller, protocol=protocol, store=None
|
||||
)
|
||||
|
||||
|
||||
|
||||
@@ -19,6 +19,7 @@ from types import SimpleNamespace
|
||||
import entity_stublib
|
||||
import pytest
|
||||
from nsh.const import (
|
||||
CONF_ADB_ENABLED,
|
||||
CONF_COMMAND_SCRIPTS,
|
||||
CONF_REMOTE_COMMAND_SCRIPT,
|
||||
SCRIPT_VAR_APP_NAME,
|
||||
@@ -317,7 +318,8 @@ def test_the_remote_entry_options_are_the_media_player_options(modules):
|
||||
player._entry.options[CONF_COMMAND_SCRIPTS]
|
||||
== entity._entry.options[CONF_COMMAND_SCRIPTS]
|
||||
)
|
||||
assert make_manager().entry is None # the harness's manager starts unbound
|
||||
# the harness's manager carries a realistic entry whose option says ADB is on
|
||||
assert make_manager().entry.options[CONF_ADB_ENABLED] is True
|
||||
|
||||
|
||||
WATCH_SCRIPT = "script.shield_remote"
|
||||
|
||||
@@ -18,7 +18,12 @@ from nsh.connection import (
|
||||
ShieldConnectionManager,
|
||||
StateStore,
|
||||
)
|
||||
from nsh.const import ADB_POLL_INTERVAL, APP_LABEL_EXIT_MARKER, APP_LABEL_GREP_SHAPES
|
||||
from nsh.const import (
|
||||
ADB_POLL_INTERVAL,
|
||||
APP_LABEL_EXIT_MARKER,
|
||||
APP_LABEL_GREP_SHAPES,
|
||||
CONF_ADB_ENABLED,
|
||||
)
|
||||
|
||||
|
||||
def test_connection_module_depends_on_no_homeassistant():
|
||||
@@ -100,8 +105,28 @@ class FakePoller:
|
||||
return self.connected
|
||||
|
||||
|
||||
def _adb_entry(**options):
|
||||
"""A realistic entry: its own option says ADB is ON (t23/O1).
|
||||
|
||||
The manager is ALWAYS built with an entry in production, and an entry whose
|
||||
options do not say ``adb_enabled: True`` means ADB is switched off — the gate
|
||||
every device path now asks directly (``connection._adb_disabled``). A harness
|
||||
that passed ``entry=None`` was therefore modelling an ADB-off entry while
|
||||
expecting traffic to flow, which is exactly the combination the t22 review
|
||||
measured. Tests that WANT the disabled behaviour build their own entry.
|
||||
"""
|
||||
return types.SimpleNamespace(
|
||||
options={CONF_ADB_ENABLED: True, **options},
|
||||
data={},
|
||||
title="Shield",
|
||||
entry_id="e1",
|
||||
)
|
||||
|
||||
|
||||
def _manager(poller=None, **kwargs) -> ShieldConnectionManager:
|
||||
return ShieldConnectionManager(hass=None, entry=None, poller=poller, **kwargs)
|
||||
return ShieldConnectionManager(
|
||||
hass=None, entry=_adb_entry(), poller=poller, **kwargs
|
||||
)
|
||||
|
||||
|
||||
class TestStateStore:
|
||||
@@ -146,7 +171,7 @@ class TestAppLabelCache:
|
||||
"application-label-de: 'TiviMate DE'"
|
||||
)
|
||||
manager = ShieldConnectionManager(
|
||||
hass=None, entry=None, poller=poller, clock=(clock or (lambda: now))
|
||||
hass=None, entry=_adb_entry(), poller=poller, clock=(clock or (lambda: now))
|
||||
)
|
||||
return manager, now
|
||||
|
||||
@@ -332,9 +357,9 @@ class TestLogcatStream:
|
||||
poller = FakePoller()
|
||||
poller.awake = True
|
||||
manager = ShieldConnectionManager(
|
||||
hass=None, entry=None, poller=poller, clock=(clock or (lambda: now))
|
||||
hass=None, entry=_adb_entry(), poller=poller, clock=(clock or (lambda: now))
|
||||
)
|
||||
manager.entry = LEntry(options={"logcat_events": bool(lines)})
|
||||
manager.entry = LEntry(options={"logcat_events": bool(lines), CONF_ADB_ENABLED: True})
|
||||
lines = lines or []
|
||||
|
||||
async def _open():
|
||||
@@ -428,9 +453,9 @@ class TestLogcatStream:
|
||||
poller = FakePoller()
|
||||
poller.awake = True
|
||||
manager = ShieldConnectionManager(
|
||||
hass=None, entry=None, poller=poller, clock=lambda: 1000.0
|
||||
hass=None, entry=_adb_entry(), poller=poller, clock=lambda: 1000.0
|
||||
)
|
||||
manager.entry = LEntry(options={"logcat_events": True})
|
||||
manager.entry = LEntry(options={"logcat_events": True, CONF_ADB_ENABLED: True})
|
||||
|
||||
# the START record spans two stream lines (wrapped cmp=)
|
||||
lines = [
|
||||
@@ -833,7 +858,7 @@ class TestLifecycle:
|
||||
poller = FakePoller()
|
||||
poller.awake = True
|
||||
manager = ShieldConnectionManager(
|
||||
hass=None, entry=None, poller_factory=lambda: poller
|
||||
hass=None, entry=_adb_entry(), poller_factory=lambda: poller
|
||||
)
|
||||
await manager.async_start()
|
||||
await asyncio.sleep(0)
|
||||
|
||||
@@ -14,6 +14,7 @@ from __future__ import annotations
|
||||
import asyncio
|
||||
import sys
|
||||
from pathlib import Path
|
||||
from types import SimpleNamespace
|
||||
|
||||
sys.path.insert(0, str(Path(__file__).resolve().parent))
|
||||
import entity_stublib # noqa: E402
|
||||
@@ -40,6 +41,7 @@ from nsh.const import ( # noqa: E402
|
||||
APP_LABEL_EXIT_MARKER,
|
||||
APP_LABEL_GREP_SHAPES,
|
||||
APP_LIST_TTL,
|
||||
CONF_ADB_ENABLED,
|
||||
CONF_APP_NAME_MAP,
|
||||
)
|
||||
from nsh.ha_adb_common import ( # noqa: E402
|
||||
@@ -260,8 +262,20 @@ class _DiscoveryPoller:
|
||||
return None
|
||||
|
||||
|
||||
def _manager(poller):
|
||||
return ShieldConnectionManager(hass=None, entry=None, poller=poller)
|
||||
|
||||
def _entry(*, adb_enabled=True, **options):
|
||||
"""A realistic entry: the ADB option is ON unless a test says otherwise (t23)."""
|
||||
return SimpleNamespace(
|
||||
options={CONF_ADB_ENABLED: adb_enabled, **options},
|
||||
data={},
|
||||
title="Shield",
|
||||
entry_id="e1",
|
||||
)
|
||||
|
||||
def _manager(poller, *, adb_enabled=True):
|
||||
return ShieldConnectionManager(
|
||||
hass=None, entry=_entry(adb_enabled=adb_enabled), poller=poller
|
||||
)
|
||||
|
||||
|
||||
class TestPackageListParsing:
|
||||
@@ -485,7 +499,7 @@ class TestDiscoverySweep:
|
||||
now = [1000.0]
|
||||
poller = _DiscoveryPoller()
|
||||
manager = ShieldConnectionManager(
|
||||
hass=None, entry=None, poller=poller, clock=lambda: now[0]
|
||||
hass=None, entry=_entry(), poller=poller, clock=lambda: now[0]
|
||||
)
|
||||
await manager.async_refresh_installed_apps()
|
||||
assert manager.installed_apps()
|
||||
@@ -542,7 +556,7 @@ class TestDiscoveryDegradesGracefully:
|
||||
"""
|
||||
|
||||
async def test_no_poller_is_an_empty_list(self):
|
||||
manager = ShieldConnectionManager(hass=None, entry=None, poller=None)
|
||||
manager = ShieldConnectionManager(hass=None, entry=_entry(), poller=None)
|
||||
|
||||
rows = await manager.async_refresh_installed_apps()
|
||||
|
||||
@@ -567,7 +581,7 @@ class TestDiscoveryDegradesGracefully:
|
||||
now = [1000.0]
|
||||
poller = _DiscoveryPoller(failure=True)
|
||||
manager = ShieldConnectionManager(
|
||||
hass=None, entry=None, poller=poller, clock=lambda: now[0]
|
||||
hass=None, entry=_entry(), poller=poller, clock=lambda: now[0]
|
||||
)
|
||||
|
||||
rows = await manager.async_refresh_installed_apps()
|
||||
|
||||
@@ -12,6 +12,7 @@ against the real library's VolumeInfo TypedDict when the package is present
|
||||
import asyncio
|
||||
import sys
|
||||
import time
|
||||
from types import SimpleNamespace
|
||||
|
||||
import pytest
|
||||
from nsh.connection import (
|
||||
@@ -28,6 +29,7 @@ from nsh.connection import (
|
||||
StateStore,
|
||||
volume_tuple,
|
||||
)
|
||||
from nsh.const import CONF_ADB_ENABLED
|
||||
|
||||
|
||||
class InvalidAuth(Exception): # noqa: N818 - the library's own class name
|
||||
@@ -152,8 +154,21 @@ class FakeProtocolClient:
|
||||
self._reconnect_task.cancel()
|
||||
|
||||
|
||||
def _adb_entry():
|
||||
"""A realistic entry: its own option says ADB is ON (t23/O1).
|
||||
|
||||
The protocol tests exercise the ADB watchdog paths too (`_watchdog_cycle`),
|
||||
and a manager with no entry means ADB is off — the gate those paths now ask.
|
||||
"""
|
||||
return SimpleNamespace(
|
||||
options={CONF_ADB_ENABLED: True}, data={}, title="Shield", entry_id="e1"
|
||||
)
|
||||
|
||||
|
||||
def _protocol_manager(client=None, **kwargs) -> ShieldConnectionManager:
|
||||
return ShieldConnectionManager(hass=None, entry=None, protocol=client, **kwargs)
|
||||
return ShieldConnectionManager(
|
||||
hass=None, entry=_adb_entry(), protocol=client, **kwargs
|
||||
)
|
||||
|
||||
|
||||
async def _drain_loop():
|
||||
@@ -454,7 +469,7 @@ class TestLifecycle:
|
||||
monkeypatch.setattr(ShieldConnectionManager, "build_protocol_client", fake_builder)
|
||||
manager = ShieldConnectionManager(
|
||||
hass=None,
|
||||
entry=None,
|
||||
entry=_adb_entry(),
|
||||
protocol_settings={
|
||||
"host": "shield.local",
|
||||
"client_name": "Home Assistant",
|
||||
@@ -561,7 +576,7 @@ class TestAdbOutageKeepsProtocolTruth:
|
||||
poller.awake = None # wake poll fails → cycle failure path
|
||||
client = FakeProtocolClient()
|
||||
manager = ShieldConnectionManager(
|
||||
hass=None, entry=None, poller=poller, protocol=client
|
||||
hass=None, entry=_adb_entry(), poller=poller, protocol=client
|
||||
)
|
||||
await manager._start_protocol()
|
||||
|
||||
@@ -683,7 +698,9 @@ class TestAdbStaysThePrimarySource:
|
||||
poller = FakeAdbForOutage()
|
||||
poller.connected = True
|
||||
poller.awake = True
|
||||
manager = ShieldConnectionManager(hass=None, entry=None, poller=poller)
|
||||
manager = ShieldConnectionManager(
|
||||
hass=None, entry=_adb_entry(), poller=poller
|
||||
)
|
||||
await manager._start_protocol()
|
||||
|
||||
assert manager._protocol is None
|
||||
|
||||
Reference in new issue
Block a user