X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Ftests%2Fapi%2Fserver%2Freverse-proxy.ts;h=484f88d674a89da3d16d164763f8ce161eacd740;hb=4c7e60bc17ee5830399bac4aa273356903421b4c;hp=4c2655f64215726408c6e75cf257d33039bd8363;hpb=09becad8e85cc4007ff862b26fa4195f3d0e7c8d;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/tests/api/server/reverse-proxy.ts b/server/tests/api/server/reverse-proxy.ts index 4c2655f64..484f88d67 100644 --- a/server/tests/api/server/reverse-proxy.ts +++ b/server/tests/api/server/reverse-proxy.ts @@ -1,82 +1,146 @@ -/* tslint:disable:no-unused-expression */ - -import 'mocha' -import * as chai from 'chai' -import { About } from '../../../../shared/models/server/about.model' -import { CustomConfig } from '../../../../shared/models/server/custom-config.model' -import { deleteCustomConfig, getAbout, getVideo, killallServers, login, reRunServer, uploadVideo, userLogin, viewVideo } from '../../utils' -const expect = chai.expect - -import { - getConfig, - flushTests, - runServer, - registerUser, getCustomConfig, setAccessTokensToServers, updateCustomConfig -} from '../../utils/index' +/* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */ + +import { expect } from 'chai' +import { cleanupTests, createSingleServer, PeerTubeServer, setAccessTokensToServers, wait } from '@shared/extra-utils' +import { HttpStatusCode } from '@shared/models' describe('Test application behind a reverse proxy', function () { - let server = null - let videoId + let server: PeerTubeServer + let videoId: string before(async function () { this.timeout(30000) - await flushTests() - server = await runServer(1) + const config = { + rates_limit: { + api: { + max: 50, + window: 5000 + }, + signup: { + max: 3, + window: 5000 + }, + login: { + max: 20 + } + }, + signup: { + limit: 20 + } + } + + server = await createSingleServer(1, config) await setAccessTokensToServers([ server ]) - const { body } = await uploadVideo(server.url, server.accessToken, {}) - videoId = body.video.uuid + const { uuid } = await server.videos.upload() + videoId = uuid }) it('Should view a video only once with the same IP by default', async function () { - await viewVideo(server.url, videoId) - await viewVideo(server.url, videoId) + this.timeout(20000) + + await server.videos.view({ id: videoId }) + await server.videos.view({ id: videoId }) - const { body } = await getVideo(server.url, videoId) - expect(body.views).to.equal(1) + // Wait the repeatable job + await wait(8000) + + const video = await server.videos.get({ id: videoId }) + expect(video.views).to.equal(1) }) it('Should view a video 2 times with the X-Forwarded-For header set', async function () { - await viewVideo(server.url, videoId, 204, '0.0.0.1,127.0.0.1') - await viewVideo(server.url, videoId, 204, '0.0.0.2,127.0.0.1') + this.timeout(20000) + + await server.videos.view({ id: videoId, xForwardedFor: '0.0.0.1,127.0.0.1' }) + await server.videos.view({ id: videoId, xForwardedFor: '0.0.0.2,127.0.0.1' }) - const { body } = await getVideo(server.url, videoId) - expect(body.views).to.equal(3) + // Wait the repeatable job + await wait(8000) + + const video = await server.videos.get({ id: videoId }) + expect(video.views).to.equal(3) }) it('Should view a video only once with the same client IP in the X-Forwarded-For header', async function () { - await viewVideo(server.url, videoId, 204, '0.0.0.4,0.0.0.3,::ffff:127.0.0.1') - await viewVideo(server.url, videoId, 204, '0.0.0.5,0.0.0.3,127.0.0.1') + this.timeout(20000) + + await server.videos.view({ id: videoId, xForwardedFor: '0.0.0.4,0.0.0.3,::ffff:127.0.0.1' }) + await server.videos.view({ id: videoId, xForwardedFor: '0.0.0.5,0.0.0.3,127.0.0.1' }) - const { body } = await getVideo(server.url, videoId) - expect(body.views).to.equal(4) + // Wait the repeatable job + await wait(8000) + + const video = await server.videos.get({ id: videoId }) + expect(video.views).to.equal(4) }) it('Should view a video two times with a different client IP in the X-Forwarded-For header', async function () { - await viewVideo(server.url, videoId, 204, '0.0.0.8,0.0.0.6,127.0.0.1') - await viewVideo(server.url, videoId, 204, '0.0.0.8,0.0.0.7,127.0.0.1') + this.timeout(20000) + + await server.videos.view({ id: videoId, xForwardedFor: '0.0.0.8,0.0.0.6,127.0.0.1' }) + await server.videos.view({ id: videoId, xForwardedFor: '0.0.0.8,0.0.0.7,127.0.0.1' }) - const { body } = await getVideo(server.url, videoId) - expect(body.views).to.equal(6) + // Wait the repeatable job + await wait(8000) + + const video = await server.videos.get({ id: videoId }) + expect(video.views).to.equal(6) }) it('Should rate limit logins', async function () { const user = { username: 'root', password: 'fail' } - for (let i = 0; i < 14; i++) { - await userLogin(server, user, 400) + for (let i = 0; i < 19; i++) { + await server.login.login({ user, expectedStatus: HttpStatusCode.BAD_REQUEST_400 }) } - await userLogin(server, user, 429) + await server.login.login({ user, expectedStatus: HttpStatusCode.TOO_MANY_REQUESTS_429 }) }) - after(async function () { - process.kill(-server.app.pid) + it('Should rate limit signup', async function () { + for (let i = 0; i < 10; i++) { + try { + await server.users.register({ username: 'test' + i }) + } catch { + // empty + } + } + + await server.users.register({ username: 'test42', expectedStatus: HttpStatusCode.TOO_MANY_REQUESTS_429 }) + }) - // Keep the logs if the test failed - if (this['ok']) { - await flushTests() + it('Should not rate limit failed signup', async function () { + this.timeout(30000) + + await wait(7000) + + for (let i = 0; i < 3; i++) { + await server.users.register({ username: 'test' + i, expectedStatus: HttpStatusCode.CONFLICT_409 }) } + + await server.users.register({ username: 'test43', expectedStatus: 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 server.videos.get({ id: videoId }) + } catch { + // don't care if it fails + } + } + + await server.videos.get({ id: videoId, expectedStatus: HttpStatusCode.TOO_MANY_REQUESTS_429 }) + }) + + after(async function () { + await cleanupTests([ server ]) }) })