Manually patch identify protocol

This commit is contained in:
Franck Royer 2021-03-15 17:29:48 +11:00
parent fe7daebbb0
commit f0233f7696
No known key found for this signature in database
GPG Key ID: A82ED75A8DFC50A4
1 changed files with 18 additions and 4 deletions

View File

@ -1,4 +1,5 @@
import test from 'ava'; import test from 'ava';
import Libp2p from 'libp2p';
import Pubsub from 'libp2p-interfaces/src/pubsub'; import Pubsub from 'libp2p-interfaces/src/pubsub';
import { NimWaku } from '../test_utils/nim_waku'; import { NimWaku } from '../test_utils/nim_waku';
@ -110,10 +111,7 @@ test('Nim-interop: js node sends message to nim node', async (t) => {
const nimWaku = new NimWaku(t.title); const nimWaku = new NimWaku(t.title);
await nimWaku.start({ staticnode: multiAddrWithId }); await nimWaku.start({ staticnode: multiAddrWithId });
// TODO: Remove this hack, tracked with https://github.com/status-im/nim-waku/issues/419 await patchPeerStore(nimWaku, node);
node.identifyService!.peerStore.protoBook.set(await nimWaku.getPeerId(), [
CODEC,
]);
await wakuRelayNode.publish(message); await wakuRelayNode.publish(message);
@ -135,3 +133,19 @@ function waitForNextData(pubsub: Pubsub): Promise<Message> {
return Message.fromBinary(msg.data); return Message.fromBinary(msg.data);
}); });
} }
// TODO: Remove this hack, tracked with https://github.com/status-im/nim-waku/issues/419
async function patchPeerStore(nimWaku: NimWaku, node: Libp2p) {
const nimPeerId = await nimWaku.getPeerId();
node.identifyService!.peerStore.protoBook.set(nimPeerId, [CODEC]);
const peer = node.peerStore.peers.get(nimPeerId.toB58String());
if (!peer) {
throw 'Did not find nim-waku node in peers';
}
peer.protocols = [CODEC];
node.peerStore.peers.set(nimPeerId.toB58String(), peer);
await new Promise((resolve) =>
node.pubsub.once('gossipsub:heartbeat', resolve)
);
}