logos-messaging/js-waku
1
0
Fork 0
mirror of https://github.com/logos-messaging/js-waku.git synced 2026-01-04 06:43:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: return first peer without traversing branch

Browse Source
This commit is contained in:
fryorcraken 2025-07-28 11:19:25 +10:00
parent 25f884e05b
commit f4a2778e02
No known key found for this signature in database
GPG Key ID: A82ED75A8DFC50A4
1 changed files with 32 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
packages/discovery/src/dns/dns.spec.ts
View File

@ -33,10 +33,12 @@ const errorBranchB = `enrtree-branch:${branchDomainD}`;
class MockDNS implements DnsClient { class MockDNS implements DnsClient {
private fqdnRes: Map<string, string[]>; private fqdnRes: Map<string, string[]>;
private fqdnThrows: string[]; private fqdnThrows: string[];
public hasThrown: boolean;
public constructor() { public constructor() {
this.fqdnRes = new Map(); this.fqdnRes = new Map();
this.fqdnThrows = []; this.fqdnThrows = [];
this.hasThrown = false;
} }
public addRes(fqdn: string, res: string[]): void { public addRes(fqdn: string, res: string[]): void {
@ -48,11 +50,18 @@ class MockDNS implements DnsClient {
} }
public resolveTXT(fqdn: string): Promise<string[]> { public resolveTXT(fqdn: string): Promise<string[]> {
if (this.fqdnThrows.includes(fqdn)) throw "Mock DNS throws."; if (this.fqdnThrows.includes(fqdn)) {
this.hasThrown = true;
console.log("throwing");
throw "Mock DNS throws.";
}
const res = this.fqdnRes.get(fqdn); const res = this.fqdnRes.get(fqdn);
if (!res) throw `Mock DNS could not resolve ${fqdn}`; if (!res) {
this.hasThrown = true;
throw `Mock DNS could not resolve ${fqdn}`;
}
return Promise.resolve(res); return Promise.resolve(res);
} }
@ -163,6 +172,7 @@ describe("DNS Node Discovery", () => {
} }
expect(peersB.length).to.eq(1); expect(peersB.length).to.eq(1);
expect(peersA[0].ip).to.eq(peersB[0].ip); expect(peersA[0].ip).to.eq(peersB[0].ip);
expect(mockDns.hasThrown).to.be.false;
}); });
}); });
@ -204,6 +214,26 @@ describe("DNS Node Discovery w/ capabilities", () => {
); );
}); });
it("return first retrieved peers without further DNS queries", async function () {
mockDns.addRes(`${rootDomain}.${host}`, multiComponentBranch);
mockDns.addRes(`${branchDomainA}.${host}`, [
mockData.enrWithWaku2RelayStore
]);
// The ENR Tree is such as there are more branches to be explored.
// But they should not be explored if it isn't asked
mockDns.addThrow(`${branchDomainB}.${host}`);
const dnsNodeDiscovery = new DnsNodeDiscovery(mockDns);
const iterator = dnsNodeDiscovery.getNextPeer([mockData.enrTree]);
const { value: peer } = await iterator.next();
expect(peer.peerId?.toString()).to.eq(
"16Uiu2HAm2HyS6brcCspSbszG9i36re2bWBVjMe3tMdnFp1Hua34F"
);
expect(mockDns.hasThrown).to.be.false;
});
it("retrieves all peers (3) when branch entries are composed of multiple strings", async function () { it("retrieves all peers (3) when branch entries are composed of multiple strings", async function () {
mockDns.addRes(`${rootDomain}.${host}`, multiComponentBranch); mockDns.addRes(`${rootDomain}.${host}`, multiComponentBranch);
mockDns.addRes(`${branchDomainA}.${host}`, [ mockDns.addRes(`${branchDomainA}.${host}`, [