mirror of
https://github.com/KeygraphHQ/juice-shop.git
synced 2026-11-10 02:31:39 -10:00
74 lines
2.0 KiB
TypeScript
74 lines
2.0 KiB
TypeScript
/*
|
|
* Copyright (c) 2014-2025 Bjoern Kimminich & the OWASP Juice Shop contributors.
|
|
* SPDX-License-Identifier: MIT
|
|
*/
|
|
|
|
import chai from 'chai'
|
|
import sinonChai from 'sinon-chai'
|
|
|
|
import validateChatBot, { checkIntentWithFunctionHandlerExists } from '../../lib/startup/validateChatBot'
|
|
import botDefaultTrainingData from '../../data/static/botDefaultTrainingData.json'
|
|
|
|
const expect = chai.expect
|
|
chai.use(sinonChai)
|
|
|
|
describe('chatBotValidation', () => {
|
|
describe('checkIntentWithHandlerExists', () => {
|
|
it('should accept training data with the expected intent and handler', () => {
|
|
const trainingData = {
|
|
data: [
|
|
{
|
|
intent: 'queries.test',
|
|
answers: [
|
|
{
|
|
action: 'function',
|
|
handler: 'testHandler'
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
|
|
expect(checkIntentWithFunctionHandlerExists(trainingData, 'queries.test', 'testHandler')).to.equal(true)
|
|
})
|
|
|
|
it('should fail if the training data lacks the expected intent', () => {
|
|
const trainingData = {
|
|
data: [
|
|
{
|
|
intent: 'queries.dummy'
|
|
}
|
|
]
|
|
}
|
|
|
|
expect(checkIntentWithFunctionHandlerExists(trainingData, 'queries.test', 'testFunction')).to.equal(false)
|
|
})
|
|
|
|
it('should fail if the training data lacks the expected handler for the given intent', () => {
|
|
const trainingData = {
|
|
data: [
|
|
{
|
|
intent: 'queries.test',
|
|
answers: [
|
|
{
|
|
action: 'function',
|
|
handler: 'dummyHandler'
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
|
|
expect(checkIntentWithFunctionHandlerExists(trainingData, 'queries.test', 'testHandler')).to.equal(false)
|
|
})
|
|
})
|
|
|
|
it('should accept the default chatbot training data', () => {
|
|
expect(validateChatBot(botDefaultTrainingData)).to.equal(true)
|
|
})
|
|
|
|
it('should fail if the chatbot training data is empty', () => {
|
|
expect(validateChatBot({ data: [] }, false)).to.equal(false)
|
|
})
|
|
})
|