First commit
This commit is contained in:
Franck Royer
commit
26ffe57d55
|
|
@ -0,0 +1,470 @@
|
|||
import { expect } from "chai";
|
||||
import { describe } from "mocha";
|
||||
import { Multiaddr } from "multiaddr";
|
||||
import PeerId from "peer-id";
|
||||
|
||||
import { delay } from "../test_utils/delay";
|
||||
|
||||
import { randomBytes } from "./crypto";
|
||||
import { fleets } from "./discovery/predefined";
|
||||
import { Protocols, Waku } from "./waku";
|
||||
import { WakuMessage } from "./waku_message";
|
||||
|
||||
describe("Test nwaku fleets", () => {
|
||||
describe("Prod Tleet", () => {
|
||||
const wakus: Waku[] = [];
|
||||
|
||||
afterEach(function () {
|
||||
wakus.forEach((waku) => {
|
||||
waku.stop().catch((e) => console.log("Waku failed to stop", e));
|
||||
});
|
||||
});
|
||||
|
||||
it("Connect", async function () {
|
||||
// This test depends on fleets.status.im being online.
|
||||
// This dependence must be removed once DNS discovery is implemented
|
||||
this.timeout(20_000);
|
||||
|
||||
const nodes = Object.values(
|
||||
fleets.fleets["wakuv2.prod"]["waku-websocket"]
|
||||
);
|
||||
|
||||
const peerIds = nodes.map((a) => {
|
||||
const ma = new Multiaddr(a);
|
||||
return ma.getPeerId()!;
|
||||
});
|
||||
|
||||
expect(nodes.length).to.eq(3);
|
||||
|
||||
const promises = nodes.map(async (node, i) => {
|
||||
wakus[i] = await Waku.create({
|
||||
bootstrap: { peers: [node] },
|
||||
});
|
||||
|
||||
return new Promise((resolve) => {
|
||||
wakus[i].libp2p.connectionManager.on("peer:connect", (connection) => {
|
||||
resolve(connection.remotePeer);
|
||||
});
|
||||
}).then((connectedPeerID) => {
|
||||
const peerId = connectedPeerID as unknown as PeerId;
|
||||
console.log("connected", peerId.toB58String());
|
||||
expect(peerId.toB58String()).to.eq(peerIds[i]);
|
||||
});
|
||||
});
|
||||
|
||||
await Promise.all(promises);
|
||||
});
|
||||
|
||||
it("Relay", async function () {
|
||||
// This test depends on fleets.status.im being online.
|
||||
// This dependence must be removed once DNS discovery is implemented
|
||||
this.timeout(60000);
|
||||
|
||||
const id = randomBytes(4).toString();
|
||||
|
||||
const nodes = Object.values(
|
||||
fleets.fleets["wakuv2.prod"]["waku-websocket"]
|
||||
);
|
||||
|
||||
expect(nodes.length).to.eq(3);
|
||||
|
||||
const promises = nodes.map(async (node, i) => {
|
||||
wakus[i] = await Waku.create({
|
||||
bootstrap: { peers: [node] },
|
||||
});
|
||||
|
||||
await wakus[i].waitForRemotePeer([Protocols.Relay]);
|
||||
console.log(node + ": ready");
|
||||
});
|
||||
|
||||
await Promise.all(promises);
|
||||
// All connected and relay ready
|
||||
|
||||
const contentTopic = "/js-waku-testing/1/relay-test/utf8";
|
||||
|
||||
const messages: Array<{ msg: string; timestamp: Date; rcvd: Date }> = [];
|
||||
|
||||
wakus.forEach((waku) => {
|
||||
waku.relay.addObserver(
|
||||
(message) => {
|
||||
messages.push({
|
||||
msg: message.payloadAsUtf8,
|
||||
timestamp: message.timestamp!,
|
||||
rcvd: new Date(),
|
||||
});
|
||||
},
|
||||
[contentTopic]
|
||||
);
|
||||
});
|
||||
|
||||
const relayPromises = wakus.map(async (waku, i) => {
|
||||
const msg = await WakuMessage.fromUtf8String(
|
||||
`sent from ${i} - ${id}`,
|
||||
contentTopic
|
||||
);
|
||||
return waku.relay.send(msg);
|
||||
});
|
||||
|
||||
await Promise.all(relayPromises);
|
||||
await delay(30000);
|
||||
|
||||
console.log(messages);
|
||||
|
||||
messages.forEach((msg) => {
|
||||
const diff = msg.rcvd.getTime() - msg.timestamp.getTime();
|
||||
console.log(msg.timestamp, msg.rcvd, diff + "ms");
|
||||
});
|
||||
|
||||
expect(messages.length).to.gte(nodes.length);
|
||||
|
||||
for (let i = 0; i < wakus.length; i++) {
|
||||
expect(messages.map((m) => m.msg)).to.contain(`sent from ${i} - ${id}`);
|
||||
}
|
||||
});
|
||||
|
||||
it("Light Push", async function () {
|
||||
// This test depends on fleets.status.im being online.
|
||||
// This dependence must be removed once DNS discovery is implemented
|
||||
this.timeout(60000);
|
||||
|
||||
const id = randomBytes(4).toString();
|
||||
|
||||
const nodes = Object.values(
|
||||
fleets.fleets["wakuv2.prod"]["waku-websocket"]
|
||||
);
|
||||
|
||||
expect(nodes.length).to.eq(3);
|
||||
|
||||
const promises = nodes.map(async (node, i) => {
|
||||
wakus[i] = await Waku.create({
|
||||
bootstrap: { peers: [node] },
|
||||
});
|
||||
|
||||
await wakus[i].waitForRemotePeer([Protocols.LightPush]);
|
||||
console.log(node + ": ready");
|
||||
});
|
||||
|
||||
await Promise.all(promises);
|
||||
// All connected and relay ready
|
||||
|
||||
const contentTopic = "/js-waku-testing/1/relay-test/utf8";
|
||||
|
||||
const messages: Array<{ msg: string; timestamp: Date; rcvd: Date }> = [];
|
||||
|
||||
wakus.forEach((waku) => {
|
||||
waku.relay.addObserver(
|
||||
(message) => {
|
||||
messages.push({
|
||||
msg: message.payloadAsUtf8,
|
||||
timestamp: message.timestamp!,
|
||||
rcvd: new Date(),
|
||||
});
|
||||
},
|
||||
[contentTopic]
|
||||
);
|
||||
});
|
||||
|
||||
const relayPromises = wakus.map(async (waku, i) => {
|
||||
const msg = await WakuMessage.fromUtf8String(
|
||||
`sent from ${i} - ${id}`,
|
||||
contentTopic
|
||||
);
|
||||
return waku.lightPush.push(msg);
|
||||
});
|
||||
|
||||
await Promise.all(relayPromises);
|
||||
|
||||
console.log(messages);
|
||||
|
||||
messages.forEach((msg) => {
|
||||
const diff = msg.rcvd.getTime() - msg.timestamp.getTime();
|
||||
console.log(msg.timestamp, msg.rcvd, diff + "ms");
|
||||
});
|
||||
|
||||
expect(messages.length).to.gte(nodes.length);
|
||||
|
||||
for (let i = 0; i < wakus.length; i++) {
|
||||
expect(messages.map((m) => m.msg)).to.contain(`sent from ${i} - ${id}`);
|
||||
}
|
||||
});
|
||||
|
||||
it("Store", async function () {
|
||||
// This test depends on fleets.status.im being online.
|
||||
// This dependence must be removed once DNS discovery is implemented
|
||||
this.timeout(30000);
|
||||
|
||||
const nodes = Object.values(
|
||||
fleets.fleets["wakuv2.prod"]["waku-websocket"]
|
||||
);
|
||||
|
||||
expect(nodes.length).to.eq(3);
|
||||
|
||||
const id = randomBytes(4).toString();
|
||||
|
||||
const promises = nodes.map(async (node, i) => {
|
||||
wakus[i] = await Waku.create({
|
||||
bootstrap: { peers: [node] },
|
||||
});
|
||||
|
||||
await wakus[i].waitForRemotePeer([Protocols.Relay]);
|
||||
console.log(node + ": ready");
|
||||
});
|
||||
|
||||
await Promise.all(promises);
|
||||
// All connected and relay ready
|
||||
|
||||
const contentTopic = "/js-waku-testing/1/store-test/utf8";
|
||||
|
||||
const relayPromises = wakus.map(async (waku, i) => {
|
||||
const msg = await WakuMessage.fromUtf8String(
|
||||
`sent from ${i} - ${id}`,
|
||||
contentTopic
|
||||
);
|
||||
return waku.relay.send(msg);
|
||||
});
|
||||
|
||||
await Promise.all(relayPromises);
|
||||
await delay(5000);
|
||||
|
||||
const storePromises = wakus.map(async (waku, index) => {
|
||||
const messages = await waku.store.queryHistory([contentTopic]);
|
||||
const payloads = messages.map((msg) => msg.payloadAsUtf8);
|
||||
console.log(index, payloads);
|
||||
|
||||
for (let i = 0; i < wakus.length; i++) {
|
||||
expect(payloads).to.contain(`sent from ${i} - ${id}`);
|
||||
}
|
||||
});
|
||||
|
||||
await Promise.all(storePromises);
|
||||
});
|
||||
});
|
||||
|
||||
describe("Test Fleet", () => {
|
||||
const wakus: Waku[] = [];
|
||||
|
||||
afterEach(function () {
|
||||
wakus.forEach((waku) => {
|
||||
waku.stop().catch((e) => console.log("Waku failed to stop", e));
|
||||
});
|
||||
});
|
||||
|
||||
it("Connect", async function () {
|
||||
// This test depends on fleets.status.im being online.
|
||||
// This dependence must be removed once DNS discovery is implemented
|
||||
this.timeout(20_000);
|
||||
|
||||
const nodes = Object.values(
|
||||
fleets.fleets["wakuv2.test"]["waku-websocket"]
|
||||
);
|
||||
|
||||
const peerIds = nodes.map((a) => {
|
||||
const ma = new Multiaddr(a);
|
||||
return ma.getPeerId()!;
|
||||
});
|
||||
|
||||
expect(nodes.length).to.eq(3);
|
||||
|
||||
const promises = nodes.map(async (node, i) => {
|
||||
wakus[i] = await Waku.create({
|
||||
bootstrap: { peers: [node] },
|
||||
});
|
||||
|
||||
return new Promise((resolve) => {
|
||||
wakus[i].libp2p.connectionManager.on("peer:connect", (connection) => {
|
||||
resolve(connection.remotePeer);
|
||||
});
|
||||
}).then((connectedPeerID) => {
|
||||
const peerId = connectedPeerID as unknown as PeerId;
|
||||
expect(peerId.toB58String()).to.eq(peerIds[i]);
|
||||
});
|
||||
});
|
||||
|
||||
await Promise.all(promises);
|
||||
});
|
||||
|
||||
it("Relay", async function () {
|
||||
// This test depends on fleets.status.im being online.
|
||||
// This dependence must be removed once DNS discovery is implemented
|
||||
this.timeout(60000);
|
||||
|
||||
const id = randomBytes(4).toString();
|
||||
|
||||
const nodes = Object.values(
|
||||
fleets.fleets["wakuv2.test"]["waku-websocket"]
|
||||
);
|
||||
|
||||
expect(nodes.length).to.eq(3);
|
||||
|
||||
const promises = nodes.map(async (node, i) => {
|
||||
wakus[i] = await Waku.create({
|
||||
bootstrap: { peers: [node] },
|
||||
});
|
||||
|
||||
await wakus[i].waitForRemotePeer([Protocols.Relay]);
|
||||
console.log(node + ": ready");
|
||||
});
|
||||
|
||||
await Promise.all(promises);
|
||||
// All connected and relay ready
|
||||
|
||||
const contentTopic = "/js-waku-testing/1/relay-test/utf8";
|
||||
|
||||
const messages: Array<{ msg: string; timestamp: Date; rcvd: Date }> = [];
|
||||
|
||||
wakus.forEach((waku) => {
|
||||
waku.relay.addObserver(
|
||||
(message) => {
|
||||
messages.push({
|
||||
msg: message.payloadAsUtf8,
|
||||
timestamp: message.timestamp!,
|
||||
rcvd: new Date(),
|
||||
});
|
||||
},
|
||||
[contentTopic]
|
||||
);
|
||||
});
|
||||
|
||||
const relayPromises = wakus.map(async (waku, i) => {
|
||||
const msg = await WakuMessage.fromUtf8String(
|
||||
`sent from ${i} - ${id}`,
|
||||
contentTopic
|
||||
);
|
||||
return waku.relay.send(msg);
|
||||
});
|
||||
|
||||
await Promise.all(relayPromises);
|
||||
await delay(20000);
|
||||
|
||||
console.log(messages);
|
||||
|
||||
messages.forEach((msg) => {
|
||||
const diff = msg.rcvd.getTime() - msg.timestamp.getTime();
|
||||
console.log(msg.timestamp, msg.rcvd, diff + "ms");
|
||||
});
|
||||
|
||||
expect(messages.length).to.gte(nodes.length);
|
||||
|
||||
for (let i = 0; i < wakus.length; i++) {
|
||||
expect(messages.map((m) => m.msg)).to.contain(`sent from ${i} - ${id}`);
|
||||
}
|
||||
});
|
||||
|
||||
it("Light Push", async function () {
|
||||
// This test depends on fleets.status.im being online.
|
||||
// This dependence must be removed once DNS discovery is implemented
|
||||
this.timeout(60000);
|
||||
|
||||
const id = randomBytes(4).toString();
|
||||
|
||||
const nodes = Object.values(
|
||||
fleets.fleets["wakuv2.test"]["waku-websocket"]
|
||||
);
|
||||
|
||||
expect(nodes.length).to.eq(3);
|
||||
|
||||
const promises = nodes.map(async (node, i) => {
|
||||
wakus[i] = await Waku.create({
|
||||
bootstrap: { peers: [node] },
|
||||
});
|
||||
|
||||
await wakus[i].waitForRemotePeer([Protocols.LightPush]);
|
||||
console.log(node + ": ready");
|
||||
});
|
||||
|
||||
await Promise.all(promises);
|
||||
// All connected and relay ready
|
||||
|
||||
const contentTopic = "/js-waku-testing/1/relay-test/utf8";
|
||||
|
||||
const messages: Array<{ msg: string; timestamp: Date; rcvd: Date }> = [];
|
||||
|
||||
wakus.forEach((waku) => {
|
||||
waku.relay.addObserver(
|
||||
(message) => {
|
||||
messages.push({
|
||||
msg: message.payloadAsUtf8,
|
||||
timestamp: message.timestamp!,
|
||||
rcvd: new Date(),
|
||||
});
|
||||
},
|
||||
[contentTopic]
|
||||
);
|
||||
});
|
||||
|
||||
const relayPromises = wakus.map(async (waku, i) => {
|
||||
const msg = await WakuMessage.fromUtf8String(
|
||||
`sent from ${i} - ${id}`,
|
||||
contentTopic
|
||||
);
|
||||
return waku.lightPush.push(msg);
|
||||
});
|
||||
|
||||
await Promise.all(relayPromises);
|
||||
|
||||
console.log(messages);
|
||||
|
||||
messages.forEach((msg) => {
|
||||
const diff = msg.rcvd.getTime() - msg.timestamp.getTime();
|
||||
console.log(msg.timestamp, msg.rcvd, diff + "ms");
|
||||
});
|
||||
|
||||
expect(messages.length).to.gte(nodes.length);
|
||||
|
||||
for (let i = 0; i < wakus.length; i++) {
|
||||
expect(messages.map((m) => m.msg)).to.contain(`sent from ${i} - ${id}`);
|
||||
}
|
||||
});
|
||||
|
||||
it("Store", async function () {
|
||||
// This test depends on fleets.status.im being online.
|
||||
// This dependence must be removed once DNS discovery is implemented
|
||||
this.timeout(30000);
|
||||
|
||||
const nodes = Object.values(
|
||||
fleets.fleets["wakuv2.test"]["waku-websocket"]
|
||||
);
|
||||
|
||||
expect(nodes.length).to.eq(3);
|
||||
|
||||
const id = randomBytes(4).toString();
|
||||
|
||||
const promises = nodes.map(async (node, i) => {
|
||||
wakus[i] = await Waku.create({
|
||||
bootstrap: { peers: [node] },
|
||||
});
|
||||
|
||||
await wakus[i].waitForRemotePeer([Protocols.Relay]);
|
||||
console.log(node + ": ready");
|
||||
});
|
||||
|
||||
await Promise.all(promises);
|
||||
// All connected and relay ready
|
||||
|
||||
const contentTopic = "/js-waku-testing/1/store-test/utf8";
|
||||
|
||||
const relayPromises = wakus.map(async (waku, i) => {
|
||||
const msg = await WakuMessage.fromUtf8String(
|
||||
`sent from ${i} - ${id}`,
|
||||
contentTopic
|
||||
);
|
||||
return waku.relay.send(msg);
|
||||
});
|
||||
|
||||
await Promise.all(relayPromises);
|
||||
await delay(5000);
|
||||
|
||||
const storePromises = wakus.map(async (waku, index) => {
|
||||
const messages = await waku.store.queryHistory([contentTopic]);
|
||||
const payloads = messages.map((msg) => msg.payloadAsUtf8);
|
||||
console.log(index, payloads);
|
||||
|
||||
for (let i = 0; i < wakus.length; i++) {
|
||||
expect(payloads).to.contain(`sent from ${i} - ${id}`);
|
||||
}
|
||||
});
|
||||
|
||||
await Promise.all(storePromises);
|
||||
});
|
||||
});
|
||||
});
|
||||
Loading…
Reference in New Issue