owncast/test/automated/api/chatmoderation.test.js

const { test } = require('@jest/globals');
var request = require('supertest');
request = request('http://127.0.0.1:8080');
const WebSocket = require('ws');

const registerChat = require('./lib/chat').registerChat;
const sendChatMessage = require('./lib/chat').sendChatMessage;

const testVisibilityMessage = {
	body: 'message ' + Math.floor(Math.random() * 100),
	type: 'CHAT',
};

var messageId;

const establishedUserFailedChatMessage = {
	body: 'this message should fail to send ' + Math.floor(Math.random() * 100),
	type: 'CHAT',
};

test('can send a chat message', async (done) => {
	const registration = await registerChat();
	const accessToken = registration.accessToken;

	sendChatMessage(testVisibilityMessage, accessToken, done);
});

test('verify we can make API call to mark message as hidden', async (done) => {
	const registration = await registerChat();
	const accessToken = registration.accessToken;
	const ws = new WebSocket(
		`ws://localhost:8080/ws?accessToken=${accessToken}`,
		{
			origin: 'http://localhost:8080',
		}
	);

	// Verify the visibility change comes through the websocket
	ws.on('message', async function incoming(message) {
		const messages = message.split('\n');
		messages.forEach(async function (message) {
			const event = JSON.parse(message);

			if (event.type === 'VISIBILITY-UPDATE') {
				ws.close();
				done();
			}
		});
	});

	const res = await request
		.get('/api/admin/chat/messages')
		.auth('admin', 'abc123')
		.expect(200);

	const message = res.body[0];
	messageId = message.id;
	await request
		.post('/api/admin/chat/messagevisibility')
		.auth('admin', 'abc123')
		.send({ idArray: [messageId], visible: false })
		.expect(200);
});

test('verify message has become hidden', async (done) => {
	await new Promise((r) => setTimeout(r, 2000));

	const res = await request
		.get('/api/admin/chat/messages')
		.expect(200)
		.auth('admin', 'abc123');

	const message = res.body.filter((obj) => {
		return obj.id === messageId;
	});
	expect(message.length).toBe(1);
	// expect(message[0].hiddenAt).toBeTruthy();
	done();
});

test('can enable established chat user mode', async (done) => {
	await request
		.post('/api/admin/config/chat/establishedusermode')
		.auth('admin', 'abc123')
		.send({ value: true })
		.expect(200);
	done();
});

test('can send a message after established user mode is enabled', async (done) => {
	const registration = await registerChat();
	const accessToken = registration.accessToken;

	sendChatMessage(establishedUserFailedChatMessage, accessToken, done);
});

test('verify rejected message is not in the chat feed', async (done) => {
	const res = await request
		.get('/api/admin/chat/messages')
		.expect(200)
		.auth('admin', 'abc123');

	const message = res.body.filter((obj) => {
		return obj.body === establishedUserFailedChatMessage.body;
	});

	expect(message.length).toBe(0);
	done();
});

test('can disable established chat user mode', async (done) => {
	await request
		.post('/api/admin/config/chat/establishedusermode')
		.auth('admin', 'abc123')
		.send({ value: false })
		.expect(200);
	done();
});