seanbetts--steam-hardware-watch/tests/test_steamtracking_client_manifests.py

import csv
import os
import stat
import subprocess
import textwrap
import unittest
from pathlib import Path
from tempfile import TemporaryDirectory


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


def write_executable(path, content):
    path.write_text(textwrap.dedent(content).lstrip(), encoding="utf-8")
    path.chmod(path.stat().st_mode | stat.S_IXUSR)


def manifest(version, hardware_file="bins_hardware_linuxarm64.zip.old", steam_file="steam_linuxarm64.zip.old"):
    return textwrap.dedent(
        f"""
        "linuxarm64"
        {{
            "version"       "{version}"
            "bins_hardware_linuxarm64"
            {{
                "file"      "{hardware_file}"
                "size"      "7590487"
                "sha2"      "hardware-sha"
            }}
            "steam_linuxarm64"
            {{
                "file"      "{steam_file}"
                "size"      "4345869"
                "sha2"      "steam-sha"
            }}
            "runtime_steamrt_linuxarm64"
            {{
                "file"      "runtime_steamrt_linuxarm64.zip.same"
                "size"      "123973554"
                "sha2"      "runtime-sha"
            }}
        }}
        """
    ).lstrip()


def read_rows(path):
    return list(csv.DictReader(path.read_text(encoding="utf-8").splitlines(), delimiter="\t"))


class SteamTrackingClientManifestTests(unittest.TestCase):
    def test_manifest_report_flags_linuxarm64_beta_version_and_block_changes(self):
        with TemporaryDirectory() as tmp:
            tmp_path = Path(tmp)
            tracking = tmp_path / "tracking"
            manifests = tracking / "ClientManifest"
            manifests.mkdir(parents=True)
            (manifests / "steam_client_beta_linuxarm64").write_text(
                manifest(
                    "1779139477",
                    hardware_file="bins_hardware_linuxarm64.zip.new",
                    steam_file="steam_linuxarm64.zip.new",
                ),
                encoding="utf-8",
            )
            run_dir = tmp_path / "runs" / "2026-05-23"
            previous_reports = tmp_path / "runs" / "2026-05-22" / "reports"
            previous_reports.mkdir(parents=True)
            previous_reports.joinpath("steamtracking-client-manifests.tsv").write_text(
                "\n".join(
                    [
                        "manifest\tarch\tversion\tblock\tfile\tsize\tsha2\tchanged_since_previous\tprevious_version\tprevious_file\tprevious_size\tprevious_sha2",
                        "steam_client_beta_linuxarm64\tlinuxarm64\t1779000000\t__manifest__\t\t\t\tno\t\t\t\t",
                        "steam_client_beta_linuxarm64\tlinuxarm64\t1779000000\tbins_hardware_linuxarm64\tbins_hardware_linuxarm64.zip.old\t7590487\thardware-sha\tno\t\t\t\t",
                        "steam_client_beta_linuxarm64\tlinuxarm64\t1779000000\tsteam_linuxarm64\tsteam_linuxarm64.zip.old\t4345869\tsteam-sha\tno\t\t\t\t",
                    ]
                )
                + "\n",
                encoding="utf-8",
            )

            result = subprocess.run(
                [
                    "python3",
                    str(ROOT / "scripts" / "parse_steamtracking_client_manifests.py"),
                    "--run-dir",
                    str(run_dir),
                    "--tracking-dir",
                    str(tracking),
                ],
                cwd=ROOT,
                text=True,
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                check=False,
            )

            self.assertEqual("", result.stderr)
            self.assertEqual(0, result.returncode)
            rows = read_rows(run_dir / "reports" / "steamtracking-client-manifests.tsv")
            manifest_row = next(
                row for row in rows if row["manifest"] == "steam_client_beta_linuxarm64" and row["block"] == "__manifest__"
            )
            hardware_row = next(
                row for row in rows if row["manifest"] == "steam_client_beta_linuxarm64" and row["block"] == "bins_hardware_linuxarm64"
            )
            self.assertEqual("1779139477", manifest_row["version"])
            self.assertEqual("yes", manifest_row["changed_since_previous"])
            self.assertEqual("yes", hardware_row["changed_since_previous"])
            self.assertEqual("bins_hardware_linuxarm64.zip.old", hardware_row["previous_file"])

            key_lines = (run_dir / "reports" / "steamtracking-client-manifests-key-lines.txt").read_text(
                encoding="utf-8"
            )
            self.assertIn("steam_client_beta_linuxarm64", key_lines)
            self.assertIn("bins_hardware_linuxarm64", key_lines)
            self.assertIn("changed=yes", key_lines)

    def test_check_steamtracking_writes_client_manifest_report(self):
        with TemporaryDirectory() as tmp:
            tmp_path = Path(tmp)
            tracking = tmp_path / "tracking"
            manifests = tracking / "ClientManifest"
            manifests.mkdir(parents=True)
            (manifests / "steam_client_beta_linuxarm64").write_text(
                manifest("1779139477", hardware_file="bins_hardware_linuxarm64.zip.new"),
                encoding="utf-8",
            )
            (tracking / "ProtobufsWebui").mkdir()
            (tracking / "ProtobufsWebui" / "service_steaminputmanager.proto").write_text(
                "message CSteamInputService_ShouldTritonPairInOobe_Request {}\n",
                encoding="utf-8",
            )
            run_dir = tmp_path / "runs" / "2026-05-23"

            result = subprocess.run(
                [str(ROOT / "scripts" / "check_steamtracking.sh"), str(run_dir), str(tracking)],
                cwd=ROOT,
                text=True,
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                check=False,
            )

            self.assertEqual("", result.stderr)
            self.assertEqual(0, result.returncode)
            self.assertTrue((run_dir / "reports" / "steamtracking-client-manifests.tsv").exists())
            key_lines = (run_dir / "reports" / "steamtracking-client-manifests-key-lines.txt").read_text(
                encoding="utf-8"
            )
            self.assertIn("steam_client_beta_linuxarm64", key_lines)

    def test_compare_runs_includes_client_manifest_key_line_deltas(self):
        with TemporaryDirectory() as tmp:
            tmp_path = Path(tmp)
            previous = tmp_path / "2026-05-22"
            current = tmp_path / "2026-05-23"
            (previous / "reports").mkdir(parents=True)
            (current / "reports").mkdir(parents=True)
            (previous / "reports" / "steamtracking-client-manifests-key-lines.txt").write_text(
                "steam_client_beta_linuxarm64 version=1779000000 arch=linuxarm64 changed=no previous=\n",
                encoding="utf-8",
            )
            (current / "reports" / "steamtracking-client-manifests-key-lines.txt").write_text(
                "steam_client_beta_linuxarm64 version=1779139477 arch=linuxarm64 changed=yes previous=1779000000\n",
                encoding="utf-8",
            )
            output = tmp_path / "compare.md"

            result = subprocess.run(
                [
                    "python3",
                    str(ROOT / "scripts" / "compare_runs.py"),
                    "--previous",
                    str(previous),
                    "--current",
                    str(current),
                    "--output",
                    str(output),
                ],
                cwd=ROOT,
                text=True,
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                check=False,
            )

            self.assertEqual("", result.stderr)
            self.assertEqual(0, result.returncode)
            compare = output.read_text(encoding="utf-8")
            self.assertIn("Steam client manifests added line", compare)
            self.assertIn("1779139477", compare)


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