From 00f39a20074234ed00ee07fff11c894579c375ac Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?dni=20=E2=9A=A1?= <office@dnilabs.com>
Date: Sun, 21 Apr 2024 13:07:42 +0200
Subject: [PATCH] test: add unit for fee_reserve and service_fee

:)

sorting
---
 tests/unit/test_services_fees.py | 75 ++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 tests/unit/test_services_fees.py

diff --git a/tests/unit/test_services_fees.py b/tests/unit/test_services_fees.py
new file mode 100644
index 00000000..14e7ea75
--- /dev/null
+++ b/tests/unit/test_services_fees.py
@@ -0,0 +1,75 @@
+import pytest
+
+from lnbits.core.services import (
+    fee_reserve,
+    fee_reserve_total,
+    service_fee,
+)
+from lnbits.settings import settings
+
+
+@pytest.mark.asyncio
+async def test_fee_reserve_internal():
+    fee = fee_reserve(10_000, internal=True)
+    assert fee == 0
+
+
+@pytest.mark.asyncio
+async def test_fee_reserve_min():
+    settings.lnbits_reserve_fee_percent = 2
+    settings.lnbits_reserve_fee_min = 500
+    fee = fee_reserve(10000)
+    assert fee == 500
+
+
+@pytest.mark.asyncio
+async def test_fee_reserve_percent():
+    settings.lnbits_reserve_fee_percent = 1
+    settings.lnbits_reserve_fee_min = 100
+    fee = fee_reserve(100000)
+    assert fee == 1000
+
+
+@pytest.mark.asyncio
+async def test_service_fee_no_wallet():
+    settings.lnbits_service_fee_wallet = ""
+    fee = service_fee(10000)
+    assert fee == 0
+
+
+@pytest.mark.asyncio
+async def test_service_fee_internal():
+    settings.lnbits_service_fee_wallet = "wallet_id"
+    settings.lnbits_service_fee_ignore_internal = True
+    fee = service_fee(10000, internal=True)
+    assert fee == 0
+
+
+@pytest.mark.asyncio
+async def test_service_fee():
+    settings.lnbits_service_fee_wallet = "wallet_id"
+    settings.lnbits_service_fee = 2
+    fee = service_fee(10000)
+    assert fee == 200
+
+
+@pytest.mark.asyncio
+async def test_service_fee_max():
+    settings.lnbits_service_fee_wallet = "wallet_id"
+    settings.lnbits_service_fee = 2
+    settings.lnbits_service_fee_max = 199
+    fee = service_fee(100_000_000)
+    assert fee / 1000 == 199
+
+
+@pytest.mark.asyncio
+async def test_fee_reserve_total():
+    settings.lnbits_reserve_fee_percent = 1
+    settings.lnbits_reserve_fee_min = 100
+    settings.lnbits_service_fee = 2
+    settings.lnbits_service_fee_wallet = "wallet_id"
+    amount = 100_000
+    fee = service_fee(amount)
+    reserve = fee_reserve(amount)
+    total = fee_reserve_total(amount)
+    assert fee + reserve == total