From 7bfc5ba23492f506e033b964c929d86cd4bc54c6 Mon Sep 17 00:00:00 2001
From: ppoelzl <33464174+ppoelzl@users.noreply.github.com>
Date: Sat, 26 Feb 2022 14:55:14 +0100
Subject: [PATCH] test: add ascendancy fixer script

---
 tests/test_fix_ascendancy_positions.py | 29 ++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 tests/test_fix_ascendancy_positions.py

diff --git a/tests/test_fix_ascendancy_positions.py b/tests/test_fix_ascendancy_positions.py
new file mode 100644
index 00000000..690dc8df
--- /dev/null
+++ b/tests/test_fix_ascendancy_positions.py
@@ -0,0 +1,29 @@
+import filecmp
+import pathlib
+import shutil
+import tempfile
+
+import fix_ascendancy_positions
+
+
+def test_fix_one() -> None:
+    raw = pathlib.Path("data.json")
+    fixed = pathlib.Path("data_fixed.json")
+    with tempfile.TemporaryDirectory() as td:
+        new = pathlib.Path(td, raw)
+        shutil.copy(raw, td)
+        fix_ascendancy_positions.fix_ascendancy_positions(new)
+        assert filecmp.cmp(new, fixed, shallow=False)
+
+
+def test_fix_all() -> None:
+    raw = pathlib.Path("data.json")
+    fixed = pathlib.Path("data_fixed.json")
+    with tempfile.TemporaryDirectory() as outer, tempfile.TemporaryDirectory(
+        dir=outer
+    ) as inner:
+        root = pathlib.Path(outer)
+        new = pathlib.Path(inner, raw)
+        shutil.copy(raw, inner)
+        fix_ascendancy_positions.main(root)
+        assert filecmp.cmp(new, fixed, shallow=False)