wa-chat-server
Version:
Watson Assistant powered chat server
158 lines (157 loc) • 7.91 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
const httpMocks = __importStar(require("node-mocks-http"));
const Session_1 = require("../../class/Session/Session");
const SessionMock_1 = require("../mock/SessionMock");
const MessageRequestFilter_1 = require("../../class/MessageRequestFilter");
const WAChatServerConfigMock_1 = require("../mock/WAChatServerConfigMock");
const Logger_1 = require("../../class/Logger/Logger");
const LoggerDriverMock_1 = require("../mock/LoggerDriverMock");
class MessageRequestFilterTest {
getLogger() {
return new Logger_1.Logger(this.getConfig(), new LoggerDriverMock_1.LoggerDriverMock(false, true));
}
getConfig() {
return new WAChatServerConfigMock_1.WAChatServerConfigMock();
}
noContextTest() {
return __awaiter(this, void 0, void 0, function* () {
it('no_context', () => __awaiter(this, void 0, void 0, function* () {
expect.assertions(1);
const session = yield new SessionMock_1.SessionMock().start();
session.setRunningAssistant('ASSISTANT');
session.data = { assistants: { ASSISTANT: {} } };
const request = yield httpMocks.createRequest({
method: 'POST',
url: '/dummy',
body: {},
});
const filter = new MessageRequestFilter_1.MessageRequestFilter(this.getLogger(), this.getConfig(), session);
expect(yield filter.applyTo(request)).toStrictEqual({ context: {} });
}));
});
}
addContextTest() {
return __awaiter(this, void 0, void 0, function* () {
it('add_context', () => __awaiter(this, void 0, void 0, function* () {
expect.assertions(1);
const session = yield new Session_1.Session('ID').start();
const context = { KEY: 'VALUE' };
session.setRunningAssistant('ASSISTANT');
session.data = { assistants: { ASSISTANT: { context } } };
const request = yield httpMocks.createRequest({
method: 'POST',
url: '/dummy',
params: {},
});
const filter = new MessageRequestFilter_1.MessageRequestFilter(this.getLogger(), this.getConfig(), session);
expect(yield filter.applyTo(request)).toStrictEqual({ context });
}));
});
}
ignoreClientDataWhenDisabledTest() {
return __awaiter(this, void 0, void 0, function* () {
it('ignore_client_data_when_disabled', () => __awaiter(this, void 0, void 0, function* () {
expect.assertions(1);
const session = yield new SessionMock_1.SessionMock().start();
const request = yield httpMocks.createRequest({
method: 'POST',
url: '/dummy',
body: { client: { k: 'v' } },
});
const filter = new MessageRequestFilter_1.MessageRequestFilter(this.getLogger(), this.getConfig(), session);
expect(yield filter.applyTo(request)).toStrictEqual({ context: {} });
}));
});
}
ignoreClientDataWhenExceedsAllowedSizeTest() {
return __awaiter(this, void 0, void 0, function* () {
it('ignore_client_data_when_exceeds_allowed_size', () => __awaiter(this, void 0, void 0, function* () {
expect.assertions(1);
const session = yield new SessionMock_1.SessionMock().start();
const config = this.getConfig();
config.client_data_max_size_in_bytes = 1; // eslint-disable-line @typescript-eslint/camelcase
const loggerDriver = new LoggerDriverMock_1.LoggerDriverMock(true, true).expect('warn', 'MessageRequestFilter, sessionId=ID:' +
' trying to pass 9B to Watson context, only 1B allowed', []);
const logger = new Logger_1.Logger(this.getConfig(), loggerDriver);
const request = yield httpMocks.createRequest({
method: 'POST',
url: '/dummy',
body: { client: { k: 'v' } },
});
const filter = new MessageRequestFilter_1.MessageRequestFilter(logger, config, session);
const result = yield filter.applyTo(request);
expect(result).toStrictEqual({ context: {} });
}));
});
}
addClientDataToContextTest() {
return __awaiter(this, void 0, void 0, function* () {
it('add_client_data_to_context', () => __awaiter(this, void 0, void 0, function* () {
expect.assertions(1);
const session = yield new SessionMock_1.SessionMock().start();
const config = this.getConfig();
config.client_data_max_size_in_bytes = 20; // eslint-disable-line @typescript-eslint/camelcase
const request = yield httpMocks.createRequest({
method: 'POST',
url: '/dummy',
body: { client: { k: 'v' } },
});
const filter = new MessageRequestFilter_1.MessageRequestFilter(this.getLogger(), config, session);
expect(yield filter.applyTo(request)).toStrictEqual({ context: { client: { k: 'v' } } });
}));
});
}
test() {
return __awaiter(this, void 0, void 0, function* () {
this.noContextTest();
this.addContextTest();
this.ignoreClientDataWhenDisabledTest();
this.ignoreClientDataWhenExceedsAllowedSizeTest();
this.addClientDataToContextTest();
});
}
}
new MessageRequestFilterTest().test();