# Mobile Verification Toolkit (MVT)
# Copyright (c) 2021-2026 The MVT Authors.
# Use of this software is governed by the MVT License 1.1 that can be found at
#   https://license.mvt.re/1.1/
"""The dump's own installed-service count must survive into the artifact.

Most builds never print the `installed services: {…}` block; they state `installedServiceCount=N` in the user's `attributes:{…}` line and
list nothing. Dropping that number makes an artifact that says "no
accessibility services" about a dump that said there are five.
"""

from mvt.android.artifacts.dumpsys_accessibility import DumpsysAccessibilityArtifact
from mvt.common.alerts import AlertLevel

from ..utils import get_artifact

AOSP_NO_LIST = """\
ACCESSIBILITY MANAGER (dumpsys accessibility)
User state[
     attributes:{id=0, touchExplorationEnabled=false, installedServiceCount=5}
     Bound services:{}
     Enabled services:{}
     Binding services:{}
     Crashed services:{}
"""

ONE_UI_WITH_LIST = """\
ACCESSIBILITY MANAGER (dumpsys accessibility)
User state[attributes:{id=0, installedServiceCount=2}
  installed services: {
    0 : com.example.app/com.example.app.Service
    1 : com.other.app/.Helper
  }
  enabled services: {
  }
"""

TWO_USERS = """\
ACCESSIBILITY MANAGER (dumpsys accessibility)
User state[attributes:{id=0, installedServiceCount=1}
  installed services: {
    0 : com.example.app/com.example.app.Service
  }
User state[attributes:{id=95, installedServiceCount=3}
     Enabled services:{}
"""

PARTIAL_TWO_USERS = """\
ACCESSIBILITY MANAGER (dumpsys accessibility)
User state[attributes:{id=0, installedServiceCount=3}
     Enabled services:{{com.example.app/com.example.app.Service}}
User state[attributes:{id=10, installedServiceCount=1}
     Enabled services:{{com.other.app/.Helper}}
"""

ZERO_COUNT = """\
ACCESSIBILITY MANAGER (dumpsys accessibility)
User state[attributes:{id=0, installedServiceCount=0}
     Enabled services:{}
"""


def _parse(content):
    artifact = DumpsysAccessibilityArtifact()
    artifact.results = []
    artifact.parse(content)
    return artifact


class TestAccessibilityInstalledServiceCount:
    def test_stated_count_without_a_list_is_kept(self):
        artifact = _parse(AOSP_NO_LIST)
        assert len(artifact.results) == 1
        record = artifact.results[0]
        assert record["installed_service_count"] == 5
        assert record["component"] is None
        # Every state flag stays unknown: the dump named no service to which a
        # state could belong.
        assert record["installed"] is None
        assert record["enabled"] is None

    def test_a_stated_count_is_reported_as_low(self):
        artifact = _parse(AOSP_NO_LIST)
        artifact.check_indicators()
        alerts = artifact.alertstore.alerts
        assert len(alerts) == 1
        # A count is a coverage statement, not a running service: it must not
        # compete with a service the dump says is enabled.
        assert alerts[0].level == AlertLevel.LOW
        assert "does not list their component names" in alerts[0].message
        assert "5 installed" in alerts[0].message

    def test_a_listed_user_carries_the_count_on_each_service(self):
        artifact = _parse(ONE_UI_WITH_LIST)
        assert len(artifact.results) == 2
        assert {record["installed_service_count"] for record in artifact.results} == {2}
        assert all(record["component"] for record in artifact.results)

    def test_only_the_unlisted_user_gets_a_count_record(self):
        artifact = _parse(TWO_USERS)
        listed = [record for record in artifact.results if record["component"]]
        unlisted = [record for record in artifact.results if not record["component"]]
        assert [record["user_id"] for record in listed] == [0]
        assert [record["user_id"] for record in unlisted] == [95]
        assert unlisted[0]["installed_service_count"] == 3

    def test_a_zero_count_adds_nothing(self):
        # "Zero installed" is a negative result the empty section already
        # states; a record for it would be noise.
        assert _parse(ZERO_COUNT).results == []

    def test_a_partly_named_count_reports_the_unnamed_rest(self):
        # The Android 14 fixture states `installedServiceCount=2` and names one
        # enabled component. The listing is incomplete, and must not read as
        # complete.
        artifact = DumpsysAccessibilityArtifact()
        artifact.results = []
        with open(
            get_artifact("android_data/dumpsys_accessibility_v14_or_later.txt")
        ) as handle:
            artifact.parse(handle.read())
        unlisted = [record for record in artifact.results if not record["component"]]
        assert len(unlisted) == 1
        assert unlisted[0]["installed_service_count"] == 2
        assert unlisted[0]["unnamed_service_count"] == 1

        artifact.check_indicators()
        low = [
            alert
            for alert in artifact.alertstore.alerts
            if alert.level == AlertLevel.LOW
        ]
        assert len(low) == 1
        assert "only 1 of them (1 unnamed)" in low[0].message

    def test_the_unnamed_rest_is_counted_per_user(self):
        # User 0 names one of three, user 10 names its only one: the gap
        # belongs to user 0 alone.
        artifact = _parse(PARTIAL_TWO_USERS)
        unlisted = [record for record in artifact.results if not record["component"]]
        assert [
            (record["user_id"], record["unnamed_service_count"]) for record in unlisted
        ] == [(0, 2)]

    def test_a_fully_named_count_adds_nothing(self):
        artifact = _parse(ONE_UI_WITH_LIST)
        assert all(
            record["unnamed_service_count"] is None for record in artifact.results
        )