analytics/tests/Strata.Analytics.Test.Unit/Biz/DataSources/DataSourceRoleAssignmentTest.cs

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<DataSourceRoleAssignmentSaveInfo, Guid>()
            {
                Updates = new List<DataSourceRoleAssignmentSaveInfo>()
                {
                    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<DataSourceRoleAssignmentSaveInfo, Guid>()
            {
                Updates = new List<DataSourceRoleAssignmentSaveInfo>()
                {
                    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<DataSourceRoleAssignmentSaveInfo, Guid>()
            {
                Deletes = new List<Guid>() { existingRoleAssigment.IdentityGuid }
            };

            // act
            await dataSourceService.SaveRoleAssignmentsAsync(5, batch, CancellationToken.None);

            // assert
            dbContext.DataSourceRoleAssignments.Should().HaveCount(0);
        }

    }
}