|
| 1 | +import asyncio |
| 2 | +import json |
| 3 | +import ssl |
| 4 | +import _thread |
| 5 | +import time |
| 6 | +import unittest |
| 7 | + |
| 8 | +import sys |
| 9 | +sys.path.append("apps/com.lightningpiggy.displaywallet/assets/") |
| 10 | +from wallet import LNBitsWallet |
| 11 | + |
| 12 | +class TestLNBitsWallet(unittest.TestCase): |
| 13 | + |
| 14 | + redraw_balance_cb_called = 0 |
| 15 | + redraw_payments_cb_called = 0 |
| 16 | + redraw_static_receive_code_cb_called = 0 |
| 17 | + error_callback_called = 0 |
| 18 | + |
| 19 | + def redraw_balance_cb(self, balance=0): |
| 20 | + print(f"redraw_callback called, balance: {balance}") |
| 21 | + self.redraw_balance_cb_called += 1 |
| 22 | + |
| 23 | + def redraw_payments_cb(self): |
| 24 | + print(f"redraw_payments_cb called") |
| 25 | + self.redraw_payments_cb_called += 1 |
| 26 | + |
| 27 | + def redraw_static_receive_code_cb(self): |
| 28 | + print(f"redraw_static_receive_code_cb called") |
| 29 | + self.redraw_static_receive_code_cb_called += 1 |
| 30 | + |
| 31 | + def error_callback(self, error): |
| 32 | + print(f"error_callback called, error: {error}") |
| 33 | + self.error_callback_called += 1 |
| 34 | + |
| 35 | + def test_it(self): |
| 36 | + print("starting test") |
| 37 | + self.wallet = LNBitsWallet("http://192.168.1.16:5000/", "5a2cf5d536ec45cb9a043071002e4449") |
| 38 | + self.wallet.start(self.redraw_balance_cb, self.redraw_payments_cb, self.redraw_static_receive_code_cb, self.error_callback) |
| 39 | + time.sleep(5) |
| 40 | + self.assertTrue(self.redraw_balance_cb_called > 0) |
| 41 | + self.assertTrue(self.redraw_payments_cb_called > 0) |
| 42 | + self.assertTrue(self.redraw_static_receive_code_cb_called == 0) # no static receive code so error 404 |
| 43 | + self.assertTrue(self.error_callback_called == 1) |
| 44 | + print("test finished") |
| 45 | + |
| 46 | + |
0 commit comments