[TS] Be more aggressive with type checks
[ldk-java] / ts / test / tests.mts
index 734ab66d3b12baf879a119905b0918c0b42f6e75..db0fe326618841a51b77104421565428272a66a1 100644 (file)
@@ -3,18 +3,21 @@ import * as ldk from "../index.mjs";
 
 const tests: Array<Function> = [];
 
+function assert(val: boolean) {
+       if (!val) throw new Error("Assertion failed");
+}
 function array_eq(a: Uint8Array, b: Uint8Array): boolean {
        return a.length == b.length && a.every((v, idx) => v == b[idx]);
 }
 
 tests.push(async () => {
        const result = rawldk.CResult_boolLightningErrorZ_ok(true);
-       console.assert(rawldk.CResult_boolLightningErrorZ_is_ok(result));
-       console.assert(rawldk.CResult_boolLightningErrorZ_get_ok(result));
+       assert(rawldk.CResult_boolLightningErrorZ_is_ok(result));
+       assert(rawldk.CResult_boolLightningErrorZ_get_ok(result));
        rawldk.CResult_boolLightningErrorZ_free(result);
        const second_res = rawldk.CResult_boolLightningErrorZ_ok(false);
-       console.assert(rawldk.CResult_boolLightningErrorZ_is_ok(second_res));
-       console.assert(!rawldk.CResult_boolLightningErrorZ_get_ok(second_res));
+       assert(rawldk.CResult_boolLightningErrorZ_is_ok(second_res));
+       assert(!rawldk.CResult_boolLightningErrorZ_get_ok(second_res));
        rawldk.CResult_boolLightningErrorZ_free(second_res);
 
        return true;
@@ -55,14 +58,23 @@ tests.push(async () => {
 });
 
 var seed_counter = 0;
-function get_chanman() {
+class Node {
+       node_id: Uint8Array;
+       node_secret: Uint8Array;
+       constructor(public chan_man: ldk.ChannelManager, public tx_broadcasted: Promise<Uint8Array>,
+       public logger: ldk.Logger, public keys_interface: ldk.KeysInterface) {
+               this.node_id = chan_man.get_our_node_id();
+               this.node_secret = (keys_interface.get_node_secret(ldk.Recipient.LDKRecipient_Node) as ldk.Result_SecretKeyNoneZ_OK).res;
+       }
+}
+function get_chanman(): Node {
        const fee_est = ldk.FeeEstimator.new_impl({
-               get_est_sat_per_1000_weight(confirmation_target: ldk.ConfirmationTarget): number {
+               get_est_sat_per_1000_weight(_confirmation_target: ldk.ConfirmationTarget): number {
                        return 253;
                }
        } as ldk.FeeEstimatorInterface);
-       var tx_broadcaster;
-       const tx_broadcasted: Promise<Uint8Array> = new Promise((resolve, reject) => {
+       var tx_broadcaster: ldk.BroadcasterInterface;
+       const tx_broadcasted: Promise<Uint8Array> = new Promise((resolve, _reject) => {
                tx_broadcaster = ldk.BroadcasterInterface.new_impl({
                        broadcast_transaction(tx: Uint8Array): void { console.log("Tx Broadcast: " + tx); resolve(tx); }
                } as ldk.BroadcasterInterfaceInterface);
@@ -74,15 +86,15 @@ function get_chanman() {
        } as ldk.LoggerInterface);
 
        const persister = ldk.Persist.new_impl({
-               persist_new_channel(channel_id: ldk.OutPoint, data: ldk.ChannelMonitor, update_id: ldk.MonitorUpdateId): ldk.Result_NoneChannelMonitorUpdateErrZ {
-                       return ldk.Result_NoneChannelMonitorUpdateErrZ.constructor_ok();
+               persist_new_channel(_channel_id: ldk.OutPoint, _data: ldk.ChannelMonitor, _update_id: ldk.MonitorUpdateId): ldk.ChannelMonitorUpdateStatus {
+                       return ldk.ChannelMonitorUpdateStatus.LDKChannelMonitorUpdateStatus_Completed;
                },
-               update_persisted_channel(channel_id: ldk.OutPoint, update: ldk.ChannelMonitorUpdate, data: ldk.ChannelMonitor, update_id: ldk.MonitorUpdateId): ldk.Result_NoneChannelMonitorUpdateErrZ {
-                       return ldk.Result_NoneChannelMonitorUpdateErrZ.constructor_ok();
+               update_persisted_channel(_channel_id: ldk.OutPoint, _update: ldk.ChannelMonitorUpdate, _data: ldk.ChannelMonitor, _update_id: ldk.MonitorUpdateId): ldk.ChannelMonitorUpdateStatus {
+                       return ldk.ChannelMonitorUpdateStatus.LDKChannelMonitorUpdateStatus_Completed;
                }
        } as ldk.PersistInterface);
 
-       const chain_monitor = ldk.ChainMonitor.constructor_new(ldk.Option_FilterZ.constructor_none(), tx_broadcaster, logger, fee_est, persister);
+       const chain_monitor = ldk.ChainMonitor.constructor_new(ldk.Option_FilterZ.constructor_none(), tx_broadcaster!, logger, fee_est, persister);
        const chain_watch: ldk.Watch = chain_monitor.as_Watch();
 
        const seed = new Uint8Array(32);
@@ -93,7 +105,8 @@ function get_chanman() {
        const config = ldk.UserConfig.constructor_default();
        const params = ldk.ChainParameters.constructor_new(ldk.Network.LDKNetwork_Testnet, ldk.BestBlock.constructor_from_genesis(ldk.Network.LDKNetwork_Testnet));
 
-       return [ldk.ChannelManager.constructor_new(fee_est, chain_watch, tx_broadcaster, logger, keys_interface, config, params), tx_broadcasted];
+       const chan_man = ldk.ChannelManager.constructor_new(fee_est, chain_watch, tx_broadcaster!, logger, keys_interface, config, params);
+       return new Node(chan_man, tx_broadcasted, logger, keys_interface);
 }
 
 function exchange_messages(a: ldk.ChannelManager, b: ldk.ChannelManager) {
@@ -109,10 +122,12 @@ function exchange_messages(a: ldk.ChannelManager, b: ldk.ChannelManager) {
                                const msg = msgs[j];
                                if (msg instanceof ldk.MessageSendEvent_SendOpenChannel) {
                                        if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
-                                       to.as_ChannelMessageHandler().handle_open_channel(from.get_our_node_id(), ldk.InitFeatures.constructor_known(), msg.msg);
+                                       const features = from.as_ChannelMessageHandler().provided_init_features(to.get_our_node_id());
+                                       to.as_ChannelMessageHandler().handle_open_channel(from.get_our_node_id(), features, msg.msg);
                                } else if (msg instanceof ldk.MessageSendEvent_SendAcceptChannel) {
                                        if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
-                                       to.as_ChannelMessageHandler().handle_accept_channel(from.get_our_node_id(), ldk.InitFeatures.constructor_known(), msg.msg);
+                                       const features = from.as_ChannelMessageHandler().provided_init_features(to.get_our_node_id());
+                                       to.as_ChannelMessageHandler().handle_accept_channel(from.get_our_node_id(), features, msg.msg);
                                } else if (msg instanceof ldk.MessageSendEvent_SendFundingCreated) {
                                        if (!array_eq(msg.node_id, to.get_our_node_id())) return false;
                                        to.as_ChannelMessageHandler().handle_funding_created(from.get_our_node_id(), msg.msg);
@@ -139,39 +154,43 @@ function assign_u64(arr: Uint8Array, offset: number, value: bigint) {
        arr[offset + 7] = Number((value >> BigInt(8 * 7)) & BigInt(0xff));
 }
 
+function get_event(chan_man: ldk.ChannelManager): ldk.Event {
+       const events: Array<ldk.Event> = [];
+       const event_handler = ldk.EventHandler.new_impl({
+               handle_event(event: ldk.Event): void {
+                       events.push(event);
+               }
+       } as ldk.EventHandlerInterface);
+
+       chan_man.as_EventsProvider().process_pending_events(event_handler);
+       assert(events.length == 1);
+       return events[0]!;
+}
+
 tests.push(async () => {
-       const peer_a = get_chanman();
-       const peer_b = get_chanman();
-       const chan_man_a: ldk.ChannelManager = peer_a[0] as ldk.ChannelManager;
-       const chan_man_b: ldk.ChannelManager = peer_b[0] as ldk.ChannelManager;
+       const a = get_chanman();
+       const b = get_chanman();
+
+       const features = a.chan_man.as_ChannelMessageHandler().provided_init_features(b.chan_man.get_our_node_id());
 
-       chan_man_a.as_ChannelMessageHandler().peer_connected(chan_man_b.get_our_node_id(), ldk.Init.constructor_new(ldk.InitFeatures.constructor_known(), ldk.Option_NetAddressZ.constructor_none()));
-       chan_man_b.as_ChannelMessageHandler().peer_connected(chan_man_a.get_our_node_id(), ldk.Init.constructor_new(ldk.InitFeatures.constructor_known(), ldk.Option_NetAddressZ.constructor_none()));
+       a.chan_man.as_ChannelMessageHandler().peer_connected(b.chan_man.get_our_node_id(), ldk.Init.constructor_new(features, ldk.Option_NetAddressZ.constructor_none()));
+       b.chan_man.as_ChannelMessageHandler().peer_connected(a.chan_man.get_our_node_id(), ldk.Init.constructor_new(features, ldk.Option_NetAddressZ.constructor_none()));
 
-       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());
+       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());
        if (chan_create_err.is_ok()) return false;
        if (!(chan_create_err instanceof ldk.Result__u832APIErrorZ_Err)) return false;
        if (!(chan_create_err.err instanceof ldk.APIError_APIMisuseError)) return false;
        if (chan_create_err.err.err != "Channel value must be at least 1000 satoshis. It was 0") return false;
 
-       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());
+       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());
        if (!chan_create_res.is_ok()) return false;
 
-       if (!exchange_messages(chan_man_a, chan_man_b)) return false;
-
-       const events: Array<ldk.Event> = [];
-       const event_handler = ldk.EventHandler.new_impl({
-               handle_event(event: ldk.Event): void {
-                       events.push(event);
-               }
-       } as ldk.EventHandlerInterface);
+       if (!exchange_messages(a.chan_man, b.chan_man)) return false;
 
-       chan_man_a.as_EventsProvider().process_pending_events(event_handler);
-       if (events.length != 1) return false;
-       if (!(events[0] instanceof ldk.Event_FundingGenerationReady)) return false;
+       const event = get_event(a.chan_man) as ldk.Event_FundingGenerationReady;
 
        // (very) manually create a funding transaction
-       const witness_pos = events[0].output_script.length + 58;
+       const witness_pos = event.output_script.length + 58;
        const funding_tx = new Uint8Array(witness_pos + 7);
        funding_tx[0] = 2; // 4-byte tx version 2
        funding_tx[4] = 0; funding_tx[5] = 1; // segwit magic bytes
@@ -180,23 +199,263 @@ tests.push(async () => {
        funding_tx[43] = 0; // 1-byte input script length 0
        funding_tx[44] = 0xff; funding_tx[45] = 0xff; funding_tx[46] = 0xff; funding_tx[47] = 0xff; // 4-byte nSequence
        funding_tx[48] = 1; // one output
-       assign_u64(funding_tx, 49, events[0].channel_value_satoshis);
-       funding_tx[57] = events[0].output_script.length; // 1-byte output script length
-       funding_tx.set(events[0].output_script, 58);
+       assign_u64(funding_tx, 49, event.channel_value_satoshis);
+       funding_tx[57] = event.output_script.length; // 1-byte output script length
+       funding_tx.set(event.output_script, 58);
        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
        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
 
-       const funding_res = chan_man_a.funding_transaction_generated(events[0].temporary_channel_id, events[0].counterparty_node_id, funding_tx);
+       const funding_res = a.chan_man.funding_transaction_generated(event.temporary_channel_id, event.counterparty_node_id, funding_tx);
        if (!(funding_res instanceof ldk.Result_NoneAPIErrorZ_OK)) return false;
 
-       if (!exchange_messages(chan_man_a, chan_man_b)) return false;
+       if (!exchange_messages(a.chan_man, b.chan_man)) return false;
 
-       const tx_broadcasted: Uint8Array = (await peer_a[1]) as Uint8Array;
+       const tx_broadcasted: Uint8Array = (await a.tx_broadcasted);
        if (!array_eq(tx_broadcasted, funding_tx)) return false;
 
        return true;
 });
 
+tests.push(async () => {
+       const a = get_chanman();
+       const b = get_chanman();
+
+       const ignorer = ldk.IgnoringMessageHandler.constructor_new();
+       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());
+       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());
+
+       var sock_b: ldk.SocketDescriptor;
+       const sock_a = ldk.SocketDescriptor.new_impl({
+               send_data(data: Uint8Array, resume_read: boolean): number {
+                       assert(pm_b.read_event(sock_b, data) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
+                       assert(resume_read);
+                       return data.length;
+               },
+               disconnect_socket(): void {
+                       assert(false);
+               },
+               eq(other: ldk.SocketDescriptor): boolean {
+                       return other.hash() == this.hash();
+               },
+               hash(): bigint {
+                       return BigInt(1);
+               }
+       } as ldk.SocketDescriptorInterface);
+       sock_b = ldk.SocketDescriptor.new_impl({
+               send_data(data: Uint8Array, resume_read: boolean): number {
+                       assert(pm_a.read_event(sock_a, data) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
+                       assert(resume_read);
+                       return data.length;
+               },
+               disconnect_socket(): void {
+                       assert(false);
+               },
+               eq(other: ldk.SocketDescriptor): boolean {
+                       return other.hash() == this.hash();
+               },
+               hash(): bigint {
+                       return BigInt(2);
+               }
+       } as ldk.SocketDescriptorInterface);
+
+       const update_fut = a.chan_man.get_persistable_update_future();
+       var update_done = false;
+       update_fut.register_callback_fn(ldk.FutureCallback.new_impl({
+               call(): void { update_done = true; }
+       }));
+       if (update_done) return false;
+
+       const v4_netaddr = ldk.NetAddress.constructor_ipv4(Uint8Array.from([42,0,42,1]), 9735);
+       assert(pm_b.new_inbound_connection(sock_b, ldk.Option_NetAddressZ.constructor_some(v4_netaddr)) instanceof ldk.Result_NonePeerHandleErrorZ_OK);
+       const init_bytes = pm_a.new_outbound_connection(b.node_id, sock_a, ldk.Option_NetAddressZ.constructor_none());
+       if (!(init_bytes instanceof ldk.Result_CVec_u8ZPeerHandleErrorZ_OK)) return false;
+       assert(pm_b.read_event(sock_b, init_bytes.res) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
+
+       assert(pm_a.get_peer_node_ids().length == 0);
+       assert(pm_b.get_peer_node_ids().length == 0);
+
+       pm_b.process_events();
+       pm_a.process_events();
+       pm_b.process_events();
+
+       assert(pm_a.get_peer_node_ids().length == 1);
+       assert(pm_b.get_peer_node_ids().length == 1);
+
+       const chan_create_res = a.chan_man.create_channel(b.node_id, BigInt(1000000), BigInt(400), BigInt(0), ldk.UserConfig.constructor_default());
+       if (!chan_create_res.is_ok()) return false;
+       if (!update_done) return false;
+
+       pm_a.process_events();
+       pm_b.process_events();
+
+       const event = get_event(a.chan_man);
+       if (!(event instanceof ldk.Event_FundingGenerationReady)) return false;
+
+       return true;
+});
+
+tests.push(async () => {
+       // Test passing onion messages through a custom trait implementation.
+       const a = get_chanman();
+       const b = get_chanman();
+
+       const ignorer = ldk.IgnoringMessageHandler.constructor_new();
+
+       var a_handled_msg = false;
+       const om_handler_a = ldk.CustomOnionMessageHandler.new_impl({
+               read_custom_message(message_type: bigint, buffer: Uint8Array): ldk.Result_COption_CustomOnionMessageContentsZDecodeErrorZ {
+                       assert(message_type == 4343n);
+                       assert(buffer.length == 44);
+                       for (var i = 0; i < 44; i++) assert(buffer[i] == 67);
+                       return ldk.Result_COption_CustomOnionMessageContentsZDecodeErrorZ.constructor_ok(ldk.Option_CustomOnionMessageContentsZ.constructor_some(ldk.CustomOnionMessageContents.new_impl({
+                               tlv_type(): bigint { return 9998n; },
+                               write(): Uint8Array { throw new Error(); }
+                       } as ldk.CustomOnionMessageContentsInterface)));
+               },
+               handle_custom_message(msg: ldk.CustomOnionMessageContents) {
+                       assert(msg.tlv_type() == 9998n);
+                       a_handled_msg = true;
+               },
+       } as ldk.CustomOnionMessageHandlerInterface);
+
+       const underlying_om_a = ldk.OnionMessenger.constructor_new(a.keys_interface, a.logger, om_handler_a);
+       const om_provider_a = {
+               next_onion_message_for_peer(peer_node_id: Uint8Array): ldk.OnionMessage {
+                       return underlying_om_a.as_OnionMessageProvider().next_onion_message_for_peer(peer_node_id);
+               }
+       } as ldk.OnionMessageProviderInterface;
+       const om_a = ldk.OnionMessageHandler.new_impl({
+               handle_onion_message(peer_node_id: Uint8Array, msg: ldk.OnionMessage) {
+                       underlying_om_a.as_OnionMessageHandler().handle_onion_message(peer_node_id, msg);
+               },
+               peer_connected(their_node_id: Uint8Array, init: ldk.Init): ldk.Result_NoneNoneZ {
+                       return underlying_om_a.as_OnionMessageHandler().peer_connected(their_node_id, init)
+               },
+               peer_disconnected(their_node_id: Uint8Array, no_connection_possible: boolean) {
+                       underlying_om_a.as_OnionMessageHandler().peer_disconnected(their_node_id, no_connection_possible);
+               },
+               provided_node_features(): ldk.NodeFeatures {
+                       return underlying_om_a.as_OnionMessageHandler().provided_node_features();
+               },
+               provided_init_features(their_node_id: Uint8Array): ldk.InitFeatures {
+                       return underlying_om_a.as_OnionMessageHandler().provided_init_features(their_node_id);
+               }
+       } as ldk.OnionMessageHandlerInterface, om_provider_a);
+
+       var b_handled_msg = false;
+       const om_handler_b = ldk.CustomOnionMessageHandler.new_impl({
+               read_custom_message(message_type: bigint, buffer: Uint8Array): ldk.Result_COption_CustomOnionMessageContentsZDecodeErrorZ {
+                       assert(message_type == 4242n);
+                       assert(buffer.length == 43);
+                       for (var i = 0; i < 43; i++) assert(buffer[i] == 66);
+                       return ldk.Result_COption_CustomOnionMessageContentsZDecodeErrorZ.constructor_ok(ldk.Option_CustomOnionMessageContentsZ.constructor_some(ldk.CustomOnionMessageContents.new_impl({
+                               tlv_type(): bigint { return 9999n; },
+                               write(): Uint8Array { throw new Error(); }
+                       } as ldk.CustomOnionMessageContentsInterface)));
+               },
+               handle_custom_message(msg: ldk.CustomOnionMessageContents) {
+                       assert(msg.tlv_type() == 9999n);
+                       b_handled_msg = true;
+               },
+       } as ldk.CustomOnionMessageHandlerInterface);
+       const om_b = ldk.OnionMessenger.constructor_new(b.keys_interface, b.logger, om_handler_b);
+
+       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());
+       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());
+
+       var sock_b: ldk.SocketDescriptor;
+       const sock_a = ldk.SocketDescriptor.new_impl({
+               send_data(data: Uint8Array, resume_read: boolean): number {
+                       assert(pm_b.read_event(sock_b, data) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
+                       assert(resume_read);
+                       return data.length;
+               },
+               disconnect_socket(): void {
+                       assert(false);
+               },
+               eq(other: ldk.SocketDescriptor): boolean {
+                       return other.hash() == this.hash();
+               },
+               hash(): bigint {
+                       return BigInt(1);
+               }
+       } as ldk.SocketDescriptorInterface);
+       sock_b = ldk.SocketDescriptor.new_impl({
+               send_data(data: Uint8Array, resume_read: boolean): number {
+                       assert(pm_a.read_event(sock_a, data) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
+                       assert(resume_read);
+                       return data.length;
+               },
+               disconnect_socket(): void {
+                       assert(false);
+               },
+               eq(other: ldk.SocketDescriptor): boolean {
+                       return other.hash() == this.hash();
+               },
+               hash(): bigint {
+                       return BigInt(2);
+               }
+       } as ldk.SocketDescriptorInterface);
+
+       const v4_netaddr = ldk.NetAddress.constructor_ipv4(Uint8Array.from([42,0,42,1]), 9735);
+       assert(pm_b.new_inbound_connection(sock_b, ldk.Option_NetAddressZ.constructor_some(v4_netaddr)) instanceof ldk.Result_NonePeerHandleErrorZ_OK);
+       const init_bytes = pm_a.new_outbound_connection(b.node_id, sock_a, ldk.Option_NetAddressZ.constructor_none());
+       if (!(init_bytes instanceof ldk.Result_CVec_u8ZPeerHandleErrorZ_OK)) return false;
+       assert(pm_b.read_event(sock_b, init_bytes.res) instanceof ldk.Result_boolPeerHandleErrorZ_OK);
+
+       assert(pm_a.get_peer_node_ids().length == 0);
+       assert(pm_b.get_peer_node_ids().length == 0);
+
+       pm_b.process_events();
+       pm_a.process_events();
+       pm_b.process_events();
+
+       assert(pm_a.get_peer_node_ids().length == 1);
+       assert(pm_b.get_peer_node_ids().length == 1);
+
+       underlying_om_a.send_custom_onion_message([], ldk.Destination.constructor_node(b.node_id), ldk.CustomOnionMessageContents.new_impl({
+                       tlv_type(): bigint { return 4242n; },
+                       write(): Uint8Array {
+                               const ret = new Uint8Array(43);
+                               for (var i = 0; i < 43; i++) ret[i] = 66;
+                               return ret;
+                       }
+               } as ldk.CustomOnionMessageContentsInterface), null);
+       pm_a.process_events();
+       assert(b_handled_msg);
+
+       om_b.send_custom_onion_message([], ldk.Destination.constructor_node(a.node_id), ldk.CustomOnionMessageContents.new_impl({
+                       tlv_type(): bigint { return 4343n; },
+                       write(): Uint8Array {
+                               const ret = new Uint8Array(44);
+                               for (var i = 0; i < 44; i++) ret[i] = 67;
+                               return ret;
+                       }
+               } as ldk.CustomOnionMessageContentsInterface), null);
+       pm_b.process_events();
+       assert(a_handled_msg);
+
+       return true;
+});
+
+tests.push(async () => {
+       // Test that we can do basic locking of a NetworkGraph
+       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]);
+       const logger = ldk.Logger.new_impl({
+               log(record: ldk.Record): void {
+                       if (record.get_level() != ldk.Level.LDKLevel_Gossip)
+                               console.log(record.get_module_path() + ": " + record.get_args());
+               }
+       } as ldk.LoggerInterface);
+       const network_graph = ldk.NetworkGraph.constructor_new(genesis_hash, logger);
+       const graph_lock_1 = network_graph.read_only();
+       graph_lock_1.free();
+       const graph_lock_2 = network_graph.read_only();
+       graph_lock_2.free();
+
+       return true;
+});
+
 async function run_tests(check_leaks: boolean) {
        var test_runs = [];
        for (const test of tests) {
@@ -208,7 +467,7 @@ async function run_tests(check_leaks: boolean) {
        console.log("all tests passed: " + result);
        if (result !== true || !check_leaks) { return result; }
 
-       const allocs_finished = new Promise((resolve, reject) => {
+       const allocs_finished = new Promise((resolve, _reject) => {
                var loop_count = 0;
                const interval_id = setInterval(() => {
                        const alloc_count = rawldk.getRemainingAllocationCount();