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

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