botium-core/src/mocks/BotiumMockMessage.js

The Selenium for Chatbots

const BotiumMockAttachment = require('./BotiumMockAttachment')
const { BotiumMockMedia, BotiumMockButton, BotiumMockCard, BotiumMockForm } = require('./BotiumMockRichMessageTypes')
const { BotiumMockAsserter, BotiumMockUserInput, BotiumMockLogicHook } = require('./BotiumMockScripting')

module.exports = class BotiumMockMessage {
  constructor (fromJson = {}) {
    this.sender = fromJson.sender
    this.channel = fromJson.channel
    this.not = fromJson.not
    this.optional = fromJson.optional
    this.messageText = fromJson.messageText
    this.media = (fromJson.media ? fromJson.media.map((a) => new BotiumMockMedia(a)) : null)
    this.buttons = (fromJson.buttons ? fromJson.buttons.map((a) => new BotiumMockButton(a)) : null)
    this.cards = (fromJson.cards ? fromJson.cards.map((a) => new BotiumMockCard(a)) : null)
    this.forms = (fromJson.forms ? fromJson.forms.map((a) => new BotiumMockForm(a)) : null)
    this.nlp = fromJson.nlp
    this.sourceData = fromJson.sourceData
    this.sourceAction = fromJson.sourceAction
    this.attachments = (fromJson.attachments ? fromJson.attachments.map((a) => new BotiumMockAttachment(a)) : null)
    this.asserters = (fromJson.asserters ? fromJson.asserters.map((a) => new BotiumMockAsserter(a)) : null)
    this.userInputs = (fromJson.userInputs ? fromJson.userInputs.map((a) => new BotiumMockUserInput(a)) : null)
    this.logicHooks = (fromJson.logicHooks ? fromJson.logicHooks.map((a) => new BotiumMockLogicHook(a)) : null)
  }

  prettify () {
    const prettifySafe = (entry) => entry.prettify ? entry.prettify(2) : `${' '.repeat(2)}<No botium object!>${JSON.stringify(entry)}`
    const lines = []
    if (this.messageText) lines.push(this.messageText)
    if (this.media && this.media.length > 0) lines.push(...this.media.map(m => prettifySafe(m)))
    if (this.buttons && this.buttons.length > 0) lines.push(...this.buttons.map(b => prettifySafe(b)))
    if (this.cards && this.cards.length > 0) lines.push(...this.cards.map(c => prettifySafe(c)))
    if (this.forms && this.forms.length > 0) lines.push(...this.forms.map(f => prettifySafe(f)))

    if (lines.length === 0) return `#${this.sender}:`

    return [
      `#${this.sender}: ${lines[0]}`,
      ...lines.slice(1)
    ].join('\n')
  }
}