m5/MicroPythonOS
0
0
Fork 0
mirror of https://github.com/m5stack/MicroPythonOS.git synced 2026-05-20 11:51:27 -07:00
Code Issues Packages Projects Releases Wiki Activity

Update Wallet tests

Browse Source
This commit is contained in:
Thomas Farstrike
2025-11-11 14:29:52 +01:00
parent 887e5fc485
commit a8f0c2e2c1
2 changed files with 117 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/manual_test_lnbitswallet.py
+61 -7
View File
@@ -4,6 +4,8 @@ import ssl
import _thread
import time
import unittest
import requests
import ujson
import sys
sys.path.append("apps/com.lightningpiggy.displaywallet/assets/")
@@ -16,8 +18,8 @@ class TestLNBitsWallet(unittest.TestCase):
redraw_static_receive_code_cb_called = 0
error_callback_called = 0
def redraw_balance_cb(self, balance=0):
print(f"redraw_callback called, balance: {balance}")
def redraw_balance_cb(self, balance_added=0):
print(f"redraw_callback called, balance_added: {balance_added}")
self.redraw_balance_cb_called += 1
def redraw_payments_cb(self):
@@ -32,15 +34,67 @@ class TestLNBitsWallet(unittest.TestCase):
print(f"error_callback called, error: {error}")
self.error_callback_called += 1
def update_balance(self, sats):
"""
Updates the user balance by 'sats' amount using the local API.
Authenticates first, then sends the balance update.
"""
try:
# Step 1: Authenticate and get access token
auth_url = "http://192.168.1.16:5000/api/v1/auth"
auth_payload = {"username": "admin", "password": "adminadmin"}
print("Authenticating...")
auth_response = requests.post( auth_url, json=auth_payload, headers={"Content-Type": "application/json"} )
if auth_response.status_code != 200:
print("Auth failed:", auth_response.text)
auth_response.close()
return False
auth_data = ujson.loads(auth_response.text)
access_token = auth_data["access_token"]
auth_response.close()
print("Authenticated, got token.")
# Step 2: Update balance
balance_url = "http://192.168.1.16:5000/users/api/v1/balance"
balance_payload = { "amount": str(sats), "id": "24e9334d39b946a3b642f5fd8c292a07" }
cookie_header = f"cookie_access_token={access_token}; is_lnbits_user_authorized=true"
print(f"Updating balance by {sats} sats...")
update_response = requests.put(
balance_url,
json=balance_payload,
headers={ "Content-Type": "application/json", "Cookie": cookie_header })
result = ujson.loads(update_response.text)
update_response.close()
if result.get("success"):
print("Balance updated successfully!")
return True
else:
print("Update failed:", result)
return False
except Exception as e:
print("Error:", e)
return False
def test_it(self):
print("starting test")
import sys
print(sys.path)
self.wallet = LNBitsWallet("http://192.168.1.16:5000/", "5a2cf5d536ec45cb9a043071002e4449")
self.wallet.start(self.redraw_balance_cb, self.redraw_payments_cb, self.redraw_static_receive_code_cb, self.error_callback)
time.sleep(5)
self.assertTrue(self.redraw_balance_cb_called > 0)
self.assertTrue(self.redraw_payments_cb_called > 0)
self.assertTrue(self.redraw_static_receive_code_cb_called == 0) # no static receive code so error 404
self.assertTrue(self.error_callback_called == 1)
time.sleep(3)
self.assertEqual(self.redraw_balance_cb_called, 1)
self.assertGreaterEqual(self.redraw_payments_cb_called, 3)
before_receive = self.redraw_payments_cb_called
self.assertEqual(self.redraw_static_receive_code_cb_called, 0) # no static receive code so error 404
self.assertEqual(self.error_callback_called, 1)
print("Everything good so far, now add a transaction...")
self.update_balance(9)
time.sleep(2) # allow some time for the notification
self.wallet.stop() # don't stop the wallet for the fullscreen QR activity
time.sleep(2)
self.assertEqual(self.redraw_balance_cb_called, 2)
self.assertGreaterEqual(self.redraw_payments_cb_called, before_receive+1)
self.assertEqual(self.redraw_static_receive_code_cb_called, 0) # no static receive code so error 404
self.assertEqual(self.error_callback_called, 1)
print("test finished")
tests/manual_test_nwcwallet.py
+56 -7
View File
@@ -4,9 +4,8 @@ import ssl
import _thread
import time
import unittest
from mpos import App, PackageManager
import mpos.apps
import requests
import ujson
import sys
sys.path.append("apps/com.lightningpiggy.displaywallet/assets/")
@@ -35,15 +34,65 @@ class TestNWCWallet(unittest.TestCase):
print(f"error_callback called, error: {error}")
self.error_callback_called += 1
def update_balance(self, sats):
"""
Updates the user balance by 'sats' amount using the local API.
Authenticates first, then sends the balance update.
"""
try:
# Step 1: Authenticate and get access token
auth_url = "http://192.168.1.16:5000/api/v1/auth"
auth_payload = {"username": "admin", "password": "adminadmin"}
print("Authenticating...")
auth_response = requests.post( auth_url, json=auth_payload, headers={"Content-Type": "application/json"} )
if auth_response.status_code != 200:
print("Auth failed:", auth_response.text)
auth_response.close()
return False
auth_data = ujson.loads(auth_response.text)
access_token = auth_data["access_token"]
auth_response.close()
print("Authenticated, got token.")
# Step 2: Update balance
balance_url = "http://192.168.1.16:5000/users/api/v1/balance"
balance_payload = { "amount": str(sats), "id": "24e9334d39b946a3b642f5fd8c292a07" }
cookie_header = f"cookie_access_token={access_token}; is_lnbits_user_authorized=true"
print(f"Updating balance by {sats} sats...")
update_response = requests.put(
balance_url,
json=balance_payload,
headers={ "Content-Type": "application/json", "Cookie": cookie_header })
result = ujson.loads(update_response.text)
update_response.close()
if result.get("success"):
print("Balance updated successfully!")
return True
else:
print("Update failed:", result)
return False
except Exception as e:
print("Error:", e)
return False
def test_it(self):
print("starting test")
self.wallet = NWCWallet("nostr+walletconnect://e46762afab282c324278351165122345f9983ea447b47943b052100321227571?relay=ws://192.168.1.16:5000/nostrclient/api/v1/relay&secret=fab0a9a11d4cf4b1d92e901a0b2c56634275e2fa1a7eb396ff1b942f95d59fd3&lud16=test@example.com")
self.wallet.start(self.redraw_balance_cb, self.redraw_payments_cb, self.redraw_static_receive_code_cb, self.error_callback)
print("\n\nWaiting a bit for the startup to be settled...")
time.sleep(15)
self.assertTrue(self.redraw_balance_cb_called > 0)
self.assertTrue(self.redraw_payments_cb_called > 0)
self.assertTrue(self.redraw_static_receive_code_cb_called > 0)
self.assertTrue(self.error_callback_called == 0)
print("\nAsserting state...")
saved = self.redraw_balance_cb_called
print(f"redraw_balance_cb_called is {self.redraw_balance_cb_called}")
self.assertGreaterEqual(self.redraw_balance_cb_called,1)
self.assertGreaterEqual(self.redraw_payments_cb_called, 1)
self.assertGreaterEqual(self.redraw_static_receive_code_cb_called, 1)
self.assertEqual(self.error_callback_called, 0)
self.update_balance(321)
time.sleep(20)
self.assertNotEqual(self.redraw_balance_cb_called,saved+1, "should be equal, but LNBits doesn't seem to send payment notifications (yet)")
self.assertGreaterEqual(self.redraw_payments_cb_called, 1)
self.assertGreaterEqual(self.redraw_static_receive_code_cb_called, 1)
print("test finished")