botium-core/src/scripting/logichook/asserter/CardsAsserter.js

The Selenium for Chatbots

const { SCRIPTING_NORMALIZE_TEXT } = require('../../../Capabilities')
const { BotiumError } = require('../../BotiumError')
const { cardsFromMsg } = require('../helpers')
const { normalizeText } = require('../../helper')

module.exports = class CardsAsserter {
  constructor (context, caps = {}) {
    this.context = context
    this.caps = caps
    this.name = 'Cards Asserter'
  }

  _evalCards (args, botMsg) {
    const allCards = cardsFromMsg(botMsg, true).reduce((acc, mc) => acc.concat([mc.text, mc.subtext, mc.content].filter(t => t).map(t => normalizeText(t, !!this.caps[SCRIPTING_NORMALIZE_TEXT]))), [])
    if (!args || args.length === 0) {
      return { allCards, cardsNotFound: [], cardsFound: allCards }
    }
    const cardsNotFound = []
    const cardsFound = []
    for (let i = 0; i < (args || []).length; i++) {
      if (allCards.findIndex(c => this.context.Match(c, normalizeText(args[i], !!this.caps[SCRIPTING_NORMALIZE_TEXT]))) < 0) {
        cardsNotFound.push(args[i])
      } else {
        cardsFound.push(args[i])
      }
    }
    return { allCards, cardsNotFound, cardsFound }
  }

  assertNotConvoStep ({ convo, convoStep, args = [], botMsg = [] }) {
    const { allCards, cardsFound } = this._evalCards(args, botMsg)
    if (cardsFound.length > 0) {
      return Promise.reject(new BotiumError(
          `${convoStep.stepTag}: Not expected card(s) with text "${cardsFound}"`,
          {
            type: 'asserter',
            source: this.name,
            params: {
              args
            },
            cause: {
              not: true,
              expected: args,
              actual: allCards,
              diff: cardsFound
            }
          }
      ))
    }
    return Promise.resolve()
  }

  assertConvoStep ({ convo, convoStep, args = [], botMsg = [] }) {
    const { allCards, cardsNotFound, cardsFound } = this._evalCards(args, botMsg)
    if (!args || args.length === 0) {
      if (!cardsFound.length) {
        return Promise.reject(new BotiumError(
          `${convoStep.stepTag}: Expected some card(s)`,
          {
            type: 'asserter',
            source: this.name,
            params: {
              args
            },
            cause: {
              not: false,
              expected: args,
              actual: allCards,
              diff: cardsNotFound
            }
          }
        ))
      }
    } else if (cardsNotFound.length > 0) {
      return Promise.reject(new BotiumError(
          `${convoStep.stepTag}: Expected card(s) with text "${cardsNotFound}"`,
          {
            type: 'asserter',
            source: this.name,
            params: {
              args
            },
            cause: {
              not: false,
              expected: args,
              actual: allCards,
              diff: cardsNotFound
            }
          }
      ))
    }
    return Promise.resolve()
  }
}