owncast/test/automated/api/chatusers.test.js

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

const registerChat = require('./lib/chat').registerChat;
const sendChatMessage = require('./lib/chat').sendChatMessage;
const sendAdminRequest = require('./lib/admin').sendAdminRequest;
const sendAdminPayload = require('./lib/admin').sendAdminPayload;
const getAdminResponse = require('./lib/admin').getAdminResponse;


const localIPAddressV4 = '127.0.0.1';
const localIPAddressV6 = '::1';

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

var userId;
var accessToken;
test('register a user', async (done) => {
  const registration = await registerChat();
  userId = registration.id;
  accessToken = registration.accessToken;
  done();
});

test('send a chat message', async (done) => {
  sendChatMessage(testVisibilityMessage, accessToken, done);
});

test('set the user as moderator', async (done) => {
  const res = await sendAdminPayload('chat/users/setmoderator', { userId: userId, isModerator: true });
  done();
});

test('verify user is a moderator', async (done) => {
  const response = await getAdminResponse('chat/users/moderators');
  const tokenCheck = response.body.filter((user) => user.id === userId);
  expect(tokenCheck).toHaveLength(1);

  done();
});

test('verify user list is populated', async (done) => {
  const ws = new WebSocket(
    `ws://localhost:8080/ws?accessToken=${accessToken}`,
    {
      origin: 'http://localhost:8080',
    }
  );

  ws.on('open', async function open() {
    const response = await getAdminResponse('chat/clients');

    expect(response.body.length).toBeGreaterThan(0);

    // Optionally, if GeoIP is configured, check the location property.
    if (fs.existsSync('../../../data/GeoLite2-City.mmdb')) {
      expect(response.body[0].geo.regionName).toBe('Localhost');
    }

    ws.close();
  });

  ws.on('error', function incoming(data) {
    console.error(data);
    ws.close();
  });

  ws.on('close', function incoming(data) {
    done();
  });
});

test('disable a user', async (done) => {
  // To allow for visually being able to see the test hiding the
  // message add a short delay.
  await new Promise((r) => setTimeout(r, 1500));

  const ws = new WebSocket(
    `ws://localhost:8080/ws?accessToken=${accessToken}`,
    {
      origin: 'http://localhost:8080',
    }
  );

  const res = await sendAdminPayload('chat/users/setenabled', { userId: userId, enabled: false });

  await new Promise((r) => setTimeout(r, 1500));
  done();
});

test('verify user is disabled', async (done) => {
  const response = await getAdminResponse('chat/users/disabled');
  const tokenCheck = response.body.filter((user) => user.id === userId);
  expect(tokenCheck).toHaveLength(1);
  done();
});

test('verify messages from user are hidden', async (done) => {
  const response = await getAdminResponse('chat/messages');
  const message = response.body.filter((obj) => {
    return obj.user.id === userId;
  });
  expect(message[0].user.disabledAt).toBeTruthy();
  done();
});

test('re-enable a user', async (done) => {
  const res = await sendAdminPayload('chat/users/setenabled', { userId: userId, enabled: true });
  done();
});

test('verify user is enabled', async (done) => {
  const response = await getAdminResponse('chat/users/disabled');
  const tokenCheck = response.body.filter((user) => user.id === userId);
  expect(tokenCheck).toHaveLength(0);

  done();
});

test('ban an ip address', async (done) => {
  const resIPv4 = await sendAdminRequest('chat/users/ipbans/create', localIPAddressV4);
  const resIPv6 = await sendAdminRequest('chat/users/ipbans/create', localIPAddressV6);
  done();
});

test('verify IP address is blocked from the ban', async (done) => {
  const response = await getAdminResponse('chat/users/ipbans');

  expect(response.body).toHaveLength(2);
  expect(onlyLocalIPAddress(response.body)).toBe(true);
  done();
});

test('verify access is denied', async (done) => {
  await request.get(`/api/chat?accessToken=${accessToken}`).expect(401);
  done();
});

test('remove an ip address ban', async (done) => {
  const resIPv4 = await sendAdminRequest('chat/users/ipbans/remove', localIPAddressV4);
  const resIPv6 = await sendAdminRequest('chat/users/ipbans/remove', localIPAddressV6);
  done();
});

test('verify IP address is no longer banned', async (done) => {
  const response = await getAdminResponse('chat/users/ipbans');

  expect(response.body).toHaveLength(0);
  done();
});

test('verify access is allowed after unban', async (done) => {
  await request.get(`/api/chat?accessToken=${accessToken}`).expect(200);
  done();
});


// This function expects the local address to be localIPAddressV4 & localIPAddressV6
function onlyLocalIPAddress(banInfo) {
  for (let i = 0; i < banInfo.length; i++) {
    if ((banInfo[i].ipAddress != localIPAddressV4) && (banInfo[i].ipAddress != localIPAddressV6)) {
      return false
    }
  }
  return true
}