1
0
mirror of https://github.com/bitwarden/server.git synced 2024-12-04 14:13:28 +01:00
bitwarden-server/test/Core.Test/OrganizationFeatures/OrganizationLicenses/SelfHostedGetOrganizationLicenseQueryTests.cs
2024-02-22 11:59:08 -05:00

96 lines
3.9 KiB
C#

using System.Text.Json;
using Bit.Core.AdminConsole.Entities;
using Bit.Core.Entities;
using Bit.Core.Exceptions;
using Bit.Core.Models.Business;
using Bit.Core.Models.OrganizationConnectionConfigs;
using Bit.Core.OrganizationFeatures.OrganizationLicenses;
using Bit.Core.Settings;
using Bit.Core.Test.AutoFixture;
using Bit.Test.Common.AutoFixture;
using Bit.Test.Common.AutoFixture.Attributes;
using Bit.Test.Common.Helpers;
using Xunit;
namespace Bit.Core.Test.OrganizationFeatures.OrganizationLicenses;
[SutProviderCustomize]
public class SelfHostedGetOrganizationLicenseQueryTests
{
private static SutProvider<SelfHostedGetOrganizationLicenseQuery> GetSutProvider(BillingSyncConfig config,
string apiResponse = null)
{
return new SutProvider<SelfHostedGetOrganizationLicenseQuery>()
.ConfigureBaseIdentityClientService($"licenses/organization/{config.CloudOrganizationId}",
HttpMethod.Get, apiResponse: apiResponse);
}
[Theory]
[BitAutoData]
[OrganizationLicenseCustomize]
public async Task GetLicenseAsync_Success(Organization organization,
OrganizationConnection<BillingSyncConfig> billingSyncConnection, BillingSyncConfig config, OrganizationLicense license)
{
var sutProvider = GetSutProvider(config, JsonSerializer.Serialize(license));
billingSyncConnection.Enabled = true;
billingSyncConnection.Config = config;
var result = await sutProvider.Sut.GetLicenseAsync(organization, billingSyncConnection);
AssertHelper.AssertPropertyEqual(result, license);
}
[Theory]
[BitAutoData]
public async Task GetLicenseAsync_WhenNotSelfHosted_Throws(Organization organization,
OrganizationConnection billingSyncConnection, BillingSyncConfig config)
{
var sutProvider = GetSutProvider(config);
sutProvider.GetDependency<IGlobalSettings>().SelfHosted = false;
var exception = await Assert.ThrowsAsync<BadRequestException>(() =>
sutProvider.Sut.GetLicenseAsync(organization, billingSyncConnection));
Assert.Contains("only available for self-hosted", exception.Message);
}
[Theory]
[BitAutoData]
public async Task GetLicenseAsync_WhenCloudCommunicationDisabled_Throws(Organization organization,
OrganizationConnection billingSyncConnection, BillingSyncConfig config)
{
var sutProvider = GetSutProvider(config);
sutProvider.GetDependency<IGlobalSettings>().EnableCloudCommunication = false;
var exception = await Assert.ThrowsAsync<BadRequestException>(() =>
sutProvider.Sut.GetLicenseAsync(organization, billingSyncConnection));
Assert.Contains("Cloud communication is disabled", exception.Message);
}
[Theory]
[BitAutoData]
public async Task GetLicenseAsync_WhenCantUseConnection_Throws(Organization organization,
OrganizationConnection<BillingSyncConfig> billingSyncConnection, BillingSyncConfig config)
{
var sutProvider = GetSutProvider(config);
billingSyncConnection.Enabled = false;
var exception = await Assert.ThrowsAsync<BadRequestException>(() =>
sutProvider.Sut.GetLicenseAsync(organization, billingSyncConnection));
Assert.Contains("Connection disabled", exception.Message);
}
[Theory]
[BitAutoData]
public async Task GetLicenseAsync_WhenNullResponse_Throws(Organization organization,
OrganizationConnection<BillingSyncConfig> billingSyncConnection, BillingSyncConfig config)
{
var sutProvider = GetSutProvider(config);
billingSyncConnection.Enabled = true;
billingSyncConnection.Config = config;
var exception = await Assert.ThrowsAsync<BadRequestException>(() =>
sutProvider.Sut.GetLicenseAsync(organization, billingSyncConnection));
Assert.Contains("An error has occurred. Check your internet connection and ensure the billing token is correct.",
exception.Message);
}
}