From 6d1525748ddc8aead2526b2e3532e03f8ad10da2 Mon Sep 17 00:00:00 2001
From: Shroominic <db2002dominic@gmail.com>
Date: Tue, 9 Sep 2025 18:27:48 +0100
Subject: [PATCH] fix tests

---
 routstr/payment/helpers.py    | 42 +++++++++++++++++++++++------------
 tests/integration/conftest.py |  8 +++++++
 2 files changed, 36 insertions(+), 14 deletions(-)

diff --git a/routstr/payment/helpers.py b/routstr/payment/helpers.py
index 7d15e2bf..33116119 100644
--- a/routstr/payment/helpers.py
+++ b/routstr/payment/helpers.py
@@ -139,22 +139,37 @@ def get_max_cost_for_model(model: str) -> int:
 
 def calculate_discounted_max_cost(max_cost_for_model: int, body: dict) -> int:
     """Calculate the discounted max cost for a request."""
+    original_max_cost_msats = max_cost_for_model
+    model = body.get("model", "unknown")
+
     if settings.fixed_pricing:
         return max_cost_for_model
 
-    print(body)
-    model_pricing = get_model_cost_info(body.get("model"))
-    print("max_cost_for_model (msats)", max_cost_for_model)
-    print("model_pricing.max_cost", model_pricing.max_cost)
-    print("model_pricing.max_cost (msats)", model_pricing.max_cost * 1000)
-    print("model_pricing.max_prompt_cost", model_pricing.max_prompt_cost)
-    print("model_pricing.max_completion_cost", model_pricing.max_completion_cost)
+    if not (model_pricing := get_model_cost_info(model)):
+        return max_cost_for_model
 
     tol = settings.tolerance_percentage
     tol_factor = max(0.0, 1 - float(tol) / 100.0)
     max_prompt_allowed_sats = model_pricing.max_prompt_cost * tol_factor
     max_completion_allowed_sats = model_pricing.max_completion_cost * tol_factor
 
+    logger.debug(
+        "Discount estimation context",
+        extra={
+            "model": model,
+            "tolerance_pct": tol,
+            "tol_factor": tol_factor,
+            "start_max_cost_msats": original_max_cost_msats,
+            "model_max_cost_sats": model_pricing.max_cost,
+            "model_max_prompt_cost_sats": model_pricing.max_prompt_cost,
+            "model_max_completion_cost_sats": model_pricing.max_completion_cost,
+            "input_rate_sats_per_token": model_pricing.prompt,
+            "output_rate_sats_per_token": model_pricing.completion,
+            "max_prompt_allowed_sats": max_prompt_allowed_sats,
+            "max_completion_allowed_sats": max_completion_allowed_sats,
+        },
+    )
+
     if messages := body.get("messages"):
         prompt_tokens = estimate_tokens(messages)
         estimated_prompt_delta_sats = (
@@ -191,13 +206,12 @@ def estimate_tokens(messages: list) -> int:
     return len(str(messages)) // 3
 
 
-def get_model_cost_info(model_id: str | None) -> Pricing:
-    if model_id is None:
-        raise HTTPException(
-            status_code=400,
-            detail=f"Model {model_id} not found",
-        )
-    return next(m for m in MODELS if m.id == model_id).sats_pricing  # type: ignore
+def get_model_cost_info(model_id: str) -> Pricing | None:
+    if not model_id or model_id == "unknown":
+        return None
+
+    model = next((m for m in MODELS if m.id == model_id), None)
+    return model.sats_pricing if model else None  # type: ignore
 
 
 def create_error_response(
diff --git a/tests/integration/conftest.py b/tests/integration/conftest.py
index 7f58c24f..311214f6 100644
--- a/tests/integration/conftest.py
+++ b/tests/integration/conftest.py
@@ -509,6 +509,10 @@ async def integration_app(
         mint_url = os.environ.get("CASHU_MINTS", "http://localhost:3338")
         from routstr.core.settings import settings as _settings
 
+        # Passthrough discounted max cost to avoid dependence on MODELS in tests
+        def _passthrough_discount(max_cost_for_model: int, body: dict) -> int:
+            return max_cost_for_model
+
         with (
             patch("routstr.core.db.engine", integration_engine),
             patch.object(_settings, "cashu_mints", [mint_url]),
@@ -522,6 +526,10 @@ async def integration_app(
             patch("websockets.connect") as mock_websockets,
             patch("routstr.payment.price.btc_usd_ask_price", return_value=50000.0),
             patch("routstr.payment.price.sats_usd_ask_price", return_value=0.0005),
+            patch(
+                "routstr.payment.helpers.calculate_discounted_max_cost",
+                side_effect=_passthrough_discount,
+            ),
         ):
             # Configure the WebSocket mock for discovery service - fast failure for performance tests
             async def mock_websocket_connect(*args: Any, **kwargs: Any) -> None: