/* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */ import 'mocha' import * as chai from 'chai' import { cleanupTests, getVideo, registerUser, uploadVideo, userLogin, viewVideo, wait } from '../../../../shared/extra-utils' import { flushAndRunServer, setAccessTokensToServers } from '../../../../shared/extra-utils/index' import { HttpStatusCode } from '../../../../shared/core-utils/miscs/http-error-codes' const expect = chai.expect describe('Test application behind a reverse proxy', function () { let server = null let videoId before(async function () { this.timeout(30000) const config = { rates_limit: { api: { max: 50, window: 5000 }, signup: { max: 3, window: 5000 }, login: { max: 20 } }, signup: { limit: 20 } } server = await flushAndRunServer(1, config) await setAccessTokensToServers([ server ]) const { body } = await uploadVideo(server.url, server.accessToken, {}) videoId = body.video.uuid }) it('Should view a video only once with the same IP by default', async function () { this.timeout(20000) await viewVideo(server.url, videoId) await viewVideo(server.url, videoId) // Wait the repeatable job await wait(8000) const { body } = await getVideo(server.url, videoId) expect(body.views).to.equal(1) }) it('Should view a video 2 times with the X-Forwarded-For header set', async function () { this.timeout(20000) await viewVideo(server.url, videoId, HttpStatusCode.NO_CONTENT_204, '0.0.0.1,127.0.0.1') await viewVideo(server.url, videoId, HttpStatusCode.NO_CONTENT_204, '0.0.0.2,127.0.0.1') // Wait the repeatable job await wait(8000) const { body } = await getVideo(server.url, videoId) expect(body.views).to.equal(3) }) it('Should view a video only once with the same client IP in the X-Forwarded-For header', async function () { this.timeout(20000) await viewVideo(server.url, videoId, HttpStatusCode.NO_CONTENT_204, '0.0.0.4,0.0.0.3,::ffff:127.0.0.1') await viewVideo(server.url, videoId, HttpStatusCode.NO_CONTENT_204, '0.0.0.5,0.0.0.3,127.0.0.1') // Wait the repeatable job await wait(8000) const { body } = await getVideo(server.url, videoId) expect(body.views).to.equal(4) }) it('Should view a video two times with a different client IP in the X-Forwarded-For header', async function () { this.timeout(20000) await viewVideo(server.url, videoId, HttpStatusCode.NO_CONTENT_204, '0.0.0.8,0.0.0.6,127.0.0.1') await viewVideo(server.url, videoId, HttpStatusCode.NO_CONTENT_204, '0.0.0.8,0.0.0.7,127.0.0.1') // Wait the repeatable job await wait(8000) const { body } = await getVideo(server.url, videoId) expect(body.views).to.equal(6) }) it('Should rate limit logins', async function () { const user = { username: 'root', password: 'fail' } for (let i = 0; i < 19; i++) { await userLogin(server, user, HttpStatusCode.BAD_REQUEST_400) } await userLogin(server, user, HttpStatusCode.TOO_MANY_REQUESTS_429) }) it('Should rate limit signup', async function () { for (let i = 0; i < 10; i++) { try { await registerUser(server.url, 'test' + i, 'password') } catch { // empty } } await registerUser(server.url, 'test42', 'password', HttpStatusCode.TOO_MANY_REQUESTS_429) }) it('Should not rate limit failed signup', async function () { this.timeout(30000) await wait(7000) for (let i = 0; i < 3; i++) { await registerUser(server.url, 'test' + i, 'password', HttpStatusCode.CONFLICT_409) } await registerUser(server.url, 'test43', 'password', HttpStatusCode.NO_CONTENT_204) }) it('Should rate limit API calls', async function () { this.timeout(30000) await wait(7000) for (let i = 0; i < 100; i++) { try { await getVideo(server.url, videoId) } catch { // don't care if it fails } } await getVideo(server.url, videoId, HttpStatusCode.TOO_MANY_REQUESTS_429) }) after(async function () { await cleanupTests([ server ]) }) })