add logging to rest session actions

This commit is contained in:
Bruno Windels 2018-09-14 12:17:22 +02:00
parent af255c6386
commit 6deb595fec
5 changed files with 58 additions and 16 deletions

View File

@ -27,10 +27,10 @@ module.exports = class RestSessionCreator {
this.cwd = cwd; this.cwd = cwd;
} }
async createSessionRange(usernames, password) { async createSessionRange(usernames, password, groupName) {
const sessionPromises = usernames.map((username) => this.createSession(username, password)); const sessionPromises = usernames.map((username) => this.createSession(username, password));
const sessions = await Promise.all(sessionPromises); const sessions = await Promise.all(sessionPromises);
return new RestMultiSession(sessions); return new RestMultiSession(sessions, groupName);
} }
async createSession(username, password) { async createSession(username, password) {

View File

@ -17,14 +17,16 @@ limitations under the License.
const request = require('request-promise-native'); const request = require('request-promise-native');
const RestRoom = require('./room'); const RestRoom = require('./room');
const {approveConsent} = require('./consent'); const {approveConsent} = require('./consent');
const Logger = require('../logger');
module.exports = class RestMultiSession { module.exports = class RestMultiSession {
constructor(sessions) { constructor(sessions, groupName) {
this.log = new Logger(groupName);
this.sessions = sessions; this.sessions = sessions;
} }
slice(start, end) { slice(start, end, groupName) {
return new RestMultiSession(this.sessions.slice(start, end)); return new RestMultiSession(this.sessions.slice(start, end), groupName);
} }
pop(userName) { pop(userName) {
@ -37,25 +39,52 @@ module.exports = class RestMultiSession {
} }
async setDisplayName(fn) { async setDisplayName(fn) {
await Promise.all(this.sessions.map((s) => s.setDisplayName(fn(s)))); this.log.step("set their display name")
await Promise.all(this.sessions.map(async (s) => {
s.log.mute();
await s.setDisplayName(fn(s));
s.log.unmute();
}));
this.log.done();
} }
async join(roomId) { async join(roomIdOrAlias) {
const rooms = await Promise.all(this.sessions.map((s) => s.join(roomId))); this.log.step(`join ${roomIdOrAlias}`)
return new RestMultiRoom(rooms); const rooms = await Promise.all(this.sessions.map(async (s) => {
s.log.mute();
const room = await s.join(roomIdOrAlias);
s.log.unmute();
return room;
}));
this.log.done();
return new RestMultiRoom(rooms, roomIdOrAlias, this.log);
} }
} }
class RestMultiRoom { class RestMultiRoom {
constructor(rooms) { constructor(rooms, roomIdOrAlias, log) {
this.rooms = rooms; this.rooms = rooms;
this.roomIdOrAlias = roomIdOrAlias;
this.log = log;
} }
async talk(message) { async talk(message) {
await Promise.all(this.rooms.map((r) => r.talk(message))); this.log.step(`say "${message}" in ${this.roomIdOrAlias}`)
await Promise.all(this.rooms.map(async (r) => {
r.log.mute();
await r.talk(message);
r.log.unmute();
}));
this.log.done();
} }
async leave() { async leave() {
await Promise.all(this.rooms.map((r) => r.leave())); this.log.step(`leave ${this.roomIdOrAlias}`)
await Promise.all(this.rooms.map(async (r) => {
r.log.mute();
await r.leave(message);
r.log.unmute();
}));
this.log.done();
} }
} }

View File

@ -18,22 +18,27 @@ const uuidv4 = require('uuid/v4');
/* no pun intented */ /* no pun intented */
module.exports = class RestRoom { module.exports = class RestRoom {
constructor(session, roomId) { constructor(session, roomId, log) {
this.session = session; this.session = session;
this._roomId = roomId; this._roomId = roomId;
this.log = log;
} }
async talk(message) { async talk(message) {
this.log.step(`says "${message}" in ${this._roomId}`)
const txId = uuidv4(); const txId = uuidv4();
await this.session._put(`/rooms/${this._roomId}/send/m.room.message/${txId}`, { await this.session._put(`/rooms/${this._roomId}/send/m.room.message/${txId}`, {
"msgtype": "m.text", "msgtype": "m.text",
"body": message "body": message
}); });
this.log.done();
return txId; return txId;
} }
async leave() { async leave() {
this.log.step(`leaves ${this._roomId}`)
await this.session._post(`/rooms/${this._roomId}/leave`); await this.session._post(`/rooms/${this._roomId}/leave`);
this.log.done();
} }
roomId() { roomId() {

View File

@ -15,11 +15,13 @@ limitations under the License.
*/ */
const request = require('request-promise-native'); const request = require('request-promise-native');
const Logger = require('../logger');
const RestRoom = require('./room'); const RestRoom = require('./room');
const {approveConsent} = require('./consent'); const {approveConsent} = require('./consent');
module.exports = class RestSession { module.exports = class RestSession {
constructor(credentials) { constructor(credentials) {
this.log = new Logger(credentials.userId);
this._credentials = credentials; this._credentials = credentials;
this._displayName = null; this._displayName = null;
} }
@ -37,18 +39,23 @@ module.exports = class RestSession {
} }
async setDisplayName(displayName) { async setDisplayName(displayName) {
this.log.step(`sets their display name to ${displayName}`);
this._displayName = displayName; this._displayName = displayName;
await this._put(`/profile/${this._credentials.userId}/displayname`, { await this._put(`/profile/${this._credentials.userId}/displayname`, {
displayname: displayName displayname: displayName
}); });
this.log.done();
} }
async join(roomIdOrAlias) { async join(roomIdOrAlias) {
this.log.step(`joins ${roomIdOrAlias}`);
const {room_id} = await this._post(`/join/${encodeURIComponent(roomIdOrAlias)}`); const {room_id} = await this._post(`/join/${encodeURIComponent(roomIdOrAlias)}`);
return new RestRoom(this, room_id); this.log.done();
return new RestRoom(this, room_id, this.log);
} }
async createRoom(name, options) { async createRoom(name, options) {
this.log.step(`creates room ${name}`);
const body = { const body = {
name, name,
}; };
@ -68,7 +75,8 @@ module.exports = class RestSession {
} }
const {room_id} = await this._post(`/createRoom`, body); const {room_id} = await this._post(`/createRoom`, body);
return new RestRoom(this, room_id); this.log.done();
return new RestRoom(this, room_id, this.log);
} }
_post(csApiPath, body) { _post(csApiPath, body) {

View File

@ -41,7 +41,7 @@ module.exports = async function scenario(createSession, restCreator) {
async function createRestUsers(restCreator) { async function createRestUsers(restCreator) {
const usernames = range(1, 10).map((i) => `charly-${i}`); const usernames = range(1, 10).map((i) => `charly-${i}`);
const charlies = await restCreator.createSessionRange(usernames, 'testtest'); const charlies = await restCreator.createSessionRange(usernames, "testtest", "charly-1..10");
await charlies.setDisplayName((s) => `Charly #${s.userName().split('-')[1]}`); await charlies.setDisplayName((s) => `Charly #${s.userName().split('-')[1]}`);
return charlies; return charlies;
} }