diff --git a/packages/tests/tests/utils.spec.ts b/packages/tests/tests/utils.spec.ts index fc31cbc8a1..89c77ebd8a 100644 --- a/packages/tests/tests/utils.spec.ts +++ b/packages/tests/tests/utils.spec.ts @@ -17,7 +17,12 @@ import chai, { expect } from "chai"; import chaiAsPromised from "chai-as-promised"; import sinon from "sinon"; -import { delay, makeLogFileName, NOISE_KEY_1 } from "../src/index.js"; +import { + delay, + makeLogFileName, + NOISE_KEY_1, + tearDownNodes +} from "../src/index.js"; import { NimGoNode } from "../src/node/node.js"; chai.use(chaiAsPromised); @@ -43,13 +48,9 @@ describe("Util: toAsyncIterator: Filter", () => { await waitForRemotePeer(waku, [Protocols.Filter, Protocols.LightPush]); }); - afterEach(async () => { - try { - await nwaku.stop(); - await waku.stop(); - } catch (err) { - console.log("Failed to stop", err); - } + afterEach(async function () { + this.timeout(10000); + await tearDownNodes(nwaku, waku); }); it("creates an iterator", async function () { @@ -99,6 +100,8 @@ describe("Util: toAsyncIterator: Filter", () => { payload: utf8ToBytes("This should be received") }); + await delay(400); + await stop(); await waku.lightPush.send(TestEncoder, { diff --git a/packages/tests/tests/waku.node.spec.ts b/packages/tests/tests/waku.node.spec.ts index 608b432bc9..3cc14ae49a 100644 --- a/packages/tests/tests/waku.node.spec.ts +++ b/packages/tests/tests/waku.node.spec.ts @@ -22,10 +22,11 @@ import { expect } from "chai"; import { makeLogFileName, - NimGoNode, NOISE_KEY_1, - NOISE_KEY_2 + NOISE_KEY_2, + tearDownNodes } from "../src/index.js"; +import { NimGoNode } from "../src/node/node.js"; const TestContentTopic = "/test/1/waku/utf8"; @@ -33,13 +34,12 @@ const TestEncoder = createPlainEncoder({ contentTopic: TestContentTopic }); describe("Waku Dial [node only]", function () { describe("Interop: NimGoNode", function () { - let waku: Waku; + let waku: LightNode; let nwaku: NimGoNode; afterEach(async function () { - !!nwaku && - nwaku.stop().catch((e) => console.log("Nwaku failed to stop", e)); - !!waku && waku.stop().catch((e) => console.log("Waku failed to stop", e)); + this.timeout(15000); + await tearDownNodes(nwaku, waku); }); it("connects to nwaku", async function () { @@ -102,8 +102,8 @@ describe("Waku Dial [node only]", function () { let nwaku: NimGoNode; afterEach(async function () { - !!nwaku && (await nwaku.stop()); - !!waku && waku.stop().catch((e) => console.log("Waku failed to stop", e)); + this.timeout(15000); + await tearDownNodes(nwaku, waku); }); it("Passing an array", async function () {