[FIX] N+1 Query in import_configs

hookwise/api.py

@@ -920,11 +920,18 @@ def restore_config() -> Any:
             return jsonify({"status": "error", "message": "No file"}), 400
         try:
             data = json.load(file)
+            # Pre-fetch all configs in data to avoid N+1 queries
+            config_ids = [c["id"] for c in data if "id" in c]
+            existing_configs = {
+                cfg.id: cfg for cfg in WebhookConfig.query.filter(WebhookConfig.id.in_(config_ids)).all()
+            }
+
             for c in data:
-                config = WebhookConfig.query.get(c["id"])
+                config = existing_configs.get(c["id"])
                 if not config:
                     config = WebhookConfig(id=c["id"])
                     db.session.add(config)
+                    existing_configs[c["id"]] = config
                 fields = [
                     "name",
                     "customer_id_default",

tests/conftest.py

@@ -1,4 +1,5 @@
 import os
+from unittest.mock import MagicMock, patch
 
 os.environ["SOCKETIO_ASYNC_MODE"] = "threading"
 os.environ["SECRET_KEY"] = "test-secret"
@@ -9,8 +10,22 @@
 
 import pytest
 
-
 @pytest.fixture(autouse=True, scope="session")
 def setup_test_env():
     # Already set at top level, but kept for clarity
     pass
+
+@pytest.fixture(autouse=True)
+def mock_redis():
+    with patch("hookwise.extensions.redis_client") as mock_ext_redis,          patch("hookwise.tasks.redis_client") as mock_tasks_redis,          patch("hookwise.api.redis_client") as mock_api_redis:
+
+        # Make redis_client.get return None by default to avoid maintenance mode
+        mock_ext_redis.get.return_value = None
+        mock_tasks_redis.get.return_value = None
+        mock_api_redis.get.return_value = None
+
+        yield {
+            "ext": mock_ext_redis,
+            "tasks": mock_tasks_redis,
+            "api": mock_api_redis
+        }

tests/test_restore_config.py

@@ -0,0 +1,72 @@
+import json
+import pytest
+from io import BytesIO
+from hookwise.extensions import db
+from hookwise.models import WebhookConfig
+from unittest.mock import patch
+
+@pytest.fixture
+def app():
+    from hookwise import create_app
+    app = create_app()
+    app.config["TESTING"] = True
+    app.config["WTF_CSRF_ENABLED"] = False
+    app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///:memory:"
+    # Ensure we have a secret key for session
+    app.config["SECRET_KEY"] = "test-secret"
+    return app
+
+@pytest.fixture
+def client(app):
+    with app.app_context():
+        db.create_all()
+        yield app.test_client()
+        db.session.remove()
+        db.drop_all()
+
+def test_restore_config_functionality(client, app):
+    with app.app_context():
+        # Create some existing configs with string IDs
+        id1 = "config-1-id"
+        id2 = "config-2-id"
+        id3 = "config-3-id"
+
+        c1 = WebhookConfig(id=id1, name="Config 1", board="Board 1")
+        c2 = WebhookConfig(id=id2, name="Config 2", board="Board 2")
+        db.session.add_all([c1, c2])
+        db.session.commit()
+
+        # Data to restore: 1 update, 1 same, 1 new
+        restore_data = [
+            {"id": id1, "name": "Config 1 Updated", "board": "Board 1 New"},
+            {"id": id2, "name": "Config 2", "board": "Board 2"},
+            {"id": id3, "name": "New Config 3", "board": "Board 3"}
+        ]
+
+        data = BytesIO(json.dumps(restore_data).encode("utf-8"))
+
+        # Bypass auth
+        with client.session_transaction() as sess:
+            sess["user_id"] = "admin"
+
+        response = client.post(
+            "/admin/restore",
+            data={"backup_file": (data, "backup.json")},
+            content_type="multipart/form-data"
+        )
+
+        assert response.status_code == 200
+        assert response.json["status"] == "success"
+
+        # Verify updates in DB
+        updated_c1 = WebhookConfig.query.get(id1)
+        assert updated_c1.name == "Config 1 Updated"
+        assert updated_c1.board == "Board 1 New"
+
+        updated_c2 = WebhookConfig.query.get(id2)
+        assert updated_c2.name == "Config 2"
+
+        new_c3 = WebhookConfig.query.get(id3)
+        assert new_c3 is not None
+        assert new_c3.name == "New Config 3"
+        assert new_c3.board == "Board 3"