bitwarden-server/test/Infrastructure.EFIntegration.Test/Repositories/OrganizationSponsorshipRepositoryTests.cs

using Bit.Core.AdminConsole.Entities;
using Bit.Core.Entities;
using Bit.Core.Test.AutoFixture.Attributes;
using Bit.Infrastructure.EFIntegration.Test.AutoFixture;
using Bit.Infrastructure.EFIntegration.Test.Repositories.EqualityComparers;
using Xunit;
using EfRepo = Bit.Infrastructure.EntityFramework.Repositories;
using SqlRepo = Bit.Infrastructure.Dapper.Repositories;

namespace Bit.Infrastructure.EFIntegration.Test.Repositories;

public class OrganizationSponsorshipRepositoryTests
{
    [CiSkippedTheory, EfOrganizationSponsorshipAutoData]
    public async void CreateAsync_Works_DataMatches(
        OrganizationSponsorship organizationSponsorship, Organization sponsoringOrg,
        List<EfRepo.OrganizationRepository> efOrgRepos,
        SqlRepo.OrganizationRepository sqlOrganizationRepo,
        SqlRepo.OrganizationSponsorshipRepository sqlOrganizationSponsorshipRepo,
        OrganizationSponsorshipCompare equalityComparer,
        List<EfRepo.OrganizationSponsorshipRepository> suts)
    {
        organizationSponsorship.SponsoredOrganizationId = null;

        var savedOrganizationSponsorships = new List<OrganizationSponsorship>();
        foreach (var (sut, orgRepo) in suts.Zip(efOrgRepos))
        {
            var efSponsoringOrg = await orgRepo.CreateAsync(sponsoringOrg);
            sut.ClearChangeTracking();
            organizationSponsorship.SponsoringOrganizationId = efSponsoringOrg.Id;

            await sut.CreateAsync(organizationSponsorship);
            sut.ClearChangeTracking();

            var savedOrganizationSponsorship = await sut.GetByIdAsync(organizationSponsorship.Id);
            savedOrganizationSponsorships.Add(savedOrganizationSponsorship);
        }

        var sqlSponsoringOrg = await sqlOrganizationRepo.CreateAsync(sponsoringOrg);
        organizationSponsorship.SponsoringOrganizationId = sqlSponsoringOrg.Id;

        var sqlOrganizationSponsorship = await sqlOrganizationSponsorshipRepo.CreateAsync(organizationSponsorship);
        savedOrganizationSponsorships.Add(await sqlOrganizationSponsorshipRepo.GetByIdAsync(sqlOrganizationSponsorship.Id));

        var distinctItems = savedOrganizationSponsorships.Distinct(equalityComparer);
        Assert.True(!distinctItems.Skip(1).Any());
    }

    [CiSkippedTheory, EfOrganizationSponsorshipAutoData]
    public async void ReplaceAsync_Works_DataMatches(OrganizationSponsorship postOrganizationSponsorship,
        OrganizationSponsorship replaceOrganizationSponsorship, Organization sponsoringOrg,
        List<EfRepo.OrganizationRepository> efOrgRepos,
        SqlRepo.OrganizationRepository sqlOrganizationRepo,
        SqlRepo.OrganizationSponsorshipRepository sqlOrganizationSponsorshipRepo,
        OrganizationSponsorshipCompare equalityComparer, List<EfRepo.OrganizationSponsorshipRepository> suts)
    {
        postOrganizationSponsorship.SponsoredOrganizationId = null;
        replaceOrganizationSponsorship.SponsoredOrganizationId = null;

        var savedOrganizationSponsorships = new List<OrganizationSponsorship>();
        foreach (var (sut, orgRepo) in suts.Zip(efOrgRepos))
        {
            var efSponsoringOrg = await orgRepo.CreateAsync(sponsoringOrg);
            sut.ClearChangeTracking();
            postOrganizationSponsorship.SponsoringOrganizationId = efSponsoringOrg.Id;
            replaceOrganizationSponsorship.SponsoringOrganizationId = efSponsoringOrg.Id;

            var postEfOrganizationSponsorship = await sut.CreateAsync(postOrganizationSponsorship);
            sut.ClearChangeTracking();

            replaceOrganizationSponsorship.Id = postEfOrganizationSponsorship.Id;
            await sut.ReplaceAsync(replaceOrganizationSponsorship);
            sut.ClearChangeTracking();

            var replacedOrganizationSponsorship = await sut.GetByIdAsync(replaceOrganizationSponsorship.Id);
            savedOrganizationSponsorships.Add(replacedOrganizationSponsorship);
        }

        var sqlSponsoringOrg = await sqlOrganizationRepo.CreateAsync(sponsoringOrg);
        postOrganizationSponsorship.SponsoringOrganizationId = sqlSponsoringOrg.Id;

        var postSqlOrganization = await sqlOrganizationSponsorshipRepo.CreateAsync(postOrganizationSponsorship);
        replaceOrganizationSponsorship.Id = postSqlOrganization.Id;
        await sqlOrganizationSponsorshipRepo.ReplaceAsync(replaceOrganizationSponsorship);
        savedOrganizationSponsorships.Add(await sqlOrganizationSponsorshipRepo.GetByIdAsync(replaceOrganizationSponsorship.Id));

        var distinctItems = savedOrganizationSponsorships.Distinct(equalityComparer);
        Assert.True(!distinctItems.Skip(1).Any());
    }

    [CiSkippedTheory, EfOrganizationSponsorshipAutoData]
    public async void DeleteAsync_Works_DataMatches(OrganizationSponsorship organizationSponsorship,
        Organization sponsoringOrg,
        List<EfRepo.OrganizationRepository> efOrgRepos,
        SqlRepo.OrganizationRepository sqlOrganizationRepo,
        SqlRepo.OrganizationSponsorshipRepository sqlOrganizationSponsorshipRepo,
        List<EfRepo.OrganizationSponsorshipRepository> suts)
    {
        organizationSponsorship.SponsoredOrganizationId = null;

        foreach (var (sut, orgRepo) in suts.Zip(efOrgRepos))
        {
            var efSponsoringOrg = await orgRepo.CreateAsync(sponsoringOrg);
            sut.ClearChangeTracking();
            organizationSponsorship.SponsoringOrganizationId = efSponsoringOrg.Id;

            var postEfOrganizationSponsorship = await sut.CreateAsync(organizationSponsorship);
            sut.ClearChangeTracking();

            var savedEfOrganizationSponsorship = await sut.GetByIdAsync(postEfOrganizationSponsorship.Id);
            sut.ClearChangeTracking();
            Assert.True(savedEfOrganizationSponsorship != null);

            await sut.DeleteAsync(savedEfOrganizationSponsorship);
            sut.ClearChangeTracking();

            savedEfOrganizationSponsorship = await sut.GetByIdAsync(savedEfOrganizationSponsorship.Id);
            Assert.True(savedEfOrganizationSponsorship == null);
        }

        var sqlSponsoringOrg = await sqlOrganizationRepo.CreateAsync(sponsoringOrg);
        organizationSponsorship.SponsoringOrganizationId = sqlSponsoringOrg.Id;

        var postSqlOrganizationSponsorship = await sqlOrganizationSponsorshipRepo.CreateAsync(organizationSponsorship);
        var savedSqlOrganizationSponsorship = await sqlOrganizationSponsorshipRepo.GetByIdAsync(postSqlOrganizationSponsorship.Id);
        Assert.True(savedSqlOrganizationSponsorship != null);

        await sqlOrganizationSponsorshipRepo.DeleteAsync(postSqlOrganizationSponsorship);
        savedSqlOrganizationSponsorship = await sqlOrganizationSponsorshipRepo.GetByIdAsync(postSqlOrganizationSponsorship.Id);
        Assert.True(savedSqlOrganizationSponsorship == null);
    }
}