mirror of
https://github.com/bitwarden/server.git
synced 2025-05-07 12:42:24 -05:00
[PM-17775] (#5699)
* Changes to allow admin to send F4E sponsorship * Fix the failing unit tests * Fix the failing test Signed-off-by: Cy Okeke <cokeke@bitwarden.com> * Merge Changes with pm-17777 Signed-off-by: Cy Okeke <cokeke@bitwarden.com> * Add changes for autoscale Signed-off-by: Cy Okeke <cokeke@bitwarden.com> * Return the right error response Signed-off-by: Cy Okeke <cokeke@bitwarden.com> * Resolve the failing unit test Signed-off-by: Cy Okeke <cokeke@bitwarden.com> --------- Signed-off-by: Cy Okeke <cokeke@bitwarden.com>
This commit is contained in:
parent
8ecd9c5fb3
commit
dc5db5673f
@ -1,4 +1,5 @@
|
|||||||
using Bit.Api.Models.Request.Organizations;
|
using Bit.Api.Models.Request.Organizations;
|
||||||
|
using Bit.Api.Models.Response;
|
||||||
using Bit.Api.Models.Response.Organizations;
|
using Bit.Api.Models.Response.Organizations;
|
||||||
using Bit.Core.AdminConsole.Enums;
|
using Bit.Core.AdminConsole.Enums;
|
||||||
using Bit.Core.AdminConsole.OrganizationFeatures.OrganizationConnections.Interfaces;
|
using Bit.Core.AdminConsole.OrganizationFeatures.OrganizationConnections.Interfaces;
|
||||||
@ -8,6 +9,7 @@ using Bit.Core.Entities;
|
|||||||
using Bit.Core.Exceptions;
|
using Bit.Core.Exceptions;
|
||||||
using Bit.Core.Models.Api.Request.OrganizationSponsorships;
|
using Bit.Core.Models.Api.Request.OrganizationSponsorships;
|
||||||
using Bit.Core.Models.Api.Response.OrganizationSponsorships;
|
using Bit.Core.Models.Api.Response.OrganizationSponsorships;
|
||||||
|
using Bit.Core.Models.Data.Organizations.OrganizationSponsorships;
|
||||||
using Bit.Core.OrganizationFeatures.OrganizationSponsorships.FamiliesForEnterprise.Interfaces;
|
using Bit.Core.OrganizationFeatures.OrganizationSponsorships.FamiliesForEnterprise.Interfaces;
|
||||||
using Bit.Core.Repositories;
|
using Bit.Core.Repositories;
|
||||||
using Bit.Core.Services;
|
using Bit.Core.Services;
|
||||||
@ -105,8 +107,11 @@ public class OrganizationSponsorshipsController : Controller
|
|||||||
model.FriendlyName,
|
model.FriendlyName,
|
||||||
model.IsAdminInitiated.GetValueOrDefault(),
|
model.IsAdminInitiated.GetValueOrDefault(),
|
||||||
model.Notes);
|
model.Notes);
|
||||||
|
if (sponsorship.OfferedToEmail != null)
|
||||||
|
{
|
||||||
await _sendSponsorshipOfferCommand.SendSponsorshipOfferAsync(sponsorship, sponsoringOrg.Name);
|
await _sendSponsorshipOfferCommand.SendSponsorshipOfferAsync(sponsorship, sponsoringOrg.Name);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
[Authorize("Application")]
|
[Authorize("Application")]
|
||||||
[HttpPost("{sponsoringOrgId}/families-for-enterprise/resend")]
|
[HttpPost("{sponsoringOrgId}/families-for-enterprise/resend")]
|
||||||
@ -246,5 +251,27 @@ public class OrganizationSponsorshipsController : Controller
|
|||||||
return new OrganizationSponsorshipSyncStatusResponseModel(lastSyncDate);
|
return new OrganizationSponsorshipSyncStatusResponseModel(lastSyncDate);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Authorize("Application")]
|
||||||
|
[HttpGet("{sponsoringOrgId}/sponsored")]
|
||||||
|
[SelfHosted(NotSelfHostedOnly = true)]
|
||||||
|
public async Task<ListResponseModel<OrganizationSponsorshipInvitesResponseModel>> GetSponsoredOrganizations(Guid sponsoringOrgId)
|
||||||
|
{
|
||||||
|
var sponsoringOrg = await _organizationRepository.GetByIdAsync(sponsoringOrgId);
|
||||||
|
if (sponsoringOrg == null)
|
||||||
|
{
|
||||||
|
throw new NotFoundException();
|
||||||
|
}
|
||||||
|
var organization = _currentContext.Organizations.First(x => x.Id == sponsoringOrg.Id);
|
||||||
|
if (!await _currentContext.OrganizationOwner(sponsoringOrg.Id) && !await _currentContext.OrganizationAdmin(sponsoringOrg.Id) && !organization.Permissions.ManageUsers)
|
||||||
|
{
|
||||||
|
throw new UnauthorizedAccessException();
|
||||||
|
}
|
||||||
|
|
||||||
|
var sponsorships = await _organizationSponsorshipRepository.GetManyBySponsoringOrganizationAsync(sponsoringOrgId);
|
||||||
|
return new ListResponseModel<OrganizationSponsorshipInvitesResponseModel>(sponsorships.Select(s =>
|
||||||
|
new OrganizationSponsorshipInvitesResponseModel(new OrganizationSponsorshipData(s))));
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
private Task<User> CurrentUser => _userService.GetUserByIdAsync(_currentContext.UserId.Value);
|
private Task<User> CurrentUser => _userService.GetUserByIdAsync(_currentContext.UserId.Value);
|
||||||
}
|
}
|
||||||
|
@ -0,0 +1,37 @@
|
|||||||
|
using Bit.Core.Enums;
|
||||||
|
using Bit.Core.Models.Data.Organizations.OrganizationSponsorships;
|
||||||
|
|
||||||
|
namespace Bit.Core.Models.Api.Response.OrganizationSponsorships;
|
||||||
|
|
||||||
|
public class OrganizationSponsorshipInvitesResponseModel : ResponseModel
|
||||||
|
{
|
||||||
|
public OrganizationSponsorshipInvitesResponseModel(OrganizationSponsorshipData sponsorshipData, string obj = "organizationSponsorship") : base(obj)
|
||||||
|
{
|
||||||
|
if (sponsorshipData == null)
|
||||||
|
{
|
||||||
|
throw new ArgumentNullException(nameof(sponsorshipData));
|
||||||
|
}
|
||||||
|
|
||||||
|
SponsoringOrganizationUserId = sponsorshipData.SponsoringOrganizationUserId;
|
||||||
|
FriendlyName = sponsorshipData.FriendlyName;
|
||||||
|
OfferedToEmail = sponsorshipData.OfferedToEmail;
|
||||||
|
PlanSponsorshipType = sponsorshipData.PlanSponsorshipType;
|
||||||
|
LastSyncDate = sponsorshipData.LastSyncDate;
|
||||||
|
ValidUntil = sponsorshipData.ValidUntil;
|
||||||
|
ToDelete = sponsorshipData.ToDelete;
|
||||||
|
IsAdminInitiated = sponsorshipData.IsAdminInitiated;
|
||||||
|
Notes = sponsorshipData.Notes;
|
||||||
|
CloudSponsorshipRemoved = sponsorshipData.CloudSponsorshipRemoved;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Guid SponsoringOrganizationUserId { get; set; }
|
||||||
|
public string FriendlyName { get; set; }
|
||||||
|
public string OfferedToEmail { get; set; }
|
||||||
|
public PlanSponsorshipType PlanSponsorshipType { get; set; }
|
||||||
|
public DateTime? LastSyncDate { get; set; }
|
||||||
|
public DateTime? ValidUntil { get; set; }
|
||||||
|
public bool ToDelete { get; set; }
|
||||||
|
public bool IsAdminInitiated { get; set; }
|
||||||
|
public string Notes { get; set; }
|
||||||
|
public bool CloudSponsorshipRemoved { get; set; }
|
||||||
|
}
|
@ -47,11 +47,12 @@ public class CreateSponsorshipCommand(
|
|||||||
throw new BadRequestException("Only confirmed users can sponsor other organizations.");
|
throw new BadRequestException("Only confirmed users can sponsor other organizations.");
|
||||||
}
|
}
|
||||||
|
|
||||||
var existingOrgSponsorship = await organizationSponsorshipRepository
|
var sponsorships =
|
||||||
.GetBySponsoringOrganizationUserIdAsync(sponsoringMember.Id);
|
await organizationSponsorshipRepository.GetManyBySponsoringOrganizationAsync(sponsoringOrganization.Id);
|
||||||
if (existingOrgSponsorship?.SponsoredOrganizationId != null)
|
var existingSponsorship = sponsorships.FirstOrDefault(s => s.FriendlyName == friendlyName);
|
||||||
|
if (existingSponsorship != null)
|
||||||
{
|
{
|
||||||
throw new BadRequestException("Can only sponsor one organization per Organization User.");
|
return existingSponsorship;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (isAdminInitiated)
|
if (isAdminInitiated)
|
||||||
@ -70,11 +71,21 @@ public class CreateSponsorshipCommand(
|
|||||||
Notes = notes
|
Notes = notes
|
||||||
};
|
};
|
||||||
|
|
||||||
|
if (!isAdminInitiated)
|
||||||
|
{
|
||||||
|
var existingOrgSponsorship = await organizationSponsorshipRepository
|
||||||
|
.GetBySponsoringOrganizationUserIdAsync(sponsoringMember.Id);
|
||||||
|
if (existingOrgSponsorship?.SponsoredOrganizationId != null)
|
||||||
|
{
|
||||||
|
throw new BadRequestException("Can only sponsor one organization per Organization User.");
|
||||||
|
}
|
||||||
|
|
||||||
if (existingOrgSponsorship != null)
|
if (existingOrgSponsorship != null)
|
||||||
{
|
{
|
||||||
// Replace existing invalid offer with our new sponsorship offer
|
// Replace existing invalid offer with our new sponsorship offer
|
||||||
sponsorship.Id = existingOrgSponsorship.Id;
|
sponsorship.Id = existingOrgSponsorship.Id;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if (isAdminInitiated && sponsoringOrganization.Seats.HasValue)
|
if (isAdminInitiated && sponsoringOrganization.Seats.HasValue)
|
||||||
{
|
{
|
||||||
|
@ -6,6 +6,7 @@ using Bit.Core.Context;
|
|||||||
using Bit.Core.Entities;
|
using Bit.Core.Entities;
|
||||||
using Bit.Core.Enums;
|
using Bit.Core.Enums;
|
||||||
using Bit.Core.Exceptions;
|
using Bit.Core.Exceptions;
|
||||||
|
using Bit.Core.Models.Data;
|
||||||
using Bit.Core.OrganizationFeatures.OrganizationSponsorships.FamiliesForEnterprise.Interfaces;
|
using Bit.Core.OrganizationFeatures.OrganizationSponsorships.FamiliesForEnterprise.Interfaces;
|
||||||
using Bit.Core.Repositories;
|
using Bit.Core.Repositories;
|
||||||
using Bit.Core.Services;
|
using Bit.Core.Services;
|
||||||
@ -13,6 +14,7 @@ using Bit.Core.Utilities;
|
|||||||
using Bit.Test.Common.AutoFixture;
|
using Bit.Test.Common.AutoFixture;
|
||||||
using Bit.Test.Common.AutoFixture.Attributes;
|
using Bit.Test.Common.AutoFixture.Attributes;
|
||||||
using NSubstitute;
|
using NSubstitute;
|
||||||
|
using NSubstitute.ReturnsExtensions;
|
||||||
using Xunit;
|
using Xunit;
|
||||||
|
|
||||||
namespace Bit.Api.Test.Billing.Controllers;
|
namespace Bit.Api.Test.Billing.Controllers;
|
||||||
@ -146,4 +148,80 @@ public class OrganizationSponsorshipsControllerTests
|
|||||||
.DidNotReceiveWithAnyArgs()
|
.DidNotReceiveWithAnyArgs()
|
||||||
.RemoveSponsorshipAsync(default);
|
.RemoveSponsorshipAsync(default);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Theory]
|
||||||
|
[BitAutoData]
|
||||||
|
public async Task GetSponsoredOrganizations_OrganizationNotFound_ThrowsNotFound(
|
||||||
|
Guid sponsoringOrgId,
|
||||||
|
SutProvider<OrganizationSponsorshipsController> sutProvider)
|
||||||
|
{
|
||||||
|
sutProvider.GetDependency<IOrganizationRepository>().GetByIdAsync(sponsoringOrgId).ReturnsNull();
|
||||||
|
|
||||||
|
await Assert.ThrowsAsync<NotFoundException>(() =>
|
||||||
|
sutProvider.Sut.GetSponsoredOrganizations(sponsoringOrgId));
|
||||||
|
|
||||||
|
await sutProvider.GetDependency<IOrganizationSponsorshipRepository>()
|
||||||
|
.DidNotReceiveWithAnyArgs()
|
||||||
|
.GetManyBySponsoringOrganizationAsync(default);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Theory]
|
||||||
|
[BitAutoData]
|
||||||
|
public async Task GetSponsoredOrganizations_NotOrganizationOwner_ThrowsNotFound(
|
||||||
|
Organization sponsoringOrg,
|
||||||
|
SutProvider<OrganizationSponsorshipsController> sutProvider)
|
||||||
|
{
|
||||||
|
// Arrange
|
||||||
|
sutProvider.GetDependency<IOrganizationRepository>().GetByIdAsync(sponsoringOrg.Id).Returns(sponsoringOrg);
|
||||||
|
sutProvider.GetDependency<ICurrentContext>().OrganizationOwner(sponsoringOrg.Id).Returns(false);
|
||||||
|
sutProvider.GetDependency<ICurrentContext>().OrganizationAdmin(sponsoringOrg.Id).Returns(false);
|
||||||
|
|
||||||
|
// Create a CurrentContextOrganization with ManageUsers set to false
|
||||||
|
var currentContextOrg = new CurrentContextOrganization
|
||||||
|
{
|
||||||
|
Id = sponsoringOrg.Id,
|
||||||
|
Permissions = new Permissions { ManageUsers = false }
|
||||||
|
};
|
||||||
|
sutProvider.GetDependency<ICurrentContext>().Organizations.Returns(new List<CurrentContextOrganization> { currentContextOrg });
|
||||||
|
|
||||||
|
// Act & Assert
|
||||||
|
await Assert.ThrowsAsync<UnauthorizedAccessException>(() =>
|
||||||
|
sutProvider.Sut.GetSponsoredOrganizations(sponsoringOrg.Id));
|
||||||
|
|
||||||
|
await sutProvider.GetDependency<IOrganizationSponsorshipRepository>()
|
||||||
|
.DidNotReceiveWithAnyArgs()
|
||||||
|
.GetManyBySponsoringOrganizationAsync(default);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Theory]
|
||||||
|
[BitAutoData]
|
||||||
|
public async Task GetSponsoredOrganizations_Success_ReturnsSponsorships(
|
||||||
|
Organization sponsoringOrg,
|
||||||
|
List<OrganizationSponsorship> sponsorships,
|
||||||
|
SutProvider<OrganizationSponsorshipsController> sutProvider)
|
||||||
|
{
|
||||||
|
// Arrange
|
||||||
|
sutProvider.GetDependency<IOrganizationRepository>().GetByIdAsync(sponsoringOrg.Id).Returns(sponsoringOrg);
|
||||||
|
sutProvider.GetDependency<ICurrentContext>().OrganizationOwner(sponsoringOrg.Id).Returns(true);
|
||||||
|
sutProvider.GetDependency<ICurrentContext>().OrganizationAdmin(sponsoringOrg.Id).Returns(false);
|
||||||
|
|
||||||
|
// Create a CurrentContextOrganization from the sponsoringOrg
|
||||||
|
var currentContextOrg = new CurrentContextOrganization
|
||||||
|
{
|
||||||
|
Id = sponsoringOrg.Id,
|
||||||
|
Permissions = new Permissions { ManageUsers = true }
|
||||||
|
};
|
||||||
|
sutProvider.GetDependency<ICurrentContext>().Organizations.Returns(new List<CurrentContextOrganization> { currentContextOrg });
|
||||||
|
|
||||||
|
sutProvider.GetDependency<IOrganizationSponsorshipRepository>()
|
||||||
|
.GetManyBySponsoringOrganizationAsync(sponsoringOrg.Id).Returns(sponsorships);
|
||||||
|
|
||||||
|
// Act
|
||||||
|
var result = await sutProvider.Sut.GetSponsoredOrganizations(sponsoringOrg.Id);
|
||||||
|
|
||||||
|
// Assert
|
||||||
|
Assert.Equal(sponsorships.Count, result.Data.Count());
|
||||||
|
await sutProvider.GetDependency<IOrganizationSponsorshipRepository>().Received(1)
|
||||||
|
.GetManyBySponsoringOrganizationAsync(sponsoringOrg.Id);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user