Compare commits

..

2 Commits

2 changed files with 34 additions and 31 deletions

View File

@ -18,6 +18,7 @@
"hyperswarm": "^4.3.5", "hyperswarm": "^4.3.5",
"prettier": "^2.7.1", "prettier": "^2.7.1",
"protoc": "^1.1.3", "protoc": "^1.1.3",
"sleep-promise": "^9.1.0",
"sodium-universal": "^3.1.0", "sodium-universal": "^3.1.0",
"tape": "^5.6.1" "tape": "^5.6.1"
} }

64
test.js
View File

@ -1,4 +1,5 @@
const test = require("tape"); const test = require("tape");
const sleep = require("sleep-promise");
const Hyperswarm = require("hyperswarm"); const Hyperswarm = require("hyperswarm");
const sodium = require("sodium-universal"); const sodium = require("sodium-universal");
const b4a = require("b4a"); const b4a = require("b4a");
@ -8,46 +9,45 @@ const crypto = require("crypto");
const topicName = crypto.randomBytes(10); const topicName = crypto.randomBytes(10);
test("Broadcast through several peers", (t) => { test("Broadcast through several peers", (t) => {
const peer1 = createPeer();
const peer2 = createPeer();
const peer3 = createPeer();
t.plan(2); t.plan(2);
Promise.all([peer1, peer2, peer3]).then((peers) => { Promise.all([createPeer(100), createPeer(300), createPeer(500)]).then(
const peer1 = peers.shift(); (peers) => {
const peer2 = peers.shift(); const peer1 = peers.shift();
const peer3 = peers.shift(); const peer2 = peers.shift();
const peer3 = peers.shift();
const flood1 = new DHTFlood({ swarm: peer1 });
const flood2 = new DHTFlood({ swarm: peer2 });
const flood3 = new DHTFlood({ swarm: peer3 });
const data = Buffer.from("Hello World");
const flood1 = new DHTFlood({ swarm: peer1 }); flood1.on("message", () => t.error("Got own message"));
const flood2 = new DHTFlood({ swarm: peer2 });
const flood3 = new DHTFlood({ swarm: peer3 });
const data = Buffer.from("Hello World");
flood1.on("message", () => t.error("Got own message")); flood2.on("message", (message) => {
t.deepEquals(message, data, "Data got broadcast");
});
flood3.on("message", (message) => {
t.deepEquals(message, data, "Data got broadcast");
});
flood2.on("message", (message) => { function maybeFlood() {
t.deepEquals(message, data, "Data got broadcast"); if (peer1.peers.size === 2) {
}); flood1.broadcast(data);
flood3.on("message", (message) => { peer1.removeListener("connection", maybeFlood);
t.deepEquals(message, data, "Data got broadcast"); }
});
function maybeFlood() {
if (peer1.peers.size === 2) {
flood1.broadcast(data);
peer1.removeListener("connection", maybeFlood);
} }
peer1.on("connection", maybeFlood);
t.teardown(() => {
[peer1, peer2, peer3].forEach((item) => item.destroy());
});
} }
);
peer1.on("connection", maybeFlood);
t.teardown(() => {
[peer1, peer2, peer3].forEach((item) => item.destroy());
});
});
}); });
async function createPeer() { async function createPeer(ms) {
await sleep(ms / 2);
const swarm = new Hyperswarm(); const swarm = new Hyperswarm();
await swarm.dht.ready(); await swarm.dht.ready();
await swarm.listen(); await swarm.listen();
@ -57,5 +57,7 @@ async function createPeer() {
swarm.join(topic); swarm.join(topic);
await sleep(ms / 2);
return swarm; return swarm;
} }