Bug fixes for uploading the files. I though the changes being made where not supposed to break legacy endpoints. Not sure what happened. Either way things are almost back to normal.

tests/unit/test_unit_hosted_file_resolver.py

@@ -0,0 +1,71 @@
+import sys
+import os
+from unittest.mock import MagicMock, patch
+
+# Add project root to path
+sys.path.append(os.getcwd())
+
+# Mock dependencies
+mock_config = MagicMock()
+mock_config.settings = MagicMock()
+sys.modules["app.config"] = mock_config
+sys.modules["redis"] = MagicMock()
+sys.modules["app.log"] = MagicMock()
+sys.modules["app.lib_general"] = MagicMock()
+
+# Valid random IDs are typically 11 or 22 chars
+VALID_RAND_ID = "Q8lR8Ai8hx2FjbQ3C_EH1Q"
+
+from app.lib_redis_helpers import lookup_id_random_pop
+
+def test_hosted_file_resolver_logic():
+    print("--- Testing ID Resolver logic (lookup_id_random_pop) ---")
+    
+    # Correctly patch the function where it is DEFINED
+    with patch('app.lib_redis_helpers.redis_lookup_id_random', side_effect=lambda record_id_random, table_name: 123 if record_id_random == VALID_RAND_ID else 999):
+        
+        # 1. Test Vision-style payload (account_id is a string)
+        payload = {
+            "account_id": VALID_RAND_ID,
+            "filename": "test.txt"
+        }
+        
+        result = lookup_id_random_pop(payload.copy())
+        print(f"Vision Payload Result: {result}")
+        
+        # Verify it resolved the string to an integer
+        assert result.get("account_id") == 123
+        
+        # 2. Test Legacy-style payload (account_id_random is a string)
+        payload_legacy = {
+            "account_id_random": VALID_RAND_ID,
+            "filename": "test.txt"
+        }
+        
+        result_legacy = lookup_id_random_pop(payload_legacy.copy())
+        print(f"Legacy Payload Result: {result_legacy}")
+        
+        # Verify it resolved and popped the random key
+        assert result_legacy.get("account_id") == 123
+        assert "account_id_random" not in result_legacy
+        
+        # 3. Test mixed/polymorphic
+        payload_poly = {
+            "link_to_type": "archive_content",
+            "link_to_id": VALID_RAND_ID
+        }
+        result_poly = lookup_id_random_pop(payload_poly.copy())
+        print(f"Polymorphic Payload Result: {result_poly}")
+        assert result_poly.get("link_to_id") == 123
+
+    print("✅ ID Resolver logic verified.")
+
+if __name__ == "__main__":
+    try:
+        test_hosted_file_resolver_logic()
+        print("\n🎉 RESOLVER LOGIC TEST PASSED!")
+    except Exception as e:
+        print(f"\n❌ TEST FAILED: {e}")
+        import traceback
+        traceback.print_exc()
+        sys.exit(1)