PeerTube/server/tests/api/friends-advanced.js

/* eslint-disable no-unused-expressions */

'use strict'

const chai = require('chai')
const each = require('async/each')
const expect = chai.expect
const series = require('async/series')

const loginUtils = require('../utils/login')
const podsUtils = require('../utils/pods')
const serversUtils = require('../utils/servers')
const videosUtils = require('../utils/videos')

describe('Test advanced friends', function () {
  let servers = []

  function makeFriends (podNumber, callback) {
    const server = servers[podNumber - 1]
    return podsUtils.makeFriends(server.url, server.accessToken, callback)
  }

  function quitFriends (podNumber, callback) {
    const server = servers[podNumber - 1]
    return podsUtils.quitFriends(server.url, server.accessToken, callback)
  }

  function removeFriend (podNumber, podNumberToRemove, callback) {
    const server = servers[podNumber - 1]
    const serverToRemove = servers[podNumberToRemove - 1]

    getFriendsList(podNumber, function (err, res) {
      if (err) throw err

      let friendsList = res.body.data
      let podToRemove = friendsList.find((friend) => (friend.host === serverToRemove.host))

      return podsUtils.quitOneFriend(server.url, server.accessToken, podToRemove.id, callback)
    })
  }

  function getFriendsList (podNumber, end) {
    const server = servers[podNumber - 1]
    return podsUtils.getFriendsList(server.url, end)
  }

  function uploadVideo (podNumber, callback) {
    const videoAttributes = {
      tags: [ 'tag1', 'tag2' ]
    }
    const server = servers[podNumber - 1]

    return videosUtils.uploadVideo(server.url, server.accessToken, videoAttributes, callback)
  }

  function getVideos (podNumber, callback) {
    return videosUtils.getVideosList(servers[podNumber - 1].url, callback)
  }

  // ---------------------------------------------------------------

  before(function (done) {
    this.timeout(120000)
    serversUtils.flushAndRunMultipleServers(6, function (serversRun, urlsRun) {
      servers = serversRun

      each(servers, function (server, callbackEach) {
        loginUtils.loginAndGetAccessToken(server, function (err, accessToken) {
          if (err) return callbackEach(err)

          server.accessToken = accessToken
          callbackEach()
        })
      }, done)
    })
  })

  it('Should make friends with two pod each in a different group', function (done) {
    this.timeout(20000)

    series([
      // Pod 3 makes friend with the first one
      function (next) {
        makeFriends(3, next)
      },
      // Pod 4 makes friend with the second one
      function (next) {
        makeFriends(4, next)
      },
      // Now if the fifth wants to make friends with the third et the first
      function (next) {
        makeFriends(5, next)
      },
      function (next) {
        setTimeout(next, 11000)
      }],
      function (err) {
        if (err) throw err

        // It should have 0 friends
        getFriendsList(5, function (err, res) {
          if (err) throw err

          expect(res.body.data.length).to.equal(0)

          done()
        })
      }
    )
  })

  it('Should quit all friends', function (done) {
    this.timeout(10000)

    series([
      function (next) {
        quitFriends(1, next)
      },
      function (next) {
        quitFriends(2, next)
      }],
      function (err) {
        if (err) throw err

        each([ 1, 2, 3, 4, 5, 6 ], function (i, callback) {
          getFriendsList(i, function (err, res) {
            if (err) throw err

            expect(res.body.data.length).to.equal(0)

            callback()
          })
        }, done)
      }
    )
  })

  it('Should make friends with the pods 1, 2, 3', function (done) {
    this.timeout(150000)

    series([
      // Pods 1, 2, 3 and 4 become friends
      function (next) {
        makeFriends(2, next)
      },
      function (next) {
        makeFriends(1, next)
      },
      function (next) {
        makeFriends(4, next)
      },
      // Check the pods 1, 2, 3 and 4 are friends
      function (next) {
        each([ 1, 2, 3, 4 ], function (i, callback) {
          getFriendsList(i, function (err, res) {
            if (err) throw err

            expect(res.body.data.length).to.equal(3)

            callback()
          })
        }, next)
      },
      // Kill pod 4
      function (next) {
        servers[3].app.kill()
        next()
      },
      // Expulse pod 4 from pod 1 and 2
      function (next) {
        uploadVideo(1, next)
      },
      function (next) {
        uploadVideo(2, next)
      },
      function (next) {
        setTimeout(next, 11000)
      },
      function (next) {
        uploadVideo(1, next)
      },
      function (next) {
        uploadVideo(2, next)
      },
      function (next) {
        setTimeout(next, 22000)
      },
      // Check the pods 1, 2 expulsed pod 4
      function (next) {
        each([ 1, 2 ], function (i, callback) {
          getFriendsList(i, function (err, res) {
            if (err) throw err

            // Pod 4 should not be our friend
            const result = res.body.data
            expect(result.length).to.equal(2)
            for (const pod of result) {
              expect(pod.host).not.equal(servers[3].host)
            }

            callback()
          })
        }, next)
      },
      // Rerun server 4
      function (next) {
        serversUtils.runServer(4, function (server) {
          servers[3].app = server.app
          next()
        })
      },
      function (next) {
        getFriendsList(4, function (err, res) {
          if (err) throw err

          // Pod 4 didn't know pod 1 and 2 removed it
          expect(res.body.data.length).to.equal(3)
          next()
        })
      },
      // Pod 6 asks pod 1, 2 and 3
      function (next) {
        makeFriends(6, next)
      },
      function (next) {
        setTimeout(next, 11000)
      }],
      function (err) {
        if (err) throw err

        getFriendsList(6, function (err, res) {
          if (err) throw err

          // Pod 4 should not be our friend
          const result = res.body.data
          expect(result.length).to.equal(3)
          for (const pod of result) {
            expect(pod.host).not.equal(servers[3].host)
          }

          done()
        })
      }
    )
  })

  it('Should pod 1 quit friends', function (done) {
    this.timeout(25000)

    series([
      // Upload a video on server 3 for aditionnal tests
      function (next) {
        uploadVideo(3, next)
      },
      function (next) {
        setTimeout(next, 15000)
      },
      function (next) {
        quitFriends(1, next)
      },
      // Remove pod 1 from pod 2
      function (next) {
        getVideos(1, function (err, res) {
          if (err) throw err

          const videos = res.body.data
          expect(videos).to.be.an('array')
          expect(videos.length).to.equal(2)

          next()
        })
      }],
      function (err) {
        if (err) throw err

        getVideos(2, function (err, res) {
          if (err) throw err

          const videos = res.body.data
          expect(videos).to.be.an('array')
          expect(videos.length).to.equal(3)
          done()
        })
      }
    )
  })

  it('Should make friends between pod 1 and 2 and exchange their videos', function (done) {
    this.timeout(20000)
    makeFriends(1, function () {
      setTimeout(function () {
        getVideos(1, function (err, res) {
          if (err) throw err

          const videos = res.body.data
          expect(videos).to.be.an('array')
          expect(videos.length).to.equal(5)

          done()
        })
      }, 11000)
    })
  })

  it('Should allow pod 6 to quit pod 1 & 2 and be friend with pod 3', function (done) {
    this.timeout(30000)

    series([
      // Pod 3 should have 4 friends
      function (next) {
        getFriendsList(3, function (err, res) {
          if (err) throw err

          const friendsList = res.body.data
          expect(friendsList).to.be.an('array')
          expect(friendsList.length).to.equal(4)

          next()
	})
      },
      // Pod 1, 2, 6 should have 3 friends each
      function (next) {
	each([ 1, 2, 6 ], function (i, callback) {
          getFriendsList(i, function (err, res) {
            if (err) throw err

            const friendsList = res.body.data
            expect(friendsList).to.be.an('array')
            expect(friendsList.length).to.equal(3)

            callback()
          })
        }, next)
      },
      function (next) {
        removeFriend(6, 1, next)
      },
      function (next) {
        removeFriend(6, 2, next)
      },
      // Pod 6 should now have only 1 friend (and it should be Pod 3)
      function (next) {
        getFriendsList(6, function (err, res) {
          if (err) throw err

          const friendsList = res.body.data
          expect(friendsList).to.be.an('array')
          expect(friendsList.length).to.equal(1)
          expect(friendsList[0].host).to.equal(servers[2].host)

          next()
	})
      },
      // Pod 1 & 2 should not know friend 6 anymore
      function (next) {
        each([ 1, 2 ], function (i, callback) {
          getFriendsList(i, function (err, res) {
            if (err) throw err

            const friendsList = res.body.data
            expect(friendsList).to.be.an('array')
            expect(friendsList.length).to.equal(2)

            callback()
          })
        }, next)
      },
      // Pod 3 should know every pod
      function (next) {
        getFriendsList(3, function (err, res) {
          if (err) throw err

          const friendsList = res.body.data
          expect(friendsList).to.be.an('array')
          expect(friendsList.length).to.equal(4)

          next()
        })
      }
    ], done)
  })

  after(function (done) {
    servers.forEach(function (server) {
      process.kill(-server.app.pid)
    })

    if (this.ok) {
      serversUtils.flushTests(done)
    } else {
      done()
    }
  })
})