From 86ecfc633316ed04b2c9ec86f3709aa7af10469a Mon Sep 17 00:00:00 2001 From: Lucas Zanek <57494138+LucasZNK@users.noreply.github.com> Date: Wed, 29 Mar 2023 17:59:17 -0300 Subject: [PATCH] whisper.addon : fixed test to new async implementation (#686) * fixed blocking code on node addon * modify the example to run async * format * added logic to see the whisper output * added logic to see the whisper output * removed extra function for more clean example * fixed whisper test to new async implementation --- examples/addon.node/__test__/whisper.spec.js | 23 +++++++++++++------- 1 file changed, 15 insertions(+), 8 deletions(-) diff --git a/examples/addon.node/__test__/whisper.spec.js b/examples/addon.node/__test__/whisper.spec.js index b7d35406..a8088b48 100644 --- a/examples/addon.node/__test__/whisper.spec.js +++ b/examples/addon.node/__test__/whisper.spec.js @@ -1,15 +1,22 @@ -const path = require('path'); -const { whisper } = require(path.join(__dirname, '../../../build/Release/whisper-addon')); +const path = require("path"); +const { whisper } = require(path.join( + __dirname, + "../../../build/Release/whisper-addon" +)); +const { promisify } = require("util"); + +const whisperAsync = promisify(whisper); const whisperParamsMock = { - language: 'en', - model: path.join(__dirname, '../../../models/ggml-base.en.bin'), - fname_inp: path.join(__dirname, '../../../samples/jfk.wav'), + language: "en", + model: path.join(__dirname, "../../../models/ggml-base.en.bin"), + fname_inp: path.join(__dirname, "../../../samples/jfk.wav"), }; describe("Run whisper.node", () => { + test("it should receive a non-empty value", async () => { + let result = await whisperAsync(whisperParamsMock); - test("it should receive a non-empty value", () => { - expect(whisper(whisperParamsMock).length).toBeGreaterThan(0); - }); + expect(result.length).toBeGreaterThan(0); + }); });