diff --git a/services/web/test/acceptance/src/helpers/Subscription.js b/services/web/test/acceptance/src/helpers/Subscription.js index b31290c52c..c44d77c418 100644 --- a/services/web/test/acceptance/src/helpers/Subscription.js +++ b/services/web/test/acceptance/src/helpers/Subscription.js @@ -83,6 +83,29 @@ class Subscription { 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) { Modules.hooks.fire( '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 diff --git a/services/web/test/acceptance/src/helpers/User.js b/services/web/test/acceptance/src/helpers/User.js index 451c183d15..e44973b8ae 100644 --- a/services/web/test/acceptance/src/helpers/User.js +++ b/services/web/test/acceptance/src/helpers/User.js @@ -229,14 +229,6 @@ class User { 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) { this.getCsrfToken(error => { if (error != null) {