Merge branch 'release/4.1.0' into develop

[github/Chocobozzz/PeerTube.git] / server / tests / helpers / request.ts

diff --git a/server/tests/helpers/request.ts b/server/tests/helpers/request.ts
index c9a2eb8310e1e5e557b513cd09e78938ba872b15..de507ba35f85d18e3153cf8a884698b2c314706d 100644 (file)
--- a/server/tests/helpers/request.ts
+++ b/server/tests/helpers/request.ts
@@ -4,8 +4,9 @@ import 'mocha'
 import { expect } from 'chai'
 import { pathExists, remove } from 'fs-extra'
 import { join } from 'path'
-import { FIXTURE_URLS, root, wait } from '../../../shared/extra-utils'
+import { root, wait } from '@shared/core-utils'
 import { doRequest, doRequestAndSaveToFile } from '../../helpers/requests'
+import { FIXTURE_URLS, Mock429 } from '../shared'
 
 describe('Request helpers', function () {
   const destPath1 = join(root(), 'test-output-1.txt')
@@ -34,6 +35,20 @@ describe('Request helpers', function () {
     throw new Error('No error thrown by do request and save to file')
   })
 
+  it('Should correctly retry on 429 error', async function () {
+    this.timeout(25000)
+
+    const mock = new Mock429()
+    const port = await mock.initialize()
+
+    const before = new Date().getTime()
+    await doRequest('http://localhost:' + port)
+
+    expect(new Date().getTime() - before).to.be.greaterThan(2000)
+
+    await mock.terminate()
+  })
+
   it('Should succeed if the file is below the limit', async function () {
     await doRequest(FIXTURE_URLS.file4K, { bodyKBLimit: 5 })
     await doRequestAndSaveToFile(FIXTURE_URLS.file4K, destPath2, { bodyKBLimit: 5 })