server/tests/api/server/slow-follows.ts

   1 /* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */
   2
   3 import 'mocha'
   4 import * as chai from 'chai'
   5 import { Job } from '@shared/models'
   6 import {
   7   cleanupTests,
   8   createMultipleServers,
   9   doubleFollow,
  10   PeerTubeServer,
  11   setAccessTokensToServers,
  12   waitJobs
  13 } from '@shared/server-commands'
  14
  15 const expect = chai.expect
  16
  17 describe('Test slow follows', function () {
  18   let servers: PeerTubeServer[] = []
  19
  20   let afterFollows: Date
  21
  22   before(async function () {
  23     this.timeout(60000)
  24
  25     servers = await createMultipleServers(3)
  26
  27     // Get the access tokens
  28     await setAccessTokensToServers(servers)
  29
  30     await doubleFollow(servers[0], servers[1])
  31     await doubleFollow(servers[0], servers[2])
  32
  33     afterFollows = new Date()
  34
  35     for (let i = 0; i < 5; i++) {
  36       await servers[0].videos.quickUpload({ name: 'video ' + i })
  37     }
  38
  39     await waitJobs(servers)
  40   })
  41
  42   it('Should only have broadcast jobs', async function () {
  43     const { data } = await servers[0].jobs.list({ jobType: 'activitypub-http-unicast', sort: '-createdAt' })
  44
  45     for (const job of data) {
  46       expect(new Date(job.createdAt)).below(afterFollows)
  47     }
  48   })
  49
  50   it('Should process bad follower', async function () {
  51     this.timeout(30000)
  52
  53     await servers[1].kill()
  54
  55     // Set server 2 as bad follower
  56     await servers[0].videos.quickUpload({ name: 'video 6' })
  57     await waitJobs(servers[0])
  58
  59     afterFollows = new Date()
  60     const filter = (job: Job) => new Date(job.createdAt) > afterFollows
  61
  62     // Resend another broadcast job
  63     await servers[0].videos.quickUpload({ name: 'video 7' })
  64     await waitJobs(servers[0])
  65
  66     const resBroadcast = await servers[0].jobs.list({ jobType: 'activitypub-http-broadcast', sort: '-createdAt' })
  67     const resUnicast = await servers[0].jobs.list({ jobType: 'activitypub-http-unicast', sort: '-createdAt' })
  68
  69     const broadcast = resBroadcast.data.filter(filter)
  70     const unicast = resUnicast.data.filter(filter)
  71
  72     expect(unicast).to.have.lengthOf(2)
  73     expect(broadcast).to.have.lengthOf(2)
  74
  75     for (const u of unicast) {
  76       expect(u.data.uri).to.equal(servers[1].url + '/inbox')
  77     }
  78
  79     for (const b of broadcast) {
  80       expect(b.data.uris).to.have.lengthOf(1)
  81       expect(b.data.uris[0]).to.equal(servers[2].url + '/inbox')
  82     }
  83   })
  84
  85   after(async function () {
  86     await cleanupTests(servers)
  87   })
  88 })