mirror of
https://github.com/boykopovar/AnyPS5.git
synced 2026-10-22 16:43:05 -05:00
88 lines
4.5 KiB
Python
88 lines
4.5 KiB
Python
import json
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
PROGRESS = Path(__file__).resolve().parents[1] / "progress.py"
|
|
|
|
|
|
def snapshot(implemented, library="libSceExample"):
|
|
return {
|
|
"libraries": {
|
|
"done": int(implemented), "total": 1, "percent": 100 if implemented else 0,
|
|
"groups": [{"name": library, "done": int(implemented), "todo": int(not implemented),
|
|
"done_names": ["Ready"] if implemented else [],
|
|
"todo_names": [] if implemented else ["Ready"]}],
|
|
},
|
|
"shaders": {"done": 0, "total": 0, "percent": 0, "groups": []},
|
|
}
|
|
|
|
|
|
class ProgressEncodingTests(unittest.TestCase):
|
|
def compare(self, base, head, encoding="cp1252"):
|
|
with tempfile.TemporaryDirectory(prefix="progress encoding ") as directory:
|
|
paths = [Path(directory) / name for name in ("base.json", "head.json")]
|
|
for path, data in zip(paths, (base, head)):
|
|
path.write_text(json.dumps(data, ensure_ascii=False), encoding="utf-8")
|
|
env = {**os.environ, "PYTHONIOENCODING": encoding, "PYTHONUTF8": "0"}
|
|
result = subprocess.run([sys.executable, "-X", "utf8=0", str(PROGRESS), "--compare", *map(str, paths)],
|
|
env=env, capture_output=True, timeout=60)
|
|
self.assertEqual(result.returncode, 0, result.stderr.decode("utf-8", errors="replace"))
|
|
return result.stdout.decode("utf-8").replace("\r\n", "\n")
|
|
|
|
def test_implemented_report_with_cp1252_stdout(self):
|
|
report = self.compare(snapshot(False), snapshot(True))
|
|
self.assertIn("\U0001f4c8 **System libraries**: 100%", report)
|
|
self.assertIn("\u2705 1 implemented", report)
|
|
self.assertIn("| libSceExample | `Ready` |", report)
|
|
|
|
def test_regression_report_with_cp1252_stdout(self):
|
|
report = self.compare(snapshot(True), snapshot(False))
|
|
self.assertIn("\U0001f4c9 **System libraries**: 0%", report)
|
|
self.assertIn("\u26a0\ufe0f 1 went back to stubs", report)
|
|
self.assertIn("-1 reverted", report)
|
|
|
|
def test_utf8_snapshot_names_are_preserved(self):
|
|
library = "libSce\u6e2c\u8a66\U0001f3ae"
|
|
report = self.compare(snapshot(False, library), snapshot(True, library))
|
|
self.assertIn(f"| {library} | `Ready` |", report)
|
|
|
|
def test_unchanged_snapshots_produce_no_output(self):
|
|
self.assertEqual(self.compare(snapshot(False), snapshot(False)), "")
|
|
|
|
def test_utf8_stdout_report_is_unchanged(self):
|
|
base, head = snapshot(False), snapshot(True)
|
|
self.assertEqual(self.compare(base, head, "utf-8"), self.compare(base, head))
|
|
|
|
def test_site_generation_preserves_unicode_with_cp1252_stdout(self):
|
|
library = "libSce\u6e2c\u8a66\U0001f3ae"
|
|
with tempfile.TemporaryDirectory(prefix="progress site encoding ") as directory:
|
|
root = Path(directory) / "source"
|
|
module = root / "core" / "libs" / "prx" / library
|
|
module.mkdir(parents=True)
|
|
(module / "Export.cpp").write_text("int APS5_VABI Ready() { return 0; }\n", encoding="utf-8")
|
|
opcodes = root / "core" / "shader" / "recompiler" / "RdnaDecoder" / "include" / "RdnaDecoder" / "RdnaOpcode.hpp"
|
|
opcodes.parent.mkdir(parents=True)
|
|
opcodes.write_text("enum class RdnaOpcode {\n SEndpgm,\n};\n", encoding="utf-8")
|
|
output = Path(directory) / "site"
|
|
env = {**os.environ, "PYTHONIOENCODING": "cp1252", "PYTHONUTF8": "0"}
|
|
result = subprocess.run(
|
|
[sys.executable, "-X", "utf8=0", str(PROGRESS), str(output), "--root", str(root),
|
|
"--module", library, "--json"], env=env, capture_output=True, timeout=30)
|
|
self.assertEqual(result.returncode, 0, result.stderr.decode("utf-8", errors="replace"))
|
|
generated = {name: (output / name).read_text(encoding="utf-8") for name in (
|
|
"progress.json", "badge-libraries.svg", "badge-shaders.svg", "progress.svg", "progress.html")}
|
|
payload = json.loads(generated["progress.json"])
|
|
self.assertEqual(payload, json.loads(result.stdout.decode("utf-8")))
|
|
self.assertEqual((payload["libraries"]["done"], payload["libraries"]["total"]), (1, 1))
|
|
self.assertEqual(payload["libraries"]["groups"][0]["name"], library)
|
|
for name in ("progress.svg", "progress.html"):
|
|
self.assertIn(library.removeprefix("libSce"), generated[name])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|