mirror of https://github.com/waku-org/js-waku.git
Move waku relay tests in dedicated file
This commit is contained in:
Franck Royer
parent
e04988d98d
commit
7d595b0c8f
|
|
@ -1,57 +1,7 @@
|
|||
import 'jest';
|
||||
import Libp2p from 'libp2p';
|
||||
import Pubsub from 'libp2p-interfaces/src/pubsub';
|
||||
|
||||
import { NimWaku } from '../test_utils/nim_waku';
|
||||
|
||||
import Waku from './waku';
|
||||
import { Message } from './waku_message';
|
||||
import { CODEC, TOPIC } from './waku_relay';
|
||||
|
||||
// TODO: Fix this, see https://github.com/ChainSafe/js-libp2p-gossipsub/issues/151
|
||||
test.skip('Publishes message', async () => {
|
||||
const message = Message.fromUtf8String('Bird bird bird, bird is the word!');
|
||||
|
||||
const [waku1, waku2] = await Promise.all([Waku.create(), Waku.create()]);
|
||||
|
||||
// Add node's 2 data to the PeerStore
|
||||
waku1.libp2p.peerStore.addressBook.set(
|
||||
waku2.libp2p.peerId,
|
||||
waku2.libp2p.multiaddrs
|
||||
);
|
||||
await waku1.libp2p.dial(waku2.libp2p.peerId);
|
||||
|
||||
await waku2.relay.subscribe();
|
||||
await new Promise((resolve) =>
|
||||
waku2.libp2p.pubsub.once('pubsub:subscription-change', (...args) =>
|
||||
resolve(args)
|
||||
)
|
||||
);
|
||||
|
||||
// Setup the promise before publishing to ensure the event is not missed
|
||||
const promise = waitForNextData(waku1.libp2p.pubsub);
|
||||
|
||||
await waku2.relay.publish(message);
|
||||
|
||||
const node1Received = await promise;
|
||||
|
||||
expect(node1Received.isEqualTo(message)).toBeTruthy();
|
||||
});
|
||||
|
||||
test('Registers waku relay protocol', async () => {
|
||||
const waku = await Waku.create();
|
||||
|
||||
const protocols = Array.from(waku.libp2p.upgrader.protocols.keys());
|
||||
|
||||
expect(protocols.findIndex((value) => value == CODEC)).toBeTruthy();
|
||||
});
|
||||
|
||||
test('Does not register any sub protocol', async () => {
|
||||
const waku = await Waku.create();
|
||||
|
||||
const protocols = Array.from(waku.libp2p.upgrader.protocols.keys());
|
||||
expect(protocols.findIndex((value) => value.match(/sub/))).toBeTruthy();
|
||||
});
|
||||
import { CODEC } from './waku_relay';
|
||||
|
||||
test('Nim-interop: nim-waku node connects to js node', async () => {
|
||||
const waku = await Waku.create();
|
||||
|
|
@ -81,113 +31,3 @@ test('Nim-interop: nim-waku node connects to js node', async () => {
|
|||
|
||||
expect(jsPeers.has(nimPeerId.toB58String())).toBeTruthy();
|
||||
});
|
||||
|
||||
test('Nim-interop: js node receives default subscription from nim node', async () => {
|
||||
const waku = await Waku.create();
|
||||
|
||||
const peerId = waku.libp2p.peerId.toB58String();
|
||||
|
||||
const localMultiaddr = waku.libp2p.multiaddrs.find((addr) =>
|
||||
addr.toString().match(/127\.0\.0\.1/)
|
||||
);
|
||||
const multiAddrWithId = localMultiaddr + '/p2p/' + peerId;
|
||||
|
||||
const nimWaku = new NimWaku(expect.getState().currentTestName);
|
||||
await nimWaku.start({ staticnode: multiAddrWithId });
|
||||
|
||||
const nimPeerId = await nimWaku.getPeerId();
|
||||
const subscribers = waku.libp2p.pubsub.getSubscribers(TOPIC);
|
||||
|
||||
expect(subscribers).toContain(nimPeerId.toB58String());
|
||||
});
|
||||
|
||||
test('Nim-interop: js node sends message to nim node', async () => {
|
||||
const message = Message.fromUtf8String('This is a message');
|
||||
const waku = await Waku.create();
|
||||
|
||||
// TODO: nim-waku does follow the `StrictNoSign` policy hence we need to change
|
||||
// it for nim-waku to process our messages. Can be removed once
|
||||
// https://github.com/status-im/nim-waku/issues/422 is fixed
|
||||
waku.libp2p.pubsub.globalSignaturePolicy = 'StrictSign';
|
||||
|
||||
const peerId = waku.libp2p.peerId.toB58String();
|
||||
const localMultiaddr = waku.libp2p.multiaddrs.find((addr) =>
|
||||
addr.toString().match(/127\.0\.0\.1/)
|
||||
);
|
||||
const multiAddrWithId = localMultiaddr + '/p2p/' + peerId;
|
||||
|
||||
const nimWaku = new NimWaku(expect.getState().currentTestName);
|
||||
await nimWaku.start({ staticnode: multiAddrWithId });
|
||||
|
||||
await patchPeerStore(nimWaku, waku.libp2p);
|
||||
|
||||
await waku.relay.publish(message);
|
||||
|
||||
await nimWaku.waitForLog('WakuMessage received');
|
||||
|
||||
const msgs = await nimWaku.messages();
|
||||
|
||||
expect(msgs[0].contentTopic).toEqual(message.contentTopic);
|
||||
expect(msgs[0].version).toEqual(message.version);
|
||||
|
||||
const payload = Buffer.from(msgs[0].payload);
|
||||
expect(Buffer.compare(payload, message.payload)).toBe(0);
|
||||
});
|
||||
|
||||
test('Nim-interop: nim node sends message to js node', async () => {
|
||||
const message = Message.fromUtf8String('Here is another message.');
|
||||
const waku = await Waku.create();
|
||||
|
||||
const peerId = waku.libp2p.peerId.toB58String();
|
||||
const localMultiaddr = waku.libp2p.multiaddrs.find((addr) =>
|
||||
addr.toString().match(/127\.0\.0\.1/)
|
||||
);
|
||||
const multiAddrWithId = localMultiaddr + '/p2p/' + peerId;
|
||||
|
||||
const nimWaku = new NimWaku(expect.getState().currentTestName);
|
||||
await nimWaku.start({ staticnode: multiAddrWithId });
|
||||
|
||||
await patchPeerStore(nimWaku, waku.libp2p);
|
||||
|
||||
await waku.relay.subscribe();
|
||||
|
||||
await new Promise((resolve) =>
|
||||
waku.libp2p.pubsub.once('gossipsub:heartbeat', resolve)
|
||||
);
|
||||
|
||||
const receivedPromise = waitForNextData(waku.libp2p.pubsub);
|
||||
|
||||
await nimWaku.sendMessage(message);
|
||||
|
||||
const receivedMsg = await receivedPromise;
|
||||
|
||||
expect(receivedMsg.contentTopic).toBe(message.contentTopic);
|
||||
expect(receivedMsg.version).toBe(message.version);
|
||||
|
||||
const payload = Buffer.from(receivedMsg.payload);
|
||||
expect(Buffer.compare(payload, message.payload)).toBe(0);
|
||||
});
|
||||
|
||||
function waitForNextData(pubsub: Pubsub): Promise<Message> {
|
||||
return new Promise((resolve) => {
|
||||
pubsub.once(TOPIC, resolve);
|
||||
}).then((msg: any) => {
|
||||
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)
|
||||
);
|
||||
}
|
||||
|
|
|
|||
|
|
@ -0,0 +1,164 @@
|
|||
import 'jest';
|
||||
import Libp2p from 'libp2p';
|
||||
import Pubsub from 'libp2p-interfaces/src/pubsub';
|
||||
|
||||
import { NimWaku } from '../test_utils/nim_waku';
|
||||
|
||||
import Waku from './waku';
|
||||
import { Message } from './waku_message';
|
||||
import { CODEC, TOPIC } from './waku_relay';
|
||||
|
||||
// TODO: Fix this, see https://github.com/ChainSafe/js-libp2p-gossipsub/issues/151
|
||||
test.skip('Publishes message', async () => {
|
||||
const message = Message.fromUtf8String('Bird bird bird, bird is the word!');
|
||||
|
||||
const [waku1, waku2] = await Promise.all([Waku.create(), Waku.create()]);
|
||||
|
||||
// Add node's 2 data to the PeerStore
|
||||
waku1.libp2p.peerStore.addressBook.set(
|
||||
waku2.libp2p.peerId,
|
||||
waku2.libp2p.multiaddrs
|
||||
);
|
||||
await waku1.libp2p.dial(waku2.libp2p.peerId);
|
||||
|
||||
await waku2.relay.subscribe();
|
||||
await new Promise((resolve) =>
|
||||
waku2.libp2p.pubsub.once('pubsub:subscription-change', (...args) =>
|
||||
resolve(args)
|
||||
)
|
||||
);
|
||||
|
||||
// Setup the promise before publishing to ensure the event is not missed
|
||||
const promise = waitForNextData(waku1.libp2p.pubsub);
|
||||
|
||||
await waku2.relay.publish(message);
|
||||
|
||||
const node1Received = await promise;
|
||||
|
||||
expect(node1Received.isEqualTo(message)).toBeTruthy();
|
||||
});
|
||||
|
||||
test('Registers waku relay protocol', async () => {
|
||||
const waku = await Waku.create();
|
||||
|
||||
const protocols = Array.from(waku.libp2p.upgrader.protocols.keys());
|
||||
|
||||
expect(protocols.findIndex((value) => value == CODEC)).toBeTruthy();
|
||||
});
|
||||
|
||||
test('Does not register any sub protocol', async () => {
|
||||
const waku = await Waku.create();
|
||||
|
||||
const protocols = Array.from(waku.libp2p.upgrader.protocols.keys());
|
||||
expect(protocols.findIndex((value) => value.match(/sub/))).toBeTruthy();
|
||||
});
|
||||
|
||||
test('Nim-interop: js node receives default subscription from nim node', async () => {
|
||||
const waku = await Waku.create();
|
||||
|
||||
const peerId = waku.libp2p.peerId.toB58String();
|
||||
|
||||
const localMultiaddr = waku.libp2p.multiaddrs.find((addr) =>
|
||||
addr.toString().match(/127\.0\.0\.1/)
|
||||
);
|
||||
const multiAddrWithId = localMultiaddr + '/p2p/' + peerId;
|
||||
|
||||
const nimWaku = new NimWaku(expect.getState().currentTestName);
|
||||
await nimWaku.start({ staticnode: multiAddrWithId });
|
||||
|
||||
const nimPeerId = await nimWaku.getPeerId();
|
||||
const subscribers = waku.libp2p.pubsub.getSubscribers(TOPIC);
|
||||
|
||||
expect(subscribers).toContain(nimPeerId.toB58String());
|
||||
});
|
||||
|
||||
test('Nim-interop: js node sends message to nim node', async () => {
|
||||
const message = Message.fromUtf8String('This is a message');
|
||||
const waku = await Waku.create();
|
||||
|
||||
// TODO: nim-waku does follow the `StrictNoSign` policy hence we need to change
|
||||
// it for nim-waku to process our messages. Can be removed once
|
||||
// https://github.com/status-im/nim-waku/issues/422 is fixed
|
||||
waku.libp2p.pubsub.globalSignaturePolicy = 'StrictSign';
|
||||
|
||||
const peerId = waku.libp2p.peerId.toB58String();
|
||||
const localMultiaddr = waku.libp2p.multiaddrs.find((addr) =>
|
||||
addr.toString().match(/127\.0\.0\.1/)
|
||||
);
|
||||
const multiAddrWithId = localMultiaddr + '/p2p/' + peerId;
|
||||
|
||||
const nimWaku = new NimWaku(expect.getState().currentTestName);
|
||||
await nimWaku.start({ staticnode: multiAddrWithId });
|
||||
|
||||
await patchPeerStore(nimWaku, waku.libp2p);
|
||||
|
||||
await waku.relay.publish(message);
|
||||
|
||||
await nimWaku.waitForLog('WakuMessage received');
|
||||
|
||||
const msgs = await nimWaku.messages();
|
||||
|
||||
expect(msgs[0].contentTopic).toEqual(message.contentTopic);
|
||||
expect(msgs[0].version).toEqual(message.version);
|
||||
|
||||
const payload = Buffer.from(msgs[0].payload);
|
||||
expect(Buffer.compare(payload, message.payload)).toBe(0);
|
||||
});
|
||||
|
||||
test('Nim-interop: nim node sends message to js node', async () => {
|
||||
const message = Message.fromUtf8String('Here is another message.');
|
||||
const waku = await Waku.create();
|
||||
|
||||
const peerId = waku.libp2p.peerId.toB58String();
|
||||
const localMultiaddr = waku.libp2p.multiaddrs.find((addr) =>
|
||||
addr.toString().match(/127\.0\.0\.1/)
|
||||
);
|
||||
const multiAddrWithId = localMultiaddr + '/p2p/' + peerId;
|
||||
|
||||
const nimWaku = new NimWaku(expect.getState().currentTestName);
|
||||
await nimWaku.start({ staticnode: multiAddrWithId });
|
||||
|
||||
await patchPeerStore(nimWaku, waku.libp2p);
|
||||
|
||||
await waku.relay.subscribe();
|
||||
|
||||
await new Promise((resolve) =>
|
||||
waku.libp2p.pubsub.once('gossipsub:heartbeat', resolve)
|
||||
);
|
||||
|
||||
const receivedPromise = waitForNextData(waku.libp2p.pubsub);
|
||||
|
||||
await nimWaku.sendMessage(message);
|
||||
|
||||
const receivedMsg = await receivedPromise;
|
||||
|
||||
expect(receivedMsg.contentTopic).toBe(message.contentTopic);
|
||||
expect(receivedMsg.version).toBe(message.version);
|
||||
|
||||
const payload = Buffer.from(receivedMsg.payload);
|
||||
expect(Buffer.compare(payload, message.payload)).toBe(0);
|
||||
});
|
||||
|
||||
function waitForNextData(pubsub: Pubsub): Promise<Message> {
|
||||
return new Promise((resolve) => {
|
||||
pubsub.once(TOPIC, resolve);
|
||||
}).then((msg: any) => {
|
||||
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)
|
||||
);
|
||||
}
|
||||
Loading…
Reference in New Issue