using System; using System.Collections.Generic; using System.Linq; using System.Threading; using System.Threading.Tasks; using FluentAssertions; using Moq; using NUnit.Framework; using Strata.Analytics.Biz.DataSources.Models; using Strata.Analytics.Biz.DataSources.Services; using Strata.Analytics.Models.Security; using Strata.Analytics.Models.Utils; using Strata.Analytics.Test.Unit.Biz.TestData; using Strata.DataSchema.Client; using Strata.Testing.Categories; namespace Strata.Analytics.Test.Unit.Biz.DataSources { [NUnit.Framework.Category(TestCategory.Unit)] [Parallelizable(ParallelScope.Self)] public class DataSourceRoleAssignmentTest { [Test] public async Task DataSourceRoleAssignment_Created_GivenRoleAssignmentWithIdentityIsNotAlreadyInDatabase() { // arrange var claimsPrincipal = IClaimsPrincipalAccessorMock.GetClaimsPrincipalAccessor(); var dbContext = AnalyticsDbContextMock.GetAnalyticsDbContext(claimsPrincipal); var dataSourceService = new DataSourceService(dbContext, claimsPrincipal, AutoMapperMock.GetAutoMapper()); var batch = new Batch() { Updates = new List() { new DataSourceRoleAssignmentSaveInfo() { IdentityGuid = Guid.NewGuid(), IdentityName="Admin", IdentityType=IdentityType.User, Role=DataSourceRole.Editor} } }; // act await dataSourceService.SaveRoleAssignmentsAsync(5, batch, CancellationToken.None); // assert dbContext.DataSourceRoleAssignments.Should().HaveCount(1); dbContext.DataSourceRoleAssignments.FirstOrDefault().DataSourceId.Should().Be(5); dbContext.DataSourceRoleAssignments.FirstOrDefault().IdentityGuid.Should().Be(batch.Updates.First().IdentityGuid); dbContext.DataSourceRoleAssignments.FirstOrDefault().IdentityName.Should().Be(batch.Updates.First().IdentityName); dbContext.DataSourceRoleAssignments.FirstOrDefault().Role.Should().Be(batch.Updates.First().Role); } [Test] public async Task DataSourceRoleAssignment_Updated_GivenRoleAssignmentWithIdentityIsAlreadyInDatabase() { // arrange var strataId = 5; var claimsPrincipal = IClaimsPrincipalAccessorMock.GetClaimsPrincipalAccessor(strataId); var dbContext = AnalyticsDbContextMock.GetAnalyticsDbContext(claimsPrincipal); var dataSourceService = new DataSourceService(dbContext, claimsPrincipal, AutoMapperMock.GetAutoMapper()); var existingRoleAssigment = new DataSourceRoleAssignment() { StrataId = strataId, DataSourceId = 5, IdentityGuid = Guid.NewGuid(), IdentityName = "Admin", IdentityType = IdentityType.User, Role = DataSourceRole.Viewer }; dbContext.DataSourceRoleAssignments.Add(existingRoleAssigment); await dbContext.SaveChangesAsync(); // change role to Editor var batch = new Batch() { Updates = new List() { new DataSourceRoleAssignmentSaveInfo() { IdentityGuid = existingRoleAssigment.IdentityGuid, IdentityName="Admin", IdentityType=IdentityType.User, Role=DataSourceRole.Editor} } }; // act await dataSourceService.SaveRoleAssignmentsAsync(5, batch, CancellationToken.None); // assert dbContext.DataSourceRoleAssignments.Should().HaveCount(1); dbContext.DataSourceRoleAssignments.FirstOrDefault().DataSourceId.Should().Be(5); dbContext.DataSourceRoleAssignments.FirstOrDefault().IdentityGuid.Should().Be(batch.Updates.First().IdentityGuid); dbContext.DataSourceRoleAssignments.FirstOrDefault().IdentityName.Should().Be(batch.Updates.First().IdentityName); dbContext.DataSourceRoleAssignments.FirstOrDefault().Role.Should().Be(batch.Updates.First().Role); } [Test] public async Task DataSourceRoleAssignment_Deleted_GivenIdentityGuidIsPartOfBatchDeletes() { // arrange var strataId = 5; var claimsPrincipal = IClaimsPrincipalAccessorMock.GetClaimsPrincipalAccessor(strataId); var dbContext = AnalyticsDbContextMock.GetAnalyticsDbContext(claimsPrincipal); var dataSourceService = new DataSourceService(dbContext, claimsPrincipal, AutoMapperMock.GetAutoMapper()); var existingRoleAssigment = new DataSourceRoleAssignment() { StrataId = strataId, DataSourceId = 5, IdentityGuid = Guid.NewGuid(), IdentityName = "Admin", IdentityType = IdentityType.User, Role = DataSourceRole.Editor }; dbContext.DataSourceRoleAssignments.Add(existingRoleAssigment); await dbContext.SaveChangesAsync(); var batch = new Batch() { Deletes = new List() { existingRoleAssigment.IdentityGuid } }; // act await dataSourceService.SaveRoleAssignmentsAsync(5, batch, CancellationToken.None); // assert dbContext.DataSourceRoleAssignments.Should().HaveCount(0); } } }