2022-12-02 20:24:30 +01:00
|
|
|
|
using Bit.Core.Entities;
|
|
|
|
|
using Bit.Core.Enums;
|
|
|
|
|
using Bit.Core.Repositories;
|
|
|
|
|
using Xunit;
|
|
|
|
|
|
|
|
|
|
namespace Bit.Infrastructure.IntegrationTest.Repositories;
|
|
|
|
|
|
|
|
|
|
public class OrganizationUserRepositoryTests
|
|
|
|
|
{
|
|
|
|
|
[DatabaseTheory, DatabaseData]
|
|
|
|
|
public async Task DeleteAsync_Works(IUserRepository userRepository,
|
|
|
|
|
IOrganizationRepository organizationRepository,
|
|
|
|
|
IOrganizationUserRepository organizationUserRepository,
|
|
|
|
|
ITestDatabaseHelper helper)
|
|
|
|
|
{
|
|
|
|
|
var user = await userRepository.CreateAsync(new User
|
|
|
|
|
{
|
|
|
|
|
Name = "Test User",
|
2023-05-30 19:25:55 +02:00
|
|
|
|
Email = $"test+{Guid.NewGuid()}@email.com",
|
2022-12-02 20:24:30 +01:00
|
|
|
|
ApiKey = "TEST",
|
|
|
|
|
SecurityStamp = "stamp",
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
var organization = await organizationRepository.CreateAsync(new Organization
|
|
|
|
|
{
|
|
|
|
|
Name = "Test Org",
|
2023-05-30 19:25:55 +02:00
|
|
|
|
BillingEmail = user.Email, // TODO: EF does not enfore this being NOT NULL
|
|
|
|
|
Plan = "Test", // TODO: EF does not enforce this being NOT NULl
|
2022-12-02 20:24:30 +01:00
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
var orgUser = await organizationUserRepository.CreateAsync(new OrganizationUser
|
|
|
|
|
{
|
|
|
|
|
OrganizationId = organization.Id,
|
|
|
|
|
UserId = user.Id,
|
|
|
|
|
Status = OrganizationUserStatusType.Confirmed,
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
helper.ClearTracker();
|
|
|
|
|
|
|
|
|
|
await organizationUserRepository.DeleteAsync(orgUser);
|
|
|
|
|
|
|
|
|
|
var newUser = await userRepository.GetByIdAsync(user.Id);
|
|
|
|
|
Assert.NotEqual(newUser.AccountRevisionDate, user.AccountRevisionDate);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[DatabaseTheory, DatabaseData]
|
|
|
|
|
public async Task DeleteManyAsync_Works(IUserRepository userRepository,
|
|
|
|
|
IOrganizationRepository organizationRepository,
|
|
|
|
|
IOrganizationUserRepository organizationUserRepository,
|
|
|
|
|
ITestDatabaseHelper helper)
|
|
|
|
|
{
|
|
|
|
|
var user1 = await userRepository.CreateAsync(new User
|
|
|
|
|
{
|
|
|
|
|
Name = "Test User 1",
|
2023-05-30 19:25:55 +02:00
|
|
|
|
Email = $"test+{Guid.NewGuid()}@email.com",
|
2022-12-02 20:24:30 +01:00
|
|
|
|
ApiKey = "TEST",
|
|
|
|
|
SecurityStamp = "stamp",
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
var user2 = await userRepository.CreateAsync(new User
|
|
|
|
|
{
|
|
|
|
|
Name = "Test User 2",
|
2023-05-30 19:25:55 +02:00
|
|
|
|
Email = $"test+{Guid.NewGuid()}@email.com",
|
2022-12-02 20:24:30 +01:00
|
|
|
|
ApiKey = "TEST",
|
|
|
|
|
SecurityStamp = "stamp",
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
var organization = await organizationRepository.CreateAsync(new Organization
|
|
|
|
|
{
|
|
|
|
|
Name = "Test Org",
|
2023-05-30 19:25:55 +02:00
|
|
|
|
BillingEmail = user1.Email, // TODO: EF does not enforce this being NOT NULl
|
|
|
|
|
Plan = "Test", // TODO: EF does not enforce this being NOT NULl
|
2022-12-02 20:24:30 +01:00
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
var orgUser1 = await organizationUserRepository.CreateAsync(new OrganizationUser
|
|
|
|
|
{
|
|
|
|
|
OrganizationId = organization.Id,
|
|
|
|
|
UserId = user1.Id,
|
2023-05-30 19:25:55 +02:00
|
|
|
|
Status = OrganizationUserStatusType.Confirmed,
|
2022-12-02 20:24:30 +01:00
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
var orgUser2 = await organizationUserRepository.CreateAsync(new OrganizationUser
|
|
|
|
|
{
|
|
|
|
|
OrganizationId = organization.Id,
|
|
|
|
|
UserId = user2.Id,
|
2023-05-30 19:25:55 +02:00
|
|
|
|
Status = OrganizationUserStatusType.Confirmed,
|
2022-12-02 20:24:30 +01:00
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
helper.ClearTracker();
|
|
|
|
|
|
|
|
|
|
await organizationUserRepository.DeleteManyAsync(new List<Guid>
|
|
|
|
|
{
|
|
|
|
|
orgUser1.Id,
|
|
|
|
|
orgUser2.Id,
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
var updatedUser1 = await userRepository.GetByIdAsync(user1.Id);
|
|
|
|
|
var updatedUser2 = await userRepository.GetByIdAsync(user2.Id);
|
|
|
|
|
|
|
|
|
|
Assert.NotEqual(updatedUser1.AccountRevisionDate, user1.AccountRevisionDate);
|
|
|
|
|
Assert.NotEqual(updatedUser2.AccountRevisionDate, user2.AccountRevisionDate);
|
|
|
|
|
}
|
|
|
|
|
}
|