fix: wire subscription service for yookassa hwid payments

2026-05-27 18:13:22 +03:00
parent 25056602d8
commit 1c9e55d797
4 changed files with 107 additions and 0 deletions
@@ -94,3 +94,105 @@ class WebAppDeviceTopupOptionsTests(IsolatedAsyncioTestCase):
        self.assertEqual(payload["extra_hwid_devices_valid_until_text"], "02.01.2099 03:04")
        self.assertEqual(payload["plans"][0]["valid_from"], valid_from.isoformat())
        self.assertEqual(payload["plans"][0]["valid_until"], active_until.isoformat())
+
+    async def test_create_payment_route_quotes_hwid_with_app_subscription_service(self):
+        tariff = SimpleNamespace(
+            key="standard",
+            billing_model="period",
+            enabled_periods=[1],
+            hwid_device_packages=SimpleNamespace(
+                rub=[SimpleNamespace(count=1)],
+                stars=[],
+            ),
+        )
+        settings = SimpleNamespace(
+            traffic_sale_mode=False,
+            tariffs_config=SimpleNamespace(require=lambda key: tariff),
+            DEFAULT_LANGUAGE="en",
+            DEFAULT_CURRENCY_SYMBOL="RUB",
+            ADMIN_IDS=[],
+        )
+        quote = {
+            "price": 25,
+            "valid_from": datetime(2099, 1, 1, tzinfo=timezone.utc),
+            "valid_until": datetime(2099, 4, 1, tzinfo=timezone.utc),
+            "pricing_period_months": 3,
+            "proration_ratio": 1.0,
+            "full_price": 25,
+        }
+        subscription_service = SimpleNamespace(
+            quote_hwid_device_topup=AsyncMock(return_value=quote)
+        )
+        request = SimpleNamespace(
+            app={
+                "settings": settings,
+                "async_session_factory": _SessionFactory(),
+                "subscription_service": subscription_service,
+            }
+        )
+        db_user = SimpleNamespace(
+            is_banned=False,
+            panel_user_uuid="panel-user",
+            language_code="en",
+            telegram_id=42,
+        )
+        sub = SimpleNamespace(tariff_key="standard")
+
+        async def _fake_create_payment(**kwargs):
+            return billing_module.web.json_response(
+                {
+                    "ok": True,
+                    "price": kwargs["price"],
+                    "hwid_valid_until": kwargs["hwid_quote"]["valid_until"].isoformat(),
+                }
+            )
+
+        with (
+            patch.object(billing_module, "_require_user_id", return_value=42),
+            patch.object(
+                billing_module,
+                "_enforce_webapp_rate_limit",
+                AsyncMock(return_value=None),
+            ),
+            patch.object(
+                billing_module,
+                "_read_json",
+                AsyncMock(
+                    return_value={
+                        "method": "yookassa",
+                        "months": 1,
+                        "device_count": 1,
+                        "tariff_key": "standard",
+                        "sale_mode": "hwid_devices",
+                    }
+                ),
+            ),
+            patch.object(
+                billing_module,
+                "_get_cached_webapp_settings",
+                return_value={"subscription_options": {}, "stars_subscription_options": {}},
+            ),
+            patch.object(
+                billing_module.user_dal,
+                "get_user_by_id",
+                AsyncMock(return_value=db_user),
+            ),
+            patch.object(
+                billing_module.subscription_dal,
+                "get_active_subscription_by_user_id",
+                AsyncMock(return_value=sub),
+            ),
+            patch.object(
+                billing_module,
+                "_create_subscription_payment",
+                AsyncMock(side_effect=_fake_create_payment),
+            ) as create_payment,
+        ):
+            response = await billing_module.create_payment_route(request)
+
+        self.assertEqual(response.status, 200)
+        payload = json.loads(response.text)
+        self.assertTrue(payload["ok"])
+        self.assertEqual(payload["price"], 25)
+        subscription_service.quote_hwid_device_topup.assert_awaited_once()
+        create_payment.assert_awaited_once()