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