Remove one pod (#76)

[github/Chocobozzz/PeerTube.git] / server / tests / utils / pods.js

diff --git a/server/tests/utils/pods.js b/server/tests/utils/pods.js
index 3664921108cfb889bc3f27e6b35c06fac8e013a7..cdabb64a63b04c9ff28c543bc3b0cf6311ca59b4 100644 (file)
--- 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