X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Ftests%2Futils%2Fpods.js;h=cdabb64a63b04c9ff28c543bc3b0cf6311ca59b4;hb=d5f5a670fcf6258d0facf9a2fd3aabd550fbc78f;hp=3664921108cfb889bc3f27e6b35c06fac8e013a7;hpb=8d30905858245f12a42fc327d2d57cbfe062d548;p=github%2FChocobozzz%2FPeerTube.git

diff --git a/server/tests/utils/pods.js b/server/tests/utils/pods.js
index 366492110..cdabb64a6 100644
--- a/server/tests/utils/pods.js
+++ b/server/tests/utils/pods.js
@@ -3,9 +3,10 @@
 const request = require('supertest')
 
 const podsUtils = {
-  getFriendsList: getFriendsList,
-  makeFriends: makeFriends,
-  quitFriends: quitFriends
+  getFriendsList,
+  makeFriends,
+  quitFriends,
+  quitOneFriend
 }
 
 // ---------------------- Export functions --------------------
@@ -27,13 +28,38 @@ function makeFriends (url, accessToken, expectedStatus, end) {
     expectedStatus = 204
   }
 
+  // Which pod makes friends with which pod
+  const friendsMatrix = {
+    'http://localhost:9001': [
+      'localhost:9002'
+    ],
+    'http://localhost:9002': [
+      'localhost:9003'
+    ],
+    'http://localhost:9003': [
+      'localhost:9001'
+    ],
+    'http://localhost:9004': [
+      'localhost:9002'
+    ],
+    'http://localhost:9005': [
+      'localhost:9001',
+      'localhost:9004'
+    ],
+    'http://localhost:9006': [
+      'localhost:9001',
+      'localhost:9002',
+      'localhost:9003'
+    ]
+  }
   const path = '/api/v1/pods/makefriends'
 
   // The first pod make friend with the third
   request(url)
-    .get(path)
+    .post(path)
     .set('Accept', 'application/json')
     .set('Authorization', 'Bearer ' + accessToken)
+    .send({ 'hosts': friendsMatrix[url] })
     .expect(expectedStatus)
     .end(function (err, res) {
       if (err) throw err
@@ -65,6 +91,26 @@ function quitFriends (url, accessToken, expectedStatus, end) {
     })
 }
 
+function quitOneFriend (url, accessToken, friendId, expectedStatus, end) {
+  if (!end) {
+    end = expectedStatus
+    expectedStatus = 204
+  }
+
+  const path = '/api/v1/pods/' + friendId
+
+  request(url)
+    .delete(path)
+    .set('Accept', 'application/json')
+    .set('Authorization', 'Bearer ' + accessToken)
+    .expect(expectedStatus)
+    .end(function (err, res) {
+      if (err) throw err
+
+      end()
+    })
+}
+
 // ---------------------------------------------------------------------------
 
 module.exports = podsUtils