From 7c7ff0d146e8f41fc84c492259b05149d8e64170 Mon Sep 17 00:00:00 2001
From: Dusk-92 <romain.thomas92@proton.me>
Date: Thu, 3 Sep 2026 23:06:41 +0200
Subject: [PATCH] Test exact QuickLoot restoration

---
 tests/test_safety.py | 31 +++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/tests/test_safety.py b/tests/test_safety.py
index b8d3112..edb4672 100644
--- a/tests/test_safety.py
+++ b/tests/test_safety.py
@@ -817,8 +817,8 @@ class SmartUpdateTests(unittest.TestCase):
         size = 0x435D3C + 16
         data = bytearray(size)
 
-        data[0x0C1ECF] = 0x75
-        data[0x0C2B25] = 0x75
+        data[0x0C1ECF:0x0C1ED1] = b"\x75\x10"
+        data[0x0C2B25:0x0C2B27] = b"\x75\x0B"
         data[0x3A4869] = 0x27
         data[0x4089B4:0x4089B8] = struct.pack("<f", 1.919862)
         data[0x435D38:0x435D3C] = b"64\x00\x00"
@@ -834,8 +834,8 @@ class SmartUpdateTests(unittest.TestCase):
             with open(path, "rb") as handle:
                 result = handle.read()
 
-        self.assertEqual(result[0x0C1ECF], 0x74)
-        self.assertEqual(result[0x0C2B25], 0x74)
+        self.assertEqual(result[0x0C1ECF:0x0C1ED1], b"\x74\x10")
+        self.assertEqual(result[0x0C2B25:0x0C2B27], b"\x74\x0B")
         self.assertEqual(result[0x3A4869], 0x14)
         self.assertAlmostEqual(
             struct.unpack("<f", result[0x4089B4:0x4089B8])[0],
@@ -848,6 +848,29 @@ class SmartUpdateTests(unittest.TestCase):
             struct.pack("<f", 3000.0),
         )
 
+    def test_superwow_reset_handles_known_always_autoloot_variant(self):
+        tool = WowSetupTool.__new__(WowSetupTool)
+        size = 0x435D3C + 16
+        data = bytearray(size)
+        data[0x0C1ECF:0x0C1ED1] = b"\x90\x90"
+        data[0x0C2B25:0x0C2B27] = b"\x90\x90"
+        data[0x3A4869] = 0x14
+        data[0x4089B4:0x4089B8] = struct.pack("<f", 1.5708)
+        data[0x435D38:0x435D3C] = b"12\x00\x00"
+
+        with tempfile.TemporaryDirectory() as root:
+            path = os.path.join(root, "WoW_Modernized.exe")
+            with open(path, "wb") as handle:
+                handle.write(data)
+
+            tool._reset_superwow_managed_exe_patches(path)
+
+            with open(path, "rb") as handle:
+                result = handle.read()
+
+        self.assertEqual(result[0x0C1ECF:0x0C1ED1], b"\x74\x10")
+        self.assertEqual(result[0x0C2B25:0x0C2B27], b"\x74\x0B")
+
     def test_superwow_signature_versions_patch_reset_policy(self):
         tool = WowSetupTool.__new__(WowSetupTool)
         tool.core_plugins = {"SuperWoWhook.dll": FakeVar(True)}