diff --git a/nwc_backend/event_handlers/__tests__/get_budget_test.py b/nwc_backend/event_handlers/__tests__/get_budget_test.py index 4f67058..18a8780 100644 --- a/nwc_backend/event_handlers/__tests__/get_budget_test.py +++ b/nwc_backend/event_handlers/__tests__/get_budget_test.py @@ -67,6 +67,7 @@ async def test_get_budget_success__spending_limit_SAT_enabled( "total_budget": 1000000, "used_budget": 0, "renews_at": round(spending_cycle.end_time.timestamp()), + "renewal_period": "monthly", } @@ -107,6 +108,7 @@ async def test_pay_invoice_success__spending_limit_USD_enabled( "total_budget": estimated_budget_currency_amount * 1000, "used_budget": 0, "renews_at": ANY, + "renewal_period": "monthly", "currency": { "code": "USD", "total_budget": total_budget_currency_amount, diff --git a/nwc_backend/event_handlers/get_budget_handler.py b/nwc_backend/event_handlers/get_budget_handler.py index ae23ccd..1115194 100644 --- a/nwc_backend/event_handlers/get_budget_handler.py +++ b/nwc_backend/event_handlers/get_budget_handler.py @@ -27,6 +27,7 @@ async def get_budget(access_token: str, request: Nip47Request) -> Nip47BudgetRes total_budget=current_spending_limit.amount * 1000, used_budget=used_budget_sats * 1000, renews_at=current_cycle_renews_at, + renewal_period=current_spending_limit.frequency.value, ) budget_estimate_response = await VaspUmaClient.instance().get_budget_estimate( @@ -56,4 +57,5 @@ async def get_budget(access_token: str, request: Nip47Request) -> Nip47BudgetRes total_budget=total_budget_sats * 1000, used_budget=used_budget_sats * 1000, renews_at=current_cycle_renews_at, + renewal_period=current_spending_limit.frequency.value, ) diff --git a/nwc_backend/models/nip47_budget.py b/nwc_backend/models/nip47_budget.py index 522c178..450c30c 100644 --- a/nwc_backend/models/nip47_budget.py +++ b/nwc_backend/models/nip47_budget.py @@ -17,6 +17,7 @@ class Nip47BudgetResponse: used_budget: Optional[int] = None total_budget: Optional[int] = None renews_at: Optional[int] = None + renewal_period: Optional[str] = None currency: Optional[Nip47BudgetCurrency] = None def to_dict(self):