mirror of
https://github.com/bitwarden/server.git
synced 2024-12-04 14:13:28 +01:00
88dd745070
* [PM-1188] add sso project to auth * [PM-1188] move sso api models to auth * [PM-1188] fix sso api model namespace & imports * [PM-1188] move core files to auth * [PM-1188] fix core sso namespace & models * [PM-1188] move sso repository files to auth * [PM-1188] fix sso repo files namespace & imports * [PM-1188] move sso sql files to auth folder * [PM-1188] move sso test files to auth folders * [PM-1188] fix sso tests namespace & imports * [PM-1188] move auth api files to auth folder * [PM-1188] fix auth api files namespace & imports * [PM-1188] move auth core files to auth folder * [PM-1188] fix auth core files namespace & imports * [PM-1188] move auth email templates to auth folder * [PM-1188] move auth email folder back into shared directory * [PM-1188] fix auth email names * [PM-1188] move auth core models to auth folder * [PM-1188] fix auth model namespace & imports * [PM-1188] add entire Identity project to auth codeowners * [PM-1188] fix auth orm files namespace & imports * [PM-1188] move auth orm files to auth folder * [PM-1188] move auth sql files to auth folder * [PM-1188] move auth tests to auth folder * [PM-1188] fix auth test files namespace & imports * [PM-1188] move emergency access api files to auth folder * [PM-1188] fix emergencyaccess api files namespace & imports * [PM-1188] move emergency access core files to auth folder * [PM-1188] fix emergency access core files namespace & imports * [PM-1188] move emergency access orm files to auth folder * [PM-1188] fix emergency access orm files namespace & imports * [PM-1188] move emergency access sql files to auth folder * [PM-1188] move emergencyaccess test files to auth folder * [PM-1188] fix emergency access test files namespace & imports * [PM-1188] move captcha files to auth folder * [PM-1188] fix captcha files namespace & imports * [PM-1188] move auth admin files into auth folder * [PM-1188] fix admin auth files namespace & imports - configure mvc to look in auth folders for views * [PM-1188] remove extra imports and formatting * [PM-1188] fix ef auth model imports * [PM-1188] fix DatabaseContextModelSnapshot paths * [PM-1188] fix grant import in ef * [PM-1188] update sqlproj * [PM-1188] move missed sqlproj files * [PM-1188] move auth ef models out of auth folder * [PM-1188] fix auth ef models namespace * [PM-1188] remove auth ef models unused imports * [PM-1188] fix imports for auth ef models * [PM-1188] fix more ef model imports * [PM-1188] fix file encodings
229 lines
9.3 KiB
C#
229 lines
9.3 KiB
C#
using Bit.Core.Auth.Entities;
|
|
using Bit.Core.Entities;
|
|
using Bit.Core.Test.AutoFixture.Attributes;
|
|
using Bit.Infrastructure.EFIntegration.Test.Auth.AutoFixture;
|
|
using Bit.Infrastructure.EFIntegration.Test.Auth.Repositories.EqualityComparers;
|
|
using Xunit;
|
|
using EfRepo = Bit.Infrastructure.EntityFramework.Repositories;
|
|
using SqlAuthRepo = Bit.Infrastructure.Dapper.Auth.Repositories;
|
|
using SqlRepo = Bit.Infrastructure.Dapper.Repositories;
|
|
|
|
namespace Bit.Infrastructure.EFIntegration.Test.Auth.Repositories;
|
|
|
|
public class SsoConfigRepositoryTests
|
|
{
|
|
[CiSkippedTheory, EfSsoConfigAutoData]
|
|
public async void CreateAsync_Works_DataMatches(SsoConfig ssoConfig, Organization org,
|
|
SsoConfigCompare equalityComparer, List<EfRepo.SsoConfigRepository> suts,
|
|
List<EfRepo.OrganizationRepository> efOrgRepos, SqlAuthRepo.SsoConfigRepository sqlSsoConfigRepo,
|
|
SqlRepo.OrganizationRepository sqlOrganizationRepo)
|
|
{
|
|
var savedSsoConfigs = new List<SsoConfig>();
|
|
|
|
foreach (var sut in suts)
|
|
{
|
|
var i = suts.IndexOf(sut);
|
|
|
|
var savedEfOrg = await efOrgRepos[i].CreateAsync(org);
|
|
sut.ClearChangeTracking();
|
|
|
|
ssoConfig.OrganizationId = savedEfOrg.Id;
|
|
var postEfSsoConfig = await sut.CreateAsync(ssoConfig);
|
|
sut.ClearChangeTracking();
|
|
|
|
var savedEfSsoConfig = await sut.GetByIdAsync(ssoConfig.Id);
|
|
Assert.True(savedEfSsoConfig != null);
|
|
savedSsoConfigs.Add(savedEfSsoConfig);
|
|
}
|
|
|
|
var sqlOrganization = await sqlOrganizationRepo.CreateAsync(org);
|
|
ssoConfig.OrganizationId = sqlOrganization.Id;
|
|
|
|
var sqlSsoConfig = await sqlSsoConfigRepo.CreateAsync(ssoConfig);
|
|
var savedSqlSsoConfig = await sqlSsoConfigRepo.GetByIdAsync(sqlSsoConfig.Id);
|
|
Assert.True(savedSqlSsoConfig != null);
|
|
savedSsoConfigs.Add(savedSqlSsoConfig);
|
|
|
|
var distinctItems = savedSsoConfigs.Distinct(equalityComparer);
|
|
Assert.True(!distinctItems.Skip(1).Any());
|
|
}
|
|
|
|
[CiSkippedTheory, EfSsoConfigAutoData]
|
|
public async void ReplaceAsync_Works_DataMatches(SsoConfig postSsoConfig, SsoConfig replaceSsoConfig,
|
|
Organization org, SsoConfigCompare equalityComparer, List<EfRepo.SsoConfigRepository> suts,
|
|
List<EfRepo.OrganizationRepository> efOrgRepos, SqlAuthRepo.SsoConfigRepository sqlSsoConfigRepo,
|
|
SqlRepo.OrganizationRepository sqlOrganizationRepo)
|
|
{
|
|
var savedSsoConfigs = new List<SsoConfig>();
|
|
|
|
foreach (var sut in suts)
|
|
{
|
|
var i = suts.IndexOf(sut);
|
|
|
|
var savedEfOrg = await efOrgRepos[i].CreateAsync(org);
|
|
sut.ClearChangeTracking();
|
|
|
|
postSsoConfig.OrganizationId = replaceSsoConfig.OrganizationId = savedEfOrg.Id;
|
|
var postEfSsoConfig = await sut.CreateAsync(postSsoConfig);
|
|
sut.ClearChangeTracking();
|
|
|
|
replaceSsoConfig.Id = postEfSsoConfig.Id;
|
|
savedSsoConfigs.Add(postEfSsoConfig);
|
|
await sut.ReplaceAsync(replaceSsoConfig);
|
|
sut.ClearChangeTracking();
|
|
|
|
var replacedSsoConfig = await sut.GetByIdAsync(replaceSsoConfig.Id);
|
|
Assert.True(replacedSsoConfig != null);
|
|
savedSsoConfigs.Add(replacedSsoConfig);
|
|
}
|
|
|
|
var sqlOrganization = await sqlOrganizationRepo.CreateAsync(org);
|
|
postSsoConfig.OrganizationId = sqlOrganization.Id;
|
|
|
|
var postSqlSsoConfig = await sqlSsoConfigRepo.CreateAsync(postSsoConfig);
|
|
replaceSsoConfig.Id = postSqlSsoConfig.Id;
|
|
savedSsoConfigs.Add(postSqlSsoConfig);
|
|
|
|
await sqlSsoConfigRepo.ReplaceAsync(replaceSsoConfig);
|
|
var replacedSqlSsoConfig = await sqlSsoConfigRepo.GetByIdAsync(replaceSsoConfig.Id);
|
|
Assert.True(replacedSqlSsoConfig != null);
|
|
savedSsoConfigs.Add(replacedSqlSsoConfig);
|
|
|
|
var distinctItems = savedSsoConfigs.Distinct(equalityComparer);
|
|
Assert.True(!distinctItems.Skip(2).Any());
|
|
}
|
|
|
|
[CiSkippedTheory, EfSsoConfigAutoData]
|
|
public async void DeleteAsync_Works_DataMatches(SsoConfig ssoConfig, Organization org, List<EfRepo.SsoConfigRepository> suts,
|
|
List<EfRepo.OrganizationRepository> efOrgRepos, SqlAuthRepo.SsoConfigRepository sqlSsoConfigRepo,
|
|
SqlRepo.OrganizationRepository sqlOrganizationRepo)
|
|
{
|
|
foreach (var sut in suts)
|
|
{
|
|
var i = suts.IndexOf(sut);
|
|
|
|
var savedEfOrg = await efOrgRepos[i].CreateAsync(org);
|
|
sut.ClearChangeTracking();
|
|
|
|
ssoConfig.OrganizationId = savedEfOrg.Id;
|
|
var postEfSsoConfig = await sut.CreateAsync(ssoConfig);
|
|
sut.ClearChangeTracking();
|
|
|
|
var savedEfSsoConfig = await sut.GetByIdAsync(postEfSsoConfig.Id);
|
|
Assert.True(savedEfSsoConfig != null);
|
|
sut.ClearChangeTracking();
|
|
|
|
await sut.DeleteAsync(savedEfSsoConfig);
|
|
var deletedEfSsoConfig = await sut.GetByIdAsync(savedEfSsoConfig.Id);
|
|
Assert.True(deletedEfSsoConfig == null);
|
|
}
|
|
|
|
var sqlOrganization = await sqlOrganizationRepo.CreateAsync(org);
|
|
ssoConfig.OrganizationId = sqlOrganization.Id;
|
|
|
|
var postSqlSsoConfig = await sqlSsoConfigRepo.CreateAsync(ssoConfig);
|
|
var savedSqlSsoConfig = await sqlSsoConfigRepo.GetByIdAsync(postSqlSsoConfig.Id);
|
|
Assert.True(savedSqlSsoConfig != null);
|
|
|
|
await sqlSsoConfigRepo.DeleteAsync(savedSqlSsoConfig);
|
|
savedSqlSsoConfig = await sqlSsoConfigRepo.GetByIdAsync(postSqlSsoConfig.Id);
|
|
Assert.True(savedSqlSsoConfig == null);
|
|
}
|
|
|
|
[CiSkippedTheory, EfSsoConfigAutoData]
|
|
public async void GetByOrganizationIdAsync_Works_DataMatches(SsoConfig ssoConfig, Organization org,
|
|
SsoConfigCompare equalityComparer, List<EfRepo.SsoConfigRepository> suts,
|
|
List<EfRepo.OrganizationRepository> efOrgRepos, SqlAuthRepo.SsoConfigRepository sqlSsoConfigRepo,
|
|
SqlRepo.OrganizationRepository sqlOrgRepo)
|
|
{
|
|
var returnedList = new List<SsoConfig>();
|
|
|
|
foreach (var sut in suts)
|
|
{
|
|
var i = suts.IndexOf(sut);
|
|
|
|
var savedEfOrg = await efOrgRepos[i].CreateAsync(org);
|
|
sut.ClearChangeTracking();
|
|
|
|
ssoConfig.OrganizationId = savedEfOrg.Id;
|
|
await sut.CreateAsync(ssoConfig);
|
|
sut.ClearChangeTracking();
|
|
|
|
var savedEfUser = await sut.GetByOrganizationIdAsync(savedEfOrg.Id);
|
|
Assert.True(savedEfUser != null);
|
|
returnedList.Add(savedEfUser);
|
|
}
|
|
|
|
var savedSqlOrg = await sqlOrgRepo.CreateAsync(org);
|
|
ssoConfig.OrganizationId = savedSqlOrg.Id;
|
|
|
|
var postSqlSsoConfig = await sqlSsoConfigRepo.CreateAsync(ssoConfig);
|
|
|
|
var savedSqlSsoConfig = await sqlSsoConfigRepo.GetByOrganizationIdAsync(ssoConfig.OrganizationId);
|
|
Assert.True(savedSqlSsoConfig != null);
|
|
returnedList.Add(savedSqlSsoConfig);
|
|
|
|
var distinctItems = returnedList.Distinct(equalityComparer);
|
|
Assert.True(!distinctItems.Skip(1).Any());
|
|
}
|
|
|
|
[CiSkippedTheory, EfSsoConfigAutoData]
|
|
public async void GetByIdentifierAsync_Works_DataMatches(SsoConfig ssoConfig, Organization org,
|
|
SsoConfigCompare equalityComparer, List<EfRepo.SsoConfigRepository> suts,
|
|
List<EfRepo.OrganizationRepository> efOrgRepos, SqlAuthRepo.SsoConfigRepository sqlSsoConfigRepo,
|
|
SqlRepo.OrganizationRepository sqlOrgRepo)
|
|
{
|
|
var returnedList = new List<SsoConfig>();
|
|
|
|
foreach (var sut in suts)
|
|
{
|
|
var i = suts.IndexOf(sut);
|
|
|
|
var savedEfOrg = await efOrgRepos[i].CreateAsync(org);
|
|
sut.ClearChangeTracking();
|
|
|
|
ssoConfig.OrganizationId = savedEfOrg.Id;
|
|
await sut.CreateAsync(ssoConfig);
|
|
sut.ClearChangeTracking();
|
|
|
|
var savedEfSsoConfig = await sut.GetByIdentifierAsync(org.Identifier);
|
|
Assert.True(savedEfSsoConfig != null);
|
|
returnedList.Add(savedEfSsoConfig);
|
|
}
|
|
|
|
var savedSqlOrg = await sqlOrgRepo.CreateAsync(org);
|
|
ssoConfig.OrganizationId = savedSqlOrg.Id;
|
|
|
|
var postSqlSsoConfig = await sqlSsoConfigRepo.CreateAsync(ssoConfig);
|
|
|
|
var savedSqlSsoConfig = await sqlSsoConfigRepo.GetByIdentifierAsync(org.Identifier);
|
|
Assert.True(savedSqlSsoConfig != null);
|
|
returnedList.Add(savedSqlSsoConfig);
|
|
|
|
var distinctItems = returnedList.Distinct(equalityComparer);
|
|
Assert.True(!distinctItems.Skip(1).Any());
|
|
}
|
|
|
|
// Testing that data matches here would involve manipulating all SsoConfig records in the db
|
|
[CiSkippedTheory, EfSsoConfigAutoData]
|
|
public async void GetManyByRevisionNotBeforeDate_Works(SsoConfig ssoConfig, DateTime notBeforeDate,
|
|
Organization org, List<EfRepo.SsoConfigRepository> suts,
|
|
List<EfRepo.OrganizationRepository> efOrgRepos)
|
|
{
|
|
foreach (var sut in suts)
|
|
{
|
|
var i = suts.IndexOf(sut);
|
|
|
|
var savedEfOrg = await efOrgRepos[i].CreateAsync(org);
|
|
sut.ClearChangeTracking();
|
|
|
|
ssoConfig.OrganizationId = savedEfOrg.Id;
|
|
await sut.CreateAsync(ssoConfig);
|
|
sut.ClearChangeTracking();
|
|
|
|
var returnedEfSsoConfigs = await sut.GetManyByRevisionNotBeforeDate(notBeforeDate);
|
|
Assert.True(returnedEfSsoConfigs.All(sc => sc.RevisionDate >= notBeforeDate));
|
|
}
|
|
}
|
|
}
|