Sfoglia il codice sorgente

Merge pull request #16860 from overleaf/jel-sso-acceptance-tests

[web] Add group SSO helpers for acceptance tests

GitOrigin-RevId: f2cec120582cecb474e705b451d06e9beced7bc9
Jessica Lawshe 2 anni fa
parent
commit
596596efb1

+ 33 - 0
services/web/test/acceptance/src/helpers/Subscription.js

@@ -83,6 +83,29 @@ class Subscription {
     Modules.hooks.fire('enableManagedUsers', this._id, callback)
     Modules.hooks.fire('enableManagedUsers', this._id, callback)
   }
   }
 
 
+  enableFeatureSSO(callback) {
+    SubscriptionModel.findOneAndUpdate(
+      { _id: new ObjectId(this._id) },
+      { 'features.groupSSO': true },
+      callback
+    )
+  }
+
+  setValidatedAndEnabledSSO(callback) {
+    db.subscriptions.findOne({ _id: new ObjectId(this._id) }, (error, doc) => {
+      if (error) {
+        return callback(error)
+      }
+      const ssoConfigId = doc.ssoConfig
+
+      db.ssoConfigs.findOneAndUpdate(
+        { _id: ssoConfigId },
+        { $set: { enabled: true, validated: true } },
+        callback
+      )
+    })
+  }
+
   getEnrollmentForUser(user, callback) {
   getEnrollmentForUser(user, callback) {
     Modules.hooks.fire(
     Modules.hooks.fire(
       'getManagedUsersEnrollmentForUser',
       'getManagedUsersEnrollmentForUser',
@@ -158,4 +181,14 @@ Object.getOwnPropertyNames(Subscription.prototype).forEach(methodName => {
   }
   }
 })
 })
 
 
+Subscription.promises.prototype.inviteUser = async function (adminUser, email) {
+  await adminUser.login()
+  return await adminUser.doRequest('POST', {
+    url: `/manage/groups/${this._id}/invites`,
+    json: {
+      email,
+    },
+  })
+}
+
 module.exports = Subscription
 module.exports = Subscription

+ 0 - 8
services/web/test/acceptance/src/helpers/User.js

@@ -229,14 +229,6 @@ class User {
     UserModel.updateOne({ _id: this.id }, { emails }, callback)
     UserModel.updateOne({ _id: this.id }, { emails }, callback)
   }
   }
 
 
-  setEnrollment(enrollment, callback) {
-    UserModel.updateOne({ _id: this.id }, { enrollment }, callback)
-  }
-
-  setSamlIdentifiers(samlIdentifiers, callback) {
-    UserModel.updateOne({ _id: this.id }, { samlIdentifiers }, callback)
-  }
-
   logout(callback) {
   logout(callback) {
     this.getCsrfToken(error => {
     this.getCsrfToken(error => {
       if (error != null) {
       if (error != null) {