7c3716535f34f77e7e0789a45b646985333c1766
[ldk-java] / ts / test / tests.mts
1 import * as rawldk from "../bindings.mjs";
2 import * as ldk from "../index.mjs";
3
4 const tests: Array<Function> = [];
5
6 function array_eq(a: Uint8Array, b: Uint8Array): boolean {
7         return a.length == b.length && a.every((v, idx) => v == b[idx]);
8 }
9
10 tests.push(async () => {
11         const result = rawldk.CResult_boolLightningErrorZ_ok(true);
12         console.assert(rawldk.CResult_boolLightningErrorZ_is_ok(result));
13         console.assert(rawldk.CResult_boolLightningErrorZ_get_ok(result));
14         rawldk.CResult_boolLightningErrorZ_free(result);
15         console.assert(rawldk.CResult_boolLightningErrorZ_ok(false) == result); // malloc doesn't need to guarantee this, but currently does
16         console.assert(rawldk.CResult_boolLightningErrorZ_is_ok(result));
17         console.assert(!rawldk.CResult_boolLightningErrorZ_get_ok(result));
18         rawldk.CResult_boolLightningErrorZ_free(result);
19
20         /*var pk_arr = [];
21         for (var i = 0; i < 33; i++) { pk_arr[i] = 42; }
22         const pk_bytes = encodeUint8Array(pk_arr);
23         const pk_res = wasm.TS_CResult_PublicKeyErrorZ_ok(pk_bytes);
24         console.assert(wasm.TS_CResult_PublicKeyErrorZ_is_ok(pk_res));
25         const pk_res_bytes = wasm.TS_LDKCResult_PublicKeyErrorZ_get_ok(pk_res);
26         wasm.TS_CResult_PublicKeyErrorZ_free(pk_res);*/
27         return true;
28 });
29
30 tests.push(async () => {
31         const ping = ldk.Ping.constructor_new(10, 2);
32         const new_ping = ldk.Ping.constructor_read(ping.write());
33         if (!(new_ping instanceof ldk.Result_PingDecodeErrorZ_OK)) return false;
34         if (!new_ping.is_ok()) return false;
35         if (new_ping.res.get_byteslen() != 2) return false;
36         if (new_ping.res.get_ponglen() != 10) return false;
37         return true;
38 });
39
40 tests.push(async () => {
41         const outpoint = ldk.OutPoint.constructor_new(new Uint8Array(32), 4);
42         const read_outpoint = ldk.OutPoint.constructor_read(outpoint.write());
43         if (!(read_outpoint instanceof ldk.Result_OutPointDecodeErrorZ_OK)) return false;
44         if (!read_outpoint.res.eq(outpoint)) return false;
45         if (read_outpoint.res.hash() != outpoint.hash()) return false;
46         const chan_id = read_outpoint.res.to_channel_id();
47         if (chan_id.length != 32) return false;
48         if (chan_id[31] != 4) return false;
49         return true;
50 });
51
52 tests.push(async () => {
53         const outpoint = ldk.OutPoint.constructor_new(new Uint8Array(32), 4);
54         const read_outpoint = ldk.OutPoint.constructor_read(outpoint.write());
55         if (!(read_outpoint instanceof ldk.Result_OutPointDecodeErrorZ_OK)) return false;
56         if (!read_outpoint.res.eq(outpoint)) return false;
57         if (read_outpoint.res.hash() != outpoint.hash()) return false;
58         const chan_id = read_outpoint.res.to_channel_id();
59         if (chan_id.length != 32) return false;
60         if (chan_id[31] != 4) return false;
61         return true;
62 });
63
64 var seed_counter = 0;
65 function get_chanman() {
66         const fee_est = ldk.FeeEstimator.new_impl({
67                 get_est_sat_per_1000_weight(confirmation_target: ldk.ConfirmationTarget): number {
68                         return 253;
69                 }
70         } as ldk.FeeEstimatorInterface);
71         const tx_broadcaster = ldk.BroadcasterInterface.new_impl({
72                 broadcast_transaction(tx: Uint8Array): void { console.log("Tx Broadcast: " + tx); }
73         } as ldk.BroadcasterInterfaceInterface);
74         const logger = ldk.Logger.new_impl({
75                 log(record: ldk.Record): void {
76                         console.log(record.get_module_path() + ": " + record.get_args());
77                 }
78         } as ldk.LoggerInterface);
79
80         const persister = ldk.Persist.new_impl({
81                 persist_new_channel(channel_id: ldk.OutPoint, data: ldk.ChannelMonitor, update_id: ldk.MonitorUpdateId): ldk.Result_NoneChannelMonitorUpdateErrZ {
82                         return ldk.Result_NoneChannelMonitorUpdateErrZ.constructor_ok();
83                 },
84                 update_persisted_channel(channel_id: ldk.OutPoint, update: ldk.ChannelMonitorUpdate, data: ldk.ChannelMonitor, update_id: ldk.MonitorUpdateId): ldk.Result_NoneChannelMonitorUpdateErrZ {
85                         return ldk.Result_NoneChannelMonitorUpdateErrZ.constructor_ok();
86                 }
87         } as ldk.PersistInterface);
88
89         const chain_monitor = ldk.ChainMonitor.constructor_new(ldk.Option_FilterZ.constructor_none(), tx_broadcaster, logger, fee_est, persister);
90         const chain_watch: ldk.Watch = chain_monitor.as_Watch();
91
92         const seed = new Uint8Array(32);
93         seed.fill(seed_counter);
94         seed_counter++;
95         const keys_manager = ldk.KeysManager.constructor_new(seed, BigInt(42), 42);
96         const keys_interface = keys_manager.as_KeysInterface();
97         const config = ldk.UserConfig.constructor_default();
98         const params = ldk.ChainParameters.constructor_new(ldk.Network.LDKNetwork_Testnet, ldk.BestBlock.constructor_from_genesis(ldk.Network.LDKNetwork_Testnet));
99
100         return ldk.ChannelManager.constructor_new(fee_est, chain_watch, tx_broadcaster, logger, keys_interface, config, params);
101 }
102
103 function exchange_messages(a: ldk.ChannelManager, b: ldk.ChannelManager) {
104         var found_msgs = true;
105         while (found_msgs) {
106                 const as_msgs = a.as_MessageSendEventsProvider().get_and_clear_pending_msg_events();
107                 const bs_msgs = b.as_MessageSendEventsProvider().get_and_clear_pending_msg_events();
108                 found_msgs = as_msgs.length != 0 || bs_msgs.length != 0;
109                 for (var i = 0; i < 2; i++) {
110                         var to: ldk.ChannelManager; var from: ldk.ChannelManager; var msgs: ldk.MessageSendEvent[];
111                         if (i == 0) { from = a; to = b; msgs = as_msgs; } else { from = b; to = a; msgs = bs_msgs; }
112                         for (var j = 0; j < msgs.length; j++) {
113                                 const msg = msgs[j];
114                                 if (msg instanceof ldk.MessageSendEvent_SendOpenChannel) {
115                                         if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
116                                         to.as_ChannelMessageHandler().handle_open_channel(from.get_our_node_id(), ldk.InitFeatures.constructor_known(), msg.msg);
117                                 } else if (msg instanceof ldk.MessageSendEvent_SendAcceptChannel) {
118                                         if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
119                                         to.as_ChannelMessageHandler().handle_accept_channel(from.get_our_node_id(), ldk.InitFeatures.constructor_known(), msg.msg);
120                                 } else {
121                                         return false;
122                                 }
123                         }
124                 }
125         }
126         return true;
127 }
128
129 tests.push(async () => {
130         const chan_man_a = get_chanman();
131         const chan_man_b = get_chanman();
132
133         chan_man_a.as_ChannelMessageHandler().peer_connected(chan_man_b.get_our_node_id(), ldk.Init.constructor_new(ldk.InitFeatures.constructor_known()));
134         chan_man_b.as_ChannelMessageHandler().peer_connected(chan_man_a.get_our_node_id(), ldk.Init.constructor_new(ldk.InitFeatures.constructor_known()));
135
136         const chan_create_err = chan_man_a.create_channel(chan_man_b.get_our_node_id(), BigInt(0), BigInt(400), BigInt(0), ldk.UserConfig.constructor_default());
137         if (chan_create_err.is_ok()) return false;
138         if (!(chan_create_err instanceof ldk.Result__u832APIErrorZ_Err)) return false;
139         if (!(chan_create_err.err instanceof ldk.APIError_APIMisuseError)) return false;
140         if (chan_create_err.err.err != "Channel value must be at least 1000 satoshis. It was 0") return false;
141
142         const chan_create_res = chan_man_a.create_channel(chan_man_b.get_our_node_id(), BigInt(1000000), BigInt(400), BigInt(0), ldk.UserConfig.constructor_default());
143         if (!chan_create_res.is_ok()) return false;
144
145         if (!exchange_messages(chan_man_a, chan_man_b)) return false;
146
147         const events: Array<ldk.Event> = [];
148         const event_handler = ldk.EventHandler.new_impl({
149                 handle_event(event: ldk.Event): void {
150                         events.push(event);
151                 }
152         } as ldk.EventHandlerInterface);
153
154         chan_man_a.as_EventsProvider().process_pending_events(event_handler);
155         if (events.length != 1) return false;
156         if (!(events[0] instanceof ldk.Event_FundingGenerationReady)) return false;
157
158         return true;
159 });
160
161 export async function run_tests(wasm_path: string) {
162         await rawldk.initializeWasm(wasm_path);
163
164         var test_runs = [];
165         for (const test of tests) {
166                 test_runs.push(test());
167         }
168         const results = await Promise.all(test_runs);
169         console.log("test results: " + results);
170         const result = results.every((v) => { return v === true });
171         console.log("all tests passed: " + result);
172         return result;
173 }