Initial commit: SHARED library with LFS for binary assets

tests/test_spectra_replay_normalize.py

@@ -0,0 +1,76 @@
+from __future__ import annotations
+
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+from spectra_replay_normalize import normalize_spectra_replay, strip_model_prefix
+
+
+def test_expands_v3_entity_delta_columns_to_legacy_path() -> None:
+    replay = {
+        "entities": [
+            {
+                "uid": "1",
+                "unit": "tankModels/ussr_kv_220",
+                "t": [1000, 250, 250],
+                "x": [1600, 16, -32],
+                "y": [64, 0, 16],
+                "z": [-3200, 160, 0],
+            }
+        ]
+    }
+
+    normalized = normalize_spectra_replay(replay)
+
+    entity = normalized["entities"][0]
+    assert entity["unit"] == "ussr_kv_220"
+    assert entity["model_path"] == "tankModels/ussr_kv_220"
+    assert entity["path"] == [
+        [1000, 100.0, 4.0, -200.0],
+        [1250, 101.0, 4.0, -190.0],
+        [1500, 99.0, 5.0, -190.0],
+    ]
+
+
+def test_expands_v3_ticket_delta_columns_to_legacy_rows() -> None:
+    replay = {"tickets": {"1": [[0, 1000, 500], [16000, -20, -30]]}}
+
+    normalized = normalize_spectra_replay(replay)
+
+    assert normalized["tickets"]["1"] == [
+        [0, 16000],
+        [1000, 15980],
+        [1500, 15950],
+    ]
+
+
+def test_strips_event_unit_model_prefixes_but_preserves_model_path() -> None:
+    replay = {
+        "events": {
+            "kills": [
+                {
+                    "offender_unit": "airModels/spitfire_ix_usa",
+                    "offended_unit": "tankModels/ussr_kv_220",
+                    "offender_pos": [1600, 320, -800],
+                    "offended_pos": [3200, 64, -1600],
+                }
+            ]
+        },
+        "players": {
+            "1": {"units": [{"unit": "tankModels/ussr_kv_220"}]},
+        },
+    }
+
+    normalized = normalize_spectra_replay(replay)
+    kill = normalized["events"]["kills"][0]
+
+    assert strip_model_prefix("tankModels/ussr_kv_220") == "ussr_kv_220"
+    assert kill["offender_unit"] == "spitfire_ix_usa"
+    assert kill["offender_unit_model_path"] == "airModels/spitfire_ix_usa"
+    assert kill["offended_unit"] == "ussr_kv_220"
+    assert kill["offended_unit_model_path"] == "tankModels/ussr_kv_220"
+    assert kill["offender_pos"] == [100.0, 20.0, -50.0]
+    assert kill["offended_pos"] == [200.0, 4.0, -100.0]
+    assert normalized["players"]["1"]["units"][0]["unit"] == "ussr_kv_220"

tests/test_spectra_ws_payload.py

@@ -0,0 +1,42 @@
+from __future__ import annotations
+
+import json
+import sys
+import unittest
+from pathlib import Path
+
+import msgpack
+import zstandard as zstd
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+from spectra_ws_payload import SpectraPayloadError, decode_spectra_ws_payload
+
+
+def _zstd(payload: bytes) -> bytes:
+    return zstd.ZstdCompressor().compress(payload)
+
+
+class SpectraWsPayloadTests(unittest.TestCase):
+    def test_decode_zstd_json_object(self) -> None:
+        replay = {"completed": [{"_id": 123, "players": {"1": {"nick": "A"}}}]}
+
+        self.assertEqual(decode_spectra_ws_payload(_zstd(json.dumps(replay).encode("utf-8"))), replay)
+
+    def test_decode_zstd_msgpack_object(self) -> None:
+        replay = {"completed": [{"_id": 123, "players": {"1": {"nick": "A"}}}]}
+
+        self.assertEqual(decode_spectra_ws_payload(_zstd(msgpack.packb(replay, use_bin_type=True))), replay)
+
+    def test_decode_plain_json_text_frame(self) -> None:
+        replay = {"_id": "abc", "teams": []}
+
+        self.assertEqual(decode_spectra_ws_payload(json.dumps(replay)), replay)
+
+    def test_unknown_payload_raises_clear_error(self) -> None:
+        with self.assertRaisesRegex(SpectraPayloadError, "neither JSON nor msgpack"):
+            decode_spectra_ws_payload(_zstd(b"not-json-or-msgpack"))
+
+
+if __name__ == "__main__":
+    unittest.main()