_ Git - ldk-java/blob - 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         const second_res = rawldk.CResult_boolLightningErrorZ_ok(false);
  16         console.assert(rawldk.CResult_boolLightningErrorZ_is_ok(second_res));
  17         console.assert(!rawldk.CResult_boolLightningErrorZ_get_ok(second_res));
  18         rawldk.CResult_boolLightningErrorZ_free(second_res);
  19
  20         return true;
  21 });
  22
  23 tests.push(async () => {
  24         const ping = ldk.Ping.constructor_new(10, 2);
  25         const new_ping = ldk.Ping.constructor_read(ping.write());
  26         if (!(new_ping instanceof ldk.Result_PingDecodeErrorZ_OK)) return false;
  27         if (!new_ping.is_ok()) return false;
  28         if (new_ping.res.get_byteslen() != 2) return false;
  29         if (new_ping.res.get_ponglen() != 10) return false;
  30         return true;
  31 });
  32
  33 tests.push(async () => {
  34         const outpoint = ldk.OutPoint.constructor_new(new Uint8Array(32), 4);
  35         const read_outpoint = ldk.OutPoint.constructor_read(outpoint.write());
  36         if (!(read_outpoint instanceof ldk.Result_OutPointDecodeErrorZ_OK)) return false;
  37         if (!read_outpoint.res.eq(outpoint)) return false;
  38         if (read_outpoint.res.hash() != outpoint.hash()) return false;
  39         const chan_id = read_outpoint.res.to_channel_id();
  40         if (chan_id.length != 32) return false;
  41         if (chan_id[31] != 4) return false;
  42         return true;
  43 });
  44
  45 tests.push(async () => {
  46         const outpoint = ldk.OutPoint.constructor_new(new Uint8Array(32), 4);
  47         const read_outpoint = ldk.OutPoint.constructor_read(outpoint.write());
  48         if (!(read_outpoint instanceof ldk.Result_OutPointDecodeErrorZ_OK)) return false;
  49         if (!read_outpoint.res.eq(outpoint)) return false;
  50         if (read_outpoint.res.hash() != outpoint.hash()) return false;
  51         const chan_id = read_outpoint.res.to_channel_id();
  52         if (chan_id.length != 32) return false;
  53         if (chan_id[31] != 4) return false;
  54         return true;
  55 });
  56
  57 var seed_counter = 0;
  58 class Node {
  59         node_id: Uint8Array;
  60         node_secret: Uint8Array;
  61         constructor(public chan_man: ldk.ChannelManager, public tx_broadcasted: Promise<Uint8Array>,
  62         public logger: ldk.Logger, public keys_interface: ldk.KeysInterface) {
  63                 this.node_id = chan_man.get_our_node_id();
  64                 this.node_secret = (keys_interface.get_node_secret(ldk.Recipient.LDKRecipient_Node) as ldk.Result_SecretKeyNoneZ_OK).res;
  65         }
  66 }
  67 function get_chanman(): Node {
  68         const fee_est = ldk.FeeEstimator.new_impl({
  69                 get_est_sat_per_1000_weight(confirmation_target: ldk.ConfirmationTarget): number {
  70                         return 253;
  71                 }
  72         } as ldk.FeeEstimatorInterface);
  73         var tx_broadcaster;
  74         const tx_broadcasted: Promise<Uint8Array> = new Promise((resolve, reject) => {
  75                 tx_broadcaster = ldk.BroadcasterInterface.new_impl({
  76                         broadcast_transaction(tx: Uint8Array): void { console.log("Tx Broadcast: " + tx); resolve(tx); }
  77                 } as ldk.BroadcasterInterfaceInterface);
  78         });
  79         const logger = ldk.Logger.new_impl({
  80                 log(record: ldk.Record): void {
  81                         console.log(record.get_module_path() + ": " + record.get_args());
  82                 }
  83         } as ldk.LoggerInterface);
  84
  85         const persister = ldk.Persist.new_impl({
  86                 persist_new_channel(channel_id: ldk.OutPoint, data: ldk.ChannelMonitor, update_id: ldk.MonitorUpdateId): ldk.ChannelMonitorUpdateStatus {
  87                         return ldk.ChannelMonitorUpdateStatus.LDKChannelMonitorUpdateStatus_Completed;
  88                 },
  89                 update_persisted_channel(channel_id: ldk.OutPoint, update: ldk.ChannelMonitorUpdate, data: ldk.ChannelMonitor, update_id: ldk.MonitorUpdateId): ldk.ChannelMonitorUpdateStatus {
  90                         return ldk.ChannelMonitorUpdateStatus.LDKChannelMonitorUpdateStatus_Completed;
  91                 }
  92         } as ldk.PersistInterface);
  93
  94         const chain_monitor = ldk.ChainMonitor.constructor_new(ldk.Option_FilterZ.constructor_none(), tx_broadcaster, logger, fee_est, persister);
  95         const chain_watch: ldk.Watch = chain_monitor.as_Watch();
  96
  97         const seed = new Uint8Array(32);
  98         seed.fill(seed_counter);
  99         seed_counter++;
 100         const keys_manager = ldk.KeysManager.constructor_new(seed, BigInt(42), 42);
 101         const keys_interface = keys_manager.as_KeysInterface();
 102         const config = ldk.UserConfig.constructor_default();
 103         const params = ldk.ChainParameters.constructor_new(ldk.Network.LDKNetwork_Testnet, ldk.BestBlock.constructor_from_genesis(ldk.Network.LDKNetwork_Testnet));
 104
 105         const chan_man = ldk.ChannelManager.constructor_new(fee_est, chain_watch, tx_broadcaster, logger, keys_interface, config, params);
 106         return new Node(chan_man, tx_broadcasted, logger, keys_interface);
 107 }
 108
 109 function exchange_messages(a: ldk.ChannelManager, b: ldk.ChannelManager) {
 110         var found_msgs = true;
 111         while (found_msgs) {
 112                 const as_msgs = a.as_MessageSendEventsProvider().get_and_clear_pending_msg_events();
 113                 const bs_msgs = b.as_MessageSendEventsProvider().get_and_clear_pending_msg_events();
 114                 found_msgs = as_msgs.length != 0 || bs_msgs.length != 0;
 115                 for (var i = 0; i < 2; i++) {
 116                         var to: ldk.ChannelManager; var from: ldk.ChannelManager; var msgs: ldk.MessageSendEvent[];
 117                         if (i == 0) { from = a; to = b; msgs = as_msgs; } else { from = b; to = a; msgs = bs_msgs; }
 118                         for (var j = 0; j < msgs.length; j++) {
 119                                 const msg = msgs[j];
 120                                 if (msg instanceof ldk.MessageSendEvent_SendOpenChannel) {
 121                                         if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
 122                                         const features = from.as_ChannelMessageHandler().provided_init_features(to.get_our_node_id());
 123                                         to.as_ChannelMessageHandler().handle_open_channel(from.get_our_node_id(), features, msg.msg);
 124                                 } else if (msg instanceof ldk.MessageSendEvent_SendAcceptChannel) {
 125                                         if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
 126                                         const features = from.as_ChannelMessageHandler().provided_init_features(to.get_our_node_id());
 127                                         to.as_ChannelMessageHandler().handle_accept_channel(from.get_our_node_id(), features, msg.msg);
 128                                 } else if (msg instanceof ldk.MessageSendEvent_SendFundingCreated) {
 129                                         if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
 130                                         to.as_ChannelMessageHandler().handle_funding_created(from.get_our_node_id(), msg.msg);
 131                                 } else if (msg instanceof ldk.MessageSendEvent_SendFundingSigned) {
 132                                         if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
 133                                         to.as_ChannelMessageHandler().handle_funding_signed(from.get_our_node_id(), msg.msg);
 134                                 } else {
 135                                         return false;
 136                                 }
 137                         }
 138                 }
 139         }
 140         return true;
 141 }
 142
 143 function assign_u64(arr: Uint8Array, offset: number, value: bigint) {
 144         arr[offset + 0] = Number((value >> BigInt(8 * 0)) & BigInt(0xff));
 145         arr[offset + 1] = Number((value >> BigInt(8 * 1)) & BigInt(0xff));
 146         arr[offset + 2] = Number((value >> BigInt(8 * 2)) & BigInt(0xff));
 147         arr[offset + 3] = Number((value >> BigInt(8 * 3)) & BigInt(0xff));
 148         arr[offset + 4] = Number((value >> BigInt(8 * 4)) & BigInt(0xff));
 149         arr[offset + 5] = Number((value >> BigInt(8 * 5)) & BigInt(0xff));
 150         arr[offset + 6] = Number((value >> BigInt(8 * 6)) & BigInt(0xff));
 151         arr[offset + 7] = Number((value >> BigInt(8 * 7)) & BigInt(0xff));
 152 }
 153
 154 function get_event(chan_man: ldk.ChannelManager): ldk.Event {
 155         const events: Array<ldk.Event> = [];
 156         const event_handler = ldk.EventHandler.new_impl({
 157                 handle_event(event: ldk.Event): void {
 158                         events.push(event);
 159                 }
 160         } as ldk.EventHandlerInterface);
 161
 162         chan_man.as_EventsProvider().process_pending_events(event_handler);
 163         console.assert(events.length == 1);
 164         return events[0];
 165 }
 166
 167 tests.push(async () => {
 168         const a = get_chanman();
 169         const b = get_chanman();
 170
 171         const features = a.chan_man.as_ChannelMessageHandler().provided_init_features(b.chan_man.get_our_node_id());
 172
 173         a.chan_man.as_ChannelMessageHandler().peer_connected(b.chan_man.get_our_node_id(), ldk.Init.constructor_new(features, ldk.Option_NetAddressZ.constructor_none()));
 174         b.chan_man.as_ChannelMessageHandler().peer_connected(a.chan_man.get_our_node_id(), ldk.Init.constructor_new(features, ldk.Option_NetAddressZ.constructor_none()));
 175
 176         const chan_create_err = a.chan_man.create_channel(b.chan_man.get_our_node_id(), BigInt(0), BigInt(400), BigInt(0), ldk.UserConfig.constructor_default());
 177         if (chan_create_err.is_ok()) return false;
 178         if (!(chan_create_err instanceof ldk.Result__u832APIErrorZ_Err)) return false;
 179         if (!(chan_create_err.err instanceof ldk.APIError_APIMisuseError)) return false;
 180         if (chan_create_err.err.err != "Channel value must be at least 1000 satoshis. It was 0") return false;
 181
 182         const chan_create_res = a.chan_man.create_channel(b.chan_man.get_our_node_id(), BigInt(1000000), BigInt(400), BigInt(0), ldk.UserConfig.constructor_default());
 183         if (!chan_create_res.is_ok()) return false;
 184
 185         if (!exchange_messages(a.chan_man, b.chan_man)) return false;
 186
 187         const event = get_event(a.chan_man) as ldk.Event_FundingGenerationReady;
 188
 189         // (very) manually create a funding transaction
 190         const witness_pos = event.output_script.length + 58;
 191         const funding_tx = new Uint8Array(witness_pos + 7);
 192         funding_tx[0] = 2; // 4-byte tx version 2
 193         funding_tx[4] = 0; funding_tx[5] = 1; // segwit magic bytes
 194         funding_tx[6] = 1; // 1-byte input count 1
 195         // 36 bytes previous outpoint all-0s
 196         funding_tx[43] = 0; // 1-byte input script length 0
 197         funding_tx[44] = 0xff; funding_tx[45] = 0xff; funding_tx[46] = 0xff; funding_tx[47] = 0xff; // 4-byte nSequence
 198         funding_tx[48] = 1; // one output
 199         assign_u64(funding_tx, 49, event.channel_value_satoshis);
 200         funding_tx[57] = event.output_script.length; // 1-byte output script length
 201         funding_tx.set(event.output_script, 58);
 202         funding_tx[witness_pos] = 1; funding_tx[witness_pos + 1] = 1; funding_tx[witness_pos + 2] = 0xff; // one witness element of size 1 with contents 0xff
 203         funding_tx[witness_pos + 3] = 0; funding_tx[witness_pos + 4] = 0; funding_tx[witness_pos + 5] = 0; funding_tx[witness_pos + 6] = 0; // lock time 0
 204
 205         const funding_res = a.chan_man.funding_transaction_generated(event.temporary_channel_id, event.counterparty_node_id, funding_tx);
 206         if (!(funding_res instanceof ldk.Result_NoneAPIErrorZ_OK)) return false;
 207
 208         if (!exchange_messages(a.chan_man, b.chan_man)) return false;
 209
 210         const tx_broadcasted: Uint8Array = (await a.tx_broadcasted);
 211         if (!array_eq(tx_broadcasted, funding_tx)) return false;
 212
 213         return true;
 214 });
 215
 216 tests.push(async () => {
 217         const a = get_chanman();
 218         const b = get_chanman();
 219
 220         const ignorer = ldk.IgnoringMessageHandler.constructor_new();
 221         const pm_a = ldk.PeerManager.constructor_new(a.chan_man.as_ChannelMessageHandler(), ignorer.as_RoutingMessageHandler(), ignorer.as_OnionMessageHandler(), a.node_secret, 0xdeadbeef, a.node_secret, a.logger, ignorer.as_CustomMessageHandler());
 222         const pm_b = ldk.PeerManager.constructor_new(b.chan_man.as_ChannelMessageHandler(), ignorer.as_RoutingMessageHandler(), ignorer.as_OnionMessageHandler(), b.node_secret, 0xdeadbeef, b.node_secret, b.logger, ignorer.as_CustomMessageHandler());
 223
 224         var sock_b: ldk.SocketDescriptor;
 225         const sock_a = ldk.SocketDescriptor.new_impl({
 226                 send_data(data: Uint8Array, resume_read: boolean): number {
 227                         console.assert(pm_b.read_event(sock_b, data) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
 228                         return data.length;
 229                 },
 230                 disconnect_socket(): void {
 231                         console.assert(false);
 232                 },
 233                 eq(other: ldk.SocketDescriptor): boolean {
 234                         return other.hash() == this.hash();
 235                 },
 236                 hash(): bigint {
 237                         return BigInt(1);
 238                 }
 239         } as ldk.SocketDescriptorInterface);
 240         sock_b = ldk.SocketDescriptor.new_impl({
 241                 send_data(data: Uint8Array, resume_read: boolean): number {
 242                         console.assert(pm_a.read_event(sock_a, data) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
 243                         return data.length;
 244                 },
 245                 disconnect_socket(): void {
 246                         console.assert(false);
 247                 },
 248                 eq(other: ldk.SocketDescriptor): boolean {
 249                         return other.hash() == this.hash();
 250                 },
 251                 hash(): bigint {
 252                         return BigInt(2);
 253                 }
 254         } as ldk.SocketDescriptorInterface);
 255
 256         const update_fut = a.chan_man.get_persistable_update_future();
 257         var update_done = false;
 258         update_fut.register_callback_fn(ldk.FutureCallback.new_impl({
 259                 call(): void { update_done = true; }
 260         }));
 261         if (update_done) return false;
 262
 263         const v4_netaddr = ldk.NetAddress.constructor_ipv4(Uint8Array.from([42,0,42,1]), 9735);
 264         console.assert(pm_b.new_inbound_connection(sock_b, ldk.Option_NetAddressZ.constructor_some(v4_netaddr)) instanceof ldk.Result_NonePeerHandleErrorZ_OK);
 265         const init_bytes = pm_a.new_outbound_connection(b.node_id, sock_a, ldk.Option_NetAddressZ.constructor_none());
 266         if (!(init_bytes instanceof ldk.Result_CVec_u8ZPeerHandleErrorZ_OK)) return false;
 267         console.assert(pm_b.read_event(sock_b, init_bytes.res) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
 268
 269         console.assert(pm_a.get_peer_node_ids().length == 0);
 270         console.assert(pm_b.get_peer_node_ids().length == 0);
 271
 272         pm_b.process_events();
 273         pm_a.process_events();
 274         pm_b.process_events();
 275
 276         console.assert(pm_a.get_peer_node_ids().length == 1);
 277         console.assert(pm_b.get_peer_node_ids().length == 1);
 278
 279         const chan_create_res = a.chan_man.create_channel(b.node_id, BigInt(1000000), BigInt(400), BigInt(0), ldk.UserConfig.constructor_default());
 280         if (!chan_create_res.is_ok()) return false;
 281         if (!update_done) return false;
 282
 283         pm_a.process_events();
 284         pm_b.process_events();
 285
 286         const event = get_event(a.chan_man);
 287         if (!(event instanceof ldk.Event_FundingGenerationReady)) return false;
 288
 289         return true;
 290 });
 291
 292 tests.push(async () => {
 293         // Test passing onion messages through a custom trait implementation.
 294         const a = get_chanman();
 295         const b = get_chanman();
 296
 297         const ignorer = ldk.IgnoringMessageHandler.constructor_new();
 298
 299         var a_handled_msg = false;
 300         const om_handler_a = ldk.CustomOnionMessageHandler.new_impl({
 301                 read_custom_message(message_type: bigint, buffer: Uint8Array): ldk.Result_COption_CustomOnionMessageContentsZDecodeErrorZ {
 302                         console.assert(message_type == 4343n);
 303                         console.assert(buffer.length == 44);
 304                         for (var i = 0; i < 44; i++) console.assert(buffer[i] == 67);
 305                         return ldk.Result_COption_CustomOnionMessageContentsZDecodeErrorZ.constructor_ok(ldk.Option_CustomOnionMessageContentsZ.constructor_some(ldk.CustomOnionMessageContents.new_impl({
 306                                 tlv_type(): bigint { return 9998n; },
 307                                 write(): Uint8Array { console.assert(false); return null; }
 308                         } as ldk.CustomOnionMessageContentsInterface)));
 309                 },
 310                 handle_custom_message(msg: ldk.CustomOnionMessageContents) {
 311                         console.assert(msg.tlv_type() == 9998n);
 312                         a_handled_msg = true;
 313                 },
 314         } as ldk.CustomOnionMessageHandlerInterface);
 315
 316         const underlying_om_a = ldk.OnionMessenger.constructor_new(a.keys_interface, a.logger, om_handler_a);
 317         const om_provider_a = {
 318                 next_onion_message_for_peer(peer_node_id: Uint8Array): ldk.OnionMessage {
 319                         return underlying_om_a.as_OnionMessageProvider().next_onion_message_for_peer(peer_node_id);
 320                 }
 321         } as ldk.OnionMessageProviderInterface;
 322         const om_a = ldk.OnionMessageHandler.new_impl({
 323                 handle_onion_message(peer_node_id: Uint8Array, msg: ldk.OnionMessage) {
 324                         underlying_om_a.as_OnionMessageHandler().handle_onion_message(peer_node_id, msg);
 325                 },
 326                 peer_connected(their_node_id: Uint8Array, init: ldk.Init): ldk.Result_NoneNoneZ {
 327                         return underlying_om_a.as_OnionMessageHandler().peer_connected(their_node_id, init)
 328                 },
 329                 peer_disconnected(their_node_id: Uint8Array, no_connection_possible: boolean) {
 330                         underlying_om_a.as_OnionMessageHandler().peer_disconnected(their_node_id, no_connection_possible);
 331                 },
 332                 provided_node_features(): ldk.NodeFeatures {
 333                         return underlying_om_a.as_OnionMessageHandler().provided_node_features();
 334                 },
 335                 provided_init_features(their_node_id: Uint8Array): ldk.InitFeatures {
 336                         return underlying_om_a.as_OnionMessageHandler().provided_init_features(their_node_id);
 337                 }
 338         } as ldk.OnionMessageHandlerInterface, om_provider_a);
 339
 340         var b_handled_msg = false;
 341         const om_handler_b = ldk.CustomOnionMessageHandler.new_impl({
 342                 read_custom_message(message_type: bigint, buffer: Uint8Array): ldk.Result_COption_CustomOnionMessageContentsZDecodeErrorZ {
 343                         console.assert(message_type == 4242n);
 344                         console.assert(buffer.length == 43);
 345                         for (var i = 0; i < 43; i++) console.assert(buffer[i] == 66);
 346                         return ldk.Result_COption_CustomOnionMessageContentsZDecodeErrorZ.constructor_ok(ldk.Option_CustomOnionMessageContentsZ.constructor_some(ldk.CustomOnionMessageContents.new_impl({
 347                                 tlv_type(): bigint { return 9999n; },
 348                                 write(): Uint8Array { console.assert(false); return null; }
 349                         } as ldk.CustomOnionMessageContentsInterface)));
 350                 },
 351                 handle_custom_message(msg: ldk.CustomOnionMessageContents) {
 352                         console.assert(msg.tlv_type() == 9999n);
 353                         b_handled_msg = true;
 354                 },
 355         } as ldk.CustomOnionMessageHandlerInterface);
 356         const om_b = ldk.OnionMessenger.constructor_new(b.keys_interface, b.logger, om_handler_b);
 357
 358         const pm_a = ldk.PeerManager.constructor_new(a.chan_man.as_ChannelMessageHandler(), ignorer.as_RoutingMessageHandler(), om_a, a.node_secret, 0xdeadbeef, a.node_secret, a.logger, ignorer.as_CustomMessageHandler());
 359         const pm_b = ldk.PeerManager.constructor_new(b.chan_man.as_ChannelMessageHandler(), ignorer.as_RoutingMessageHandler(), om_b.as_OnionMessageHandler(), b.node_secret, 0xdeadbeef, b.node_secret, b.logger, ignorer.as_CustomMessageHandler());
 360
 361         var sock_b: ldk.SocketDescriptor;
 362         const sock_a = ldk.SocketDescriptor.new_impl({
 363                 send_data(data: Uint8Array, resume_read: boolean): number {
 364                         console.assert(pm_b.read_event(sock_b, data) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
 365                         return data.length;
 366                 },
 367                 disconnect_socket(): void {
 368                         console.assert(false);
 369                 },
 370                 eq(other: ldk.SocketDescriptor): boolean {
 371                         return other.hash() == this.hash();
 372                 },
 373                 hash(): bigint {
 374                         return BigInt(1);
 375                 }
 376         } as ldk.SocketDescriptorInterface);
 377         sock_b = ldk.SocketDescriptor.new_impl({
 378                 send_data(data: Uint8Array, resume_read: boolean): number {
 379                         console.assert(pm_a.read_event(sock_a, data) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
 380                         return data.length;
 381                 },
 382                 disconnect_socket(): void {
 383                         console.assert(false);
 384                 },
 385                 eq(other: ldk.SocketDescriptor): boolean {
 386                         return other.hash() == this.hash();
 387                 },
 388                 hash(): bigint {
 389                         return BigInt(2);
 390                 }
 391         } as ldk.SocketDescriptorInterface);
 392
 393         const v4_netaddr = ldk.NetAddress.constructor_ipv4(Uint8Array.from([42,0,42,1]), 9735);
 394         console.assert(pm_b.new_inbound_connection(sock_b, ldk.Option_NetAddressZ.constructor_some(v4_netaddr)) instanceof ldk.Result_NonePeerHandleErrorZ_OK);
 395         const init_bytes = pm_a.new_outbound_connection(b.node_id, sock_a, ldk.Option_NetAddressZ.constructor_none());
 396         if (!(init_bytes instanceof ldk.Result_CVec_u8ZPeerHandleErrorZ_OK)) return false;
 397         console.assert(pm_b.read_event(sock_b, init_bytes.res) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
 398
 399         console.assert(pm_a.get_peer_node_ids().length == 0);
 400         console.assert(pm_b.get_peer_node_ids().length == 0);
 401
 402         pm_b.process_events();
 403         pm_a.process_events();
 404         pm_b.process_events();
 405
 406         console.assert(pm_a.get_peer_node_ids().length == 1);
 407         console.assert(pm_b.get_peer_node_ids().length == 1);
 408
 409         underlying_om_a.send_custom_onion_message([], ldk.Destination.constructor_node(b.node_id), ldk.CustomOnionMessageContents.new_impl({
 410                         tlv_type(): bigint { return 4242n; },
 411                         write(): Uint8Array {
 412                                 const ret = new Uint8Array(43);
 413                                 for (var i = 0; i < 43; i++) ret[i] = 66;
 414                                 return ret;
 415                         }
 416                 } as ldk.CustomOnionMessageContentsInterface), null);
 417         pm_a.process_events();
 418         console.assert(b_handled_msg);
 419
 420         om_b.send_custom_onion_message([], ldk.Destination.constructor_node(a.node_id), ldk.CustomOnionMessageContents.new_impl({
 421                         tlv_type(): bigint { return 4343n; },
 422                         write(): Uint8Array {
 423                                 const ret = new Uint8Array(44);
 424                                 for (var i = 0; i < 44; i++) ret[i] = 67;
 425                                 return ret;
 426                         }
 427                 } as ldk.CustomOnionMessageContentsInterface), null);
 428         pm_b.process_events();
 429         console.assert(a_handled_msg);
 430
 431         return true;
 432 });
 433
 434 tests.push(async () => {
 435         // Test that we can do basic locking of a NetworkGraph
 436         const genesis_hash = new Uint8Array([0x00, 0x00, 0x00, 0x00, 0x00, 0x19, 0xd6, 0x68, 0x9c, 0x08, 0x5a, 0xe1, 0x65, 0x83, 0x1e, 0x93, 0x4f, 0xf7, 0x63, 0xae, 0x46, 0xa2, 0xa6, 0xc1, 0x72, 0xb3, 0xf1, 0xb6, 0x0a, 0x8c, 0xe2, 0x6f]);
 437         const logger = ldk.Logger.new_impl({
 438                 log(record: ldk.Record): void {
 439                         if (record.get_level() != ldk.Level.LDKLevel_Gossip)
 440                                 console.log(record.get_module_path() + ": " + record.get_args());
 441                 }
 442         } as ldk.LoggerInterface);
 443         const network_graph = ldk.NetworkGraph.constructor_new(genesis_hash, logger);
 444         const graph_lock_1 = network_graph.read_only();
 445         graph_lock_1.free();
 446         const graph_lock_2 = network_graph.read_only();
 447         graph_lock_2.free();
 448
 449         return true;
 450 });
 451
 452 async function run_tests(check_leaks: boolean) {
 453         var test_runs = [];
 454         for (const test of tests) {
 455                 test_runs.push(test());
 456         }
 457         const results = await Promise.all(test_runs);
 458         console.log("test results: " + results);
 459         const result = results.every((v) => { return v === true });
 460         console.log("all tests passed: " + result);
 461         if (result !== true || !check_leaks) { return result; }
 462
 463         const allocs_finished = new Promise((resolve, reject) => {
 464                 var loop_count = 0;
 465                 const interval_id = setInterval(() => {
 466                         const alloc_count = rawldk.getRemainingAllocationCount();
 467                         if (loop_count % 20 == 0)
 468                                 console.log("Remaining LDK allocation count: " + alloc_count);
 469
 470                         // chromium with --js-flags="--expose-gc" exposes a `window.gc()` which we call if we can
 471                         // @ts-ignore window.gc is considered a type error in TS
 472                         if (typeof window !== "undefined" && typeof window.gc !== "undefined") window.gc();
 473
 474                         // Note that there are currently 9 leaks in the above tests. At least some are known - look for XXX in bindings.c
 475                         if (alloc_count <= 10) { clearInterval(interval_id); rawldk.debugPrintRemainingAllocs(); resolve(true); }
 476                         loop_count += 1;
 477                         if (loop_count > 30*2) { clearInterval(interval_id); rawldk.debugPrintRemainingAllocs(); resolve(false); }
 478                 }, 500);
 479         });
 480         return allocs_finished;
 481 }
 482
 483 export async function run_tests_web(wasm_path: string, check_leaks: boolean = true) {
 484         await ldk.initializeWasmWebFetch(wasm_path);
 485         return await run_tests(check_leaks);
 486 }
 487
 488 export async function run_tests_node(wasm_file: Uint8Array, check_leaks: boolean = true) {
 489         await ldk.initializeWasmFromBinary(wasm_file);
 490         return await run_tests(check_leaks);
 491 }