mirror of
https://github.com/bitwarden/server.git
synced 2025-06-28 06:36:15 -05:00
Add UpdateCollectionCommand and associated interface with validation logic
* Implement UpdateCollectionCommand to handle collection updates with organization checks and access permissions. * Introduce IUpdateCollectionCommand interface for defining the collection update contract. * Add unit tests for UpdateCollectionCommand to validate various scenarios including permission checks and error handling.
This commit is contained in:
parent
448692fc6e
commit
2e2cf2e0df
@ -0,0 +1,17 @@
|
|||||||
|
using Bit.Core.Entities;
|
||||||
|
using Bit.Core.Models.Data;
|
||||||
|
|
||||||
|
namespace Bit.Core.OrganizationFeatures.OrganizationCollections.Interfaces;
|
||||||
|
|
||||||
|
public interface IUpdateCollectionCommand
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// Updates a collection.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="collection">The collection to update.</param>
|
||||||
|
/// <param name="groups">(Optional) The groups that will have access to the collection.</param>
|
||||||
|
/// <param name="users">(Optional) The users that will have access to the collection.</param>
|
||||||
|
/// <returns>The updated collection.</returns>
|
||||||
|
Task<Collection> UpdateAsync(Collection collection, IEnumerable<CollectionAccessSelection> groups = null,
|
||||||
|
IEnumerable<CollectionAccessSelection> users = null);
|
||||||
|
}
|
@ -0,0 +1,59 @@
|
|||||||
|
using Bit.Core.Entities;
|
||||||
|
using Bit.Core.Exceptions;
|
||||||
|
using Bit.Core.Models.Data;
|
||||||
|
using Bit.Core.OrganizationFeatures.OrganizationCollections.Interfaces;
|
||||||
|
using Bit.Core.Repositories;
|
||||||
|
using Bit.Core.Services;
|
||||||
|
|
||||||
|
namespace Bit.Core.OrganizationFeatures.OrganizationCollections;
|
||||||
|
|
||||||
|
public class UpdateCollectionCommand : IUpdateCollectionCommand
|
||||||
|
{
|
||||||
|
private readonly IEventService _eventService;
|
||||||
|
private readonly IOrganizationRepository _organizationRepository;
|
||||||
|
private readonly ICollectionRepository _collectionRepository;
|
||||||
|
|
||||||
|
public UpdateCollectionCommand(
|
||||||
|
IEventService eventService,
|
||||||
|
IOrganizationRepository organizationRepository,
|
||||||
|
ICollectionRepository collectionRepository)
|
||||||
|
{
|
||||||
|
_eventService = eventService;
|
||||||
|
_organizationRepository = organizationRepository;
|
||||||
|
_collectionRepository = collectionRepository;
|
||||||
|
}
|
||||||
|
|
||||||
|
public async Task<Collection> UpdateAsync(Collection collection, IEnumerable<CollectionAccessSelection> groups = null,
|
||||||
|
IEnumerable<CollectionAccessSelection> users = null)
|
||||||
|
{
|
||||||
|
var org = await _organizationRepository.GetByIdAsync(collection.OrganizationId);
|
||||||
|
if (org == null)
|
||||||
|
{
|
||||||
|
throw new BadRequestException("Organization not found");
|
||||||
|
}
|
||||||
|
|
||||||
|
var groupsList = groups?.ToList();
|
||||||
|
var usersList = users?.ToList();
|
||||||
|
|
||||||
|
// Cannot use Manage with ReadOnly/HidePasswords permissions
|
||||||
|
var invalidAssociations = groupsList?.Where(cas => cas.Manage && (cas.ReadOnly || cas.HidePasswords));
|
||||||
|
if (invalidAssociations?.Any() ?? false)
|
||||||
|
{
|
||||||
|
throw new BadRequestException("The Manage property is mutually exclusive and cannot be true while the ReadOnly or HidePasswords properties are also true.");
|
||||||
|
}
|
||||||
|
|
||||||
|
// A collection should always have someone with Can Manage permissions
|
||||||
|
var groupHasManageAccess = groupsList?.Any(g => g.Manage) ?? false;
|
||||||
|
var userHasManageAccess = usersList?.Any(u => u.Manage) ?? false;
|
||||||
|
if (!groupHasManageAccess && !userHasManageAccess && !org.AllowAdminAccessToAllCollectionItems)
|
||||||
|
{
|
||||||
|
throw new BadRequestException(
|
||||||
|
"At least one member or group must have can manage permission.");
|
||||||
|
}
|
||||||
|
|
||||||
|
await _collectionRepository.ReplaceAsync(collection, org.UseGroups ? groupsList : null, usersList);
|
||||||
|
await _eventService.LogCollectionEventAsync(collection, Enums.EventType.Collection_Updated);
|
||||||
|
|
||||||
|
return collection;
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,169 @@
|
|||||||
|
using Bit.Core.AdminConsole.Entities;
|
||||||
|
using Bit.Core.Entities;
|
||||||
|
using Bit.Core.Enums;
|
||||||
|
using Bit.Core.Exceptions;
|
||||||
|
using Bit.Core.Models.Data;
|
||||||
|
using Bit.Core.OrganizationFeatures.OrganizationCollections;
|
||||||
|
using Bit.Core.Repositories;
|
||||||
|
using Bit.Core.Services;
|
||||||
|
using Bit.Core.Test.AutoFixture;
|
||||||
|
using Bit.Core.Test.AutoFixture.OrganizationFixtures;
|
||||||
|
using Bit.Test.Common.AutoFixture;
|
||||||
|
using Bit.Test.Common.AutoFixture.Attributes;
|
||||||
|
using NSubstitute;
|
||||||
|
using Xunit;
|
||||||
|
|
||||||
|
namespace Bit.Core.Test.OrganizationFeatures.OrganizationCollections;
|
||||||
|
|
||||||
|
[SutProviderCustomize]
|
||||||
|
[OrganizationCustomize]
|
||||||
|
public class UpdateCollectionCommandTests
|
||||||
|
{
|
||||||
|
[Theory, BitAutoData]
|
||||||
|
public async Task UpdateAsync_WithoutGroupsAndUsers_ReplacesCollection(
|
||||||
|
Organization organization, Collection collection, SutProvider<UpdateCollectionCommand> sutProvider)
|
||||||
|
{
|
||||||
|
var creationDate = collection.CreationDate;
|
||||||
|
sutProvider.GetDependency<IOrganizationRepository>()
|
||||||
|
.GetByIdAsync(organization.Id)
|
||||||
|
.Returns(organization);
|
||||||
|
var utcNow = DateTime.UtcNow;
|
||||||
|
|
||||||
|
await sutProvider.Sut.UpdateAsync(collection, null, null);
|
||||||
|
|
||||||
|
await sutProvider.GetDependency<ICollectionRepository>()
|
||||||
|
.Received(1)
|
||||||
|
.ReplaceAsync(
|
||||||
|
collection,
|
||||||
|
Arg.Is<List<CollectionAccessSelection>>(l => l == null),
|
||||||
|
Arg.Is<List<CollectionAccessSelection>>(l => l == null));
|
||||||
|
await sutProvider.GetDependency<IEventService>()
|
||||||
|
.Received(1)
|
||||||
|
.LogCollectionEventAsync(collection, EventType.Collection_Updated);
|
||||||
|
Assert.Equal(collection.CreationDate, creationDate);
|
||||||
|
Assert.True(collection.RevisionDate - utcNow < TimeSpan.FromSeconds(1));
|
||||||
|
}
|
||||||
|
|
||||||
|
[Theory, BitAutoData]
|
||||||
|
public async Task UpdateAsync_WithGroupsAndUsers_ReplacesCollectionWithGroupsAndUsers(
|
||||||
|
Organization organization, Collection collection,
|
||||||
|
[CollectionAccessSelectionCustomize(true)] IEnumerable<CollectionAccessSelection> groups,
|
||||||
|
IEnumerable<CollectionAccessSelection> users,
|
||||||
|
SutProvider<UpdateCollectionCommand> sutProvider)
|
||||||
|
{
|
||||||
|
var creationDate = collection.CreationDate;
|
||||||
|
organization.UseGroups = true;
|
||||||
|
sutProvider.GetDependency<IOrganizationRepository>()
|
||||||
|
.GetByIdAsync(organization.Id)
|
||||||
|
.Returns(organization);
|
||||||
|
var utcNow = DateTime.UtcNow;
|
||||||
|
|
||||||
|
await sutProvider.Sut.UpdateAsync(collection, groups, users);
|
||||||
|
|
||||||
|
await sutProvider.GetDependency<ICollectionRepository>()
|
||||||
|
.Received(1)
|
||||||
|
.ReplaceAsync(
|
||||||
|
collection,
|
||||||
|
Arg.Is<List<CollectionAccessSelection>>(l => l.Any(i => i.Manage == true)),
|
||||||
|
Arg.Any<List<CollectionAccessSelection>>());
|
||||||
|
await sutProvider.GetDependency<IEventService>()
|
||||||
|
.Received(1)
|
||||||
|
.LogCollectionEventAsync(collection, EventType.Collection_Updated);
|
||||||
|
Assert.Equal(collection.CreationDate, creationDate);
|
||||||
|
Assert.True(collection.RevisionDate - utcNow < TimeSpan.FromSeconds(1));
|
||||||
|
}
|
||||||
|
|
||||||
|
[Theory, BitAutoData]
|
||||||
|
public async Task UpdateAsync_WithOrganizationUseGroupDisabled_ReplacesCollectionWithoutGroups(
|
||||||
|
Organization organization, Collection collection,
|
||||||
|
[CollectionAccessSelectionCustomize] IEnumerable<CollectionAccessSelection> groups,
|
||||||
|
[CollectionAccessSelectionCustomize(true)] IEnumerable<CollectionAccessSelection> users,
|
||||||
|
SutProvider<UpdateCollectionCommand> sutProvider)
|
||||||
|
{
|
||||||
|
var creationDate = collection.CreationDate;
|
||||||
|
organization.UseGroups = false;
|
||||||
|
sutProvider.GetDependency<IOrganizationRepository>()
|
||||||
|
.GetByIdAsync(organization.Id)
|
||||||
|
.Returns(organization);
|
||||||
|
var utcNow = DateTime.UtcNow;
|
||||||
|
|
||||||
|
await sutProvider.Sut.UpdateAsync(collection, groups, users);
|
||||||
|
|
||||||
|
await sutProvider.GetDependency<ICollectionRepository>()
|
||||||
|
.Received(1)
|
||||||
|
.ReplaceAsync(
|
||||||
|
collection,
|
||||||
|
Arg.Is<List<CollectionAccessSelection>>(l => l == null),
|
||||||
|
Arg.Is<List<CollectionAccessSelection>>(l => l.Any(i => i.Manage == true)));
|
||||||
|
await sutProvider.GetDependency<IEventService>()
|
||||||
|
.Received(1)
|
||||||
|
.LogCollectionEventAsync(collection, EventType.Collection_Updated);
|
||||||
|
Assert.Equal(collection.CreationDate, creationDate);
|
||||||
|
Assert.True(collection.RevisionDate - utcNow < TimeSpan.FromSeconds(1));
|
||||||
|
}
|
||||||
|
|
||||||
|
[Theory, BitAutoData]
|
||||||
|
public async Task UpdateAsync_WithNonExistingOrganizationId_ThrowsBadRequest(
|
||||||
|
Collection collection, SutProvider<UpdateCollectionCommand> sutProvider)
|
||||||
|
{
|
||||||
|
var ex = await Assert.ThrowsAsync<BadRequestException>(() => sutProvider.Sut.UpdateAsync(collection));
|
||||||
|
Assert.Contains("Organization not found", ex.Message);
|
||||||
|
await sutProvider.GetDependency<ICollectionRepository>()
|
||||||
|
.DidNotReceiveWithAnyArgs()
|
||||||
|
.ReplaceAsync(default);
|
||||||
|
await sutProvider.GetDependency<ICollectionRepository>()
|
||||||
|
.DidNotReceiveWithAnyArgs()
|
||||||
|
.ReplaceAsync(default, default, default);
|
||||||
|
await sutProvider.GetDependency<IEventService>()
|
||||||
|
.DidNotReceiveWithAnyArgs()
|
||||||
|
.LogCollectionEventAsync(default, default);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Theory, BitAutoData]
|
||||||
|
public async Task UpdateAsync_WithoutManageAccess_ThrowsBadRequest(
|
||||||
|
Organization organization, Collection collection,
|
||||||
|
[CollectionAccessSelectionCustomize] IEnumerable<CollectionAccessSelection> users,
|
||||||
|
SutProvider<UpdateCollectionCommand> sutProvider)
|
||||||
|
{
|
||||||
|
organization.AllowAdminAccessToAllCollectionItems = false;
|
||||||
|
sutProvider.GetDependency<IOrganizationRepository>()
|
||||||
|
.GetByIdAsync(organization.Id)
|
||||||
|
.Returns(organization);
|
||||||
|
|
||||||
|
var ex = await Assert.ThrowsAsync<BadRequestException>(() => sutProvider.Sut.UpdateAsync(collection, null, users));
|
||||||
|
Assert.Contains("At least one member or group must have can manage permission.", ex.Message);
|
||||||
|
await sutProvider.GetDependency<ICollectionRepository>()
|
||||||
|
.DidNotReceiveWithAnyArgs()
|
||||||
|
.ReplaceAsync(default);
|
||||||
|
await sutProvider.GetDependency<ICollectionRepository>()
|
||||||
|
.DidNotReceiveWithAnyArgs()
|
||||||
|
.ReplaceAsync(default, default, default);
|
||||||
|
await sutProvider.GetDependency<IEventService>()
|
||||||
|
.DidNotReceiveWithAnyArgs()
|
||||||
|
.LogCollectionEventAsync(default, default);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Theory, BitAutoData]
|
||||||
|
public async Task UpdateAsync_WithInvalidManageAssociations_ThrowsBadRequest(
|
||||||
|
Organization organization, Collection collection, SutProvider<UpdateCollectionCommand> sutProvider)
|
||||||
|
{
|
||||||
|
sutProvider.GetDependency<IOrganizationRepository>().GetByIdAsync(organization.Id).Returns(organization);
|
||||||
|
|
||||||
|
var invalidGroups = new List<CollectionAccessSelection>
|
||||||
|
{
|
||||||
|
new() { Id = Guid.NewGuid(), Manage = true, HidePasswords = true }
|
||||||
|
};
|
||||||
|
|
||||||
|
var ex = await Assert.ThrowsAsync<BadRequestException>(() => sutProvider.Sut.UpdateAsync(collection, invalidGroups, null));
|
||||||
|
Assert.Contains("The Manage property is mutually exclusive and cannot be true while the ReadOnly or HidePasswords properties are also true.", ex.Message);
|
||||||
|
await sutProvider.GetDependency<ICollectionRepository>()
|
||||||
|
.DidNotReceiveWithAnyArgs()
|
||||||
|
.ReplaceAsync(default);
|
||||||
|
await sutProvider.GetDependency<ICollectionRepository>()
|
||||||
|
.DidNotReceiveWithAnyArgs()
|
||||||
|
.ReplaceAsync(default, default, default);
|
||||||
|
await sutProvider.GetDependency<IEventService>()
|
||||||
|
.DidNotReceiveWithAnyArgs()
|
||||||
|
.LogCollectionEventAsync(default, default);
|
||||||
|
}
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user