server/tests/api/views/videos-views-cleaner.ts

   1 /* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */
   2
   3 import * as chai from 'chai'
   4 import { wait } from '@shared/core-utils'
   5 import {
   6   cleanupTests,
   7   createMultipleServers,
   8   doubleFollow,
   9   killallServers,
  10   PeerTubeServer,
  11   setAccessTokensToServers,
  12   waitJobs
  13 } from '@shared/server-commands'
  14
  15 const expect = chai.expect
  16
  17 describe('Test video views cleaner', function () {
  18   let servers: PeerTubeServer[]
  19
  20   let videoIdServer1: string
  21   let videoIdServer2: string
  22
  23   before(async function () {
  24     this.timeout(120000)
  25
  26     servers = await createMultipleServers(2)
  27     await setAccessTokensToServers(servers)
  28
  29     await doubleFollow(servers[0], servers[1])
  30
  31     videoIdServer1 = (await servers[0].videos.quickUpload({ name: 'video server 1' })).uuid
  32     videoIdServer2 = (await servers[1].videos.quickUpload({ name: 'video server 2' })).uuid
  33
  34     await waitJobs(servers)
  35
  36     await servers[0].views.simulateView({ id: videoIdServer1 })
  37     await servers[1].views.simulateView({ id: videoIdServer1 })
  38     await servers[0].views.simulateView({ id: videoIdServer2 })
  39     await servers[1].views.simulateView({ id: videoIdServer2 })
  40
  41     await waitJobs(servers)
  42   })
  43
  44   it('Should not clean old video views', async function () {
  45     this.timeout(50000)
  46
  47     await killallServers([ servers[0] ])
  48
  49     await servers[0].run({ views: { videos: { remote: { max_age: '10 days' } } } })
  50
  51     await wait(6000)
  52
  53     // Should still have views
  54
  55     {
  56       for (const server of servers) {
  57         const total = await server.sql.countVideoViewsOf(videoIdServer1)
  58         expect(total).to.equal(2, 'Server ' + server.serverNumber + ' does not have the correct amount of views')
  59       }
  60     }
  61
  62     {
  63       for (const server of servers) {
  64         const total = await server.sql.countVideoViewsOf(videoIdServer2)
  65         expect(total).to.equal(2, 'Server ' + server.serverNumber + ' does not have the correct amount of views')
  66       }
  67     }
  68   })
  69
  70   it('Should clean old video views', async function () {
  71     this.timeout(50000)
  72
  73     await killallServers([ servers[0] ])
  74
  75     await servers[0].run({ views: { videos: { remote: { max_age: '5 seconds' } } } })
  76
  77     await wait(6000)
  78
  79     // Should still have views
  80
  81     {
  82       for (const server of servers) {
  83         const total = await server.sql.countVideoViewsOf(videoIdServer1)
  84         expect(total).to.equal(2)
  85       }
  86     }
  87
  88     {
  89       const totalServer1 = await servers[0].sql.countVideoViewsOf(videoIdServer2)
  90       expect(totalServer1).to.equal(0)
  91
  92       const totalServer2 = await servers[1].sql.countVideoViewsOf(videoIdServer2)
  93       expect(totalServer2).to.equal(2)
  94     }
  95   })
  96
  97   after(async function () {
  98     await cleanupTests(servers)
  99   })
 100 })