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

[web] Add group SSO helpers for acceptance tests

GitOrigin-RevId: f2cec120582cecb474e705b451d06e9beced7bc9
This commit is contained in:
Jessica Lawshe 2024-02-01 10:10:04 -06:00 committed by Copybot
parent 3748621bcd
commit 596596efb1
2 changed files with 33 additions and 8 deletions

View file

@ -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

View file

@ -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) {