From cebfed03823e8fdbd4742aaad0408bf2d5a85b82 Mon Sep 17 00:00:00 2001
From: Leifer Mendez <leifer33@gmail.com>
Date: Mon, 23 Jan 2023 20:27:55 +0100
Subject: [PATCH] test(bot): :zap: more test and endflow

---
 __test__/04-case.test.js        | 82 +++++++++++++++++++++++++++++++++
 packages/bot/core/core.class.js | 22 ++++++++-
 2 files changed, 102 insertions(+), 2 deletions(-)
 create mode 100644 __test__/04-case.test.js

diff --git a/__test__/04-case.test.js b/__test__/04-case.test.js
new file mode 100644
index 0000000..2291a0c
--- /dev/null
+++ b/__test__/04-case.test.js
@@ -0,0 +1,82 @@
+const { test } = require('uvu')
+const assert = require('uvu/assert')
+const MOCK_DB = require('../packages/database/src/mock')
+const PROVIDER_DB = require('../packages/provider/src/mock')
+const {
+    addKeyword,
+    createBot,
+    createFlow,
+    createProvider,
+} = require('../packages/bot/index')
+
+/**
+ * Falsear peticion async
+ * @param {*} fakeData
+ * @returns
+ */
+const fakeHTTP = async (fakeData = []) => {
+    console.log('⚡ Server request!')
+    await delay(50)
+    console.log('⚡ Server return!')
+    const data = fakeData.map((u, i) => ({ body: `${i + 1} ${u}` }))
+    console.log(data)
+    return Promise.resolve(data)
+}
+
+test(`[Caso - 04] Romper flujo (endFlow)`, async () => {
+    const MOCK_VALUES = [
+        'Bienvenido te envio muchas marcas (5510)',
+        'Seleccione marca del auto a cotizar, con el *número* correspondiente',
+        'Seleccione la sub marca del auto a cotizar, con el *número* correspondiente:',
+        'Los precios rondan:',
+    ]
+    const provider = createProvider(PROVIDER_DB)
+    const database = new MOCK_DB()
+
+    const flujoPrincipal = addKeyword(['hola'])
+        .addAnswer(MOCK_VALUES[0], null, async (ctx, { flowDynamic }) => {
+            console.log('execute...')
+            const data = await fakeHTTP(['Ford', 'GM', 'BMW'])
+            return flowDynamic(data)
+        })
+        .addAnswer(MOCK_VALUES[1], null, async (ctx, { endFlow }) => {
+            return endFlow()
+        })
+        .addAnswer(MOCK_VALUES[2], null, async (ctx, { flowDynamic }) => {
+            const data = await fakeHTTP(['Usado', 'Nuevos'])
+            return flowDynamic(data)
+        })
+        .addAnswer(MOCK_VALUES[3], null, async (ctx, { flowDynamic }) => {
+            const data = await fakeHTTP(['1000', '2000', '3000'])
+            return flowDynamic(data)
+        })
+
+    createBot({
+        database,
+        flow: createFlow([flujoPrincipal]),
+        provider,
+    })
+
+    provider.delaySendMessage(0, 'message', {
+        from: '000',
+        body: 'hola',
+    })
+
+    await delay(1200)
+    const getHistory = database.listHistory.map((i) => i.answer)
+    assert.is(MOCK_VALUES[0], getHistory[0])
+
+    //FlowDynamic
+    assert.is('1 Ford', getHistory[1])
+    assert.is('2 GM', getHistory[2])
+    assert.is('3 BMW', getHistory[3])
+
+    assert.is(MOCK_VALUES[1], getHistory[4])
+    assert.is(undefined, getHistory[5])
+})
+
+test.run()
+
+function delay(ms) {
+    return new Promise((res) => setTimeout(res, ms))
+}
diff --git a/packages/bot/core/core.class.js b/packages/bot/core/core.class.js
index 30e6f4d..f7051b3 100644
--- a/packages/bot/core/core.class.js
+++ b/packages/bot/core/core.class.js
@@ -72,11 +72,12 @@ class CoreClass {
         const { body, from } = messageCtxInComming
         let msgToSend = []
         let fallBackFlag = false
+        let endFlowFlag = false
         if (this.generalArgs.blackList.includes(from)) return
         if (!body) return
         if (!body.length) return
 
-        const prevMsg = await this.databaseClass.getPrevByNumber(from)
+        let prevMsg = await this.databaseClass.getPrevByNumber(from)
         const refToContinue = this.flowClass.findBySerialize(
             prevMsg?.refSerialize
         )
@@ -90,13 +91,28 @@ class CoreClass {
             this.databaseClass.save(ctxByNumber)
         }
 
+        // 📄 Limpiar cola de procesos
+        const clearQueue = () => {
+            QueuePrincipal.pendingPromise = false
+            QueuePrincipal.queue = []
+        }
+
+        // 📄 Finalizar flujo
+        const endFlow = async () => {
+            prevMsg = null
+            endFlowFlag = true
+            clearQueue()
+            return
+        }
+
         // 📄 Esta funcion se encarga de enviar un array de mensajes dentro de este ctx
         const sendFlow = async (messageToSend, numberOrId) => {
             // [1 Paso] esto esta bien!
-            if (prevMsg?.options?.capture) await cbEveryCtx(prevMsg?.ref)
 
+            if (prevMsg?.options?.capture) await cbEveryCtx(prevMsg?.ref)
             const queue = []
             for (const ctxMessage of messageToSend) {
+                if (endFlowFlag) return
                 const delayMs = ctxMessage?.options?.delay || 0
                 if (delayMs) await delay(delayMs)
                 QueuePrincipal.enqueue(() =>
@@ -145,6 +161,7 @@ class CoreClass {
                 })
                 .slice(0, optListMsg.limit)
 
+            if (endFlowFlag) return
             for (const msg of parseListMsg) {
                 await this.sendProviderAndSave(from, msg)
             }
@@ -163,6 +180,7 @@ class CoreClass {
             return this.flowClass.allCallbacks[inRef](messageCtxInComming, {
                 fallBack,
                 flowDynamic,
+                endFlow,
             })
         }