mirror of
https://github.com/bitwarden/server.git
synced 2024-12-05 14:23:30 +01:00
a4ddb4b212
* move remaining Organization domain files * namespaces will be updated in a separate commit
100 lines
4.3 KiB
C#
100 lines
4.3 KiB
C#
using System.Text.Json;
|
|
using Bit.Core.AdminConsole.Entities;
|
|
using Bit.Core.Auth.Enums;
|
|
using Bit.Core.Auth.Models;
|
|
using Bit.Test.Common.Helpers;
|
|
using Xunit;
|
|
|
|
namespace Bit.Core.Test.Entities;
|
|
|
|
public class OrganizationTests
|
|
{
|
|
private static readonly Dictionary<TwoFactorProviderType, TwoFactorProvider> _testConfig = new Dictionary<TwoFactorProviderType, TwoFactorProvider>()
|
|
{
|
|
[TwoFactorProviderType.OrganizationDuo] = new TwoFactorProvider
|
|
{
|
|
Enabled = true,
|
|
MetaData = new Dictionary<string, object>
|
|
{
|
|
["IKey"] = "IKey_value",
|
|
["SKey"] = "SKey_value",
|
|
["Host"] = "Host_value",
|
|
},
|
|
}
|
|
};
|
|
|
|
|
|
[Fact]
|
|
public void SetTwoFactorProviders_Success()
|
|
{
|
|
var organization = new Organization();
|
|
organization.SetTwoFactorProviders(_testConfig);
|
|
|
|
using var jsonDocument = JsonDocument.Parse(organization.TwoFactorProviders);
|
|
var root = jsonDocument.RootElement;
|
|
|
|
var duo = AssertHelper.AssertJsonProperty(root, "6", JsonValueKind.Object);
|
|
AssertHelper.AssertJsonProperty(duo, "Enabled", JsonValueKind.True);
|
|
var duoMetaData = AssertHelper.AssertJsonProperty(duo, "MetaData", JsonValueKind.Object);
|
|
var iKey = AssertHelper.AssertJsonProperty(duoMetaData, "IKey", JsonValueKind.String).GetString();
|
|
Assert.Equal("IKey_value", iKey);
|
|
var sKey = AssertHelper.AssertJsonProperty(duoMetaData, "SKey", JsonValueKind.String).GetString();
|
|
Assert.Equal("SKey_value", sKey);
|
|
var host = AssertHelper.AssertJsonProperty(duoMetaData, "Host", JsonValueKind.String).GetString();
|
|
Assert.Equal("Host_value", host);
|
|
}
|
|
|
|
[Fact]
|
|
public void GetTwoFactorProviders_Success()
|
|
{
|
|
// This is to get rid of the cached dictionary the SetTwoFactorProviders keeps so we can fully test the JSON reading
|
|
// It intent is to mimic a storing of the entity in the database and it being read later
|
|
var tempOrganization = new Organization();
|
|
tempOrganization.SetTwoFactorProviders(_testConfig);
|
|
var organization = new Organization
|
|
{
|
|
TwoFactorProviders = tempOrganization.TwoFactorProviders,
|
|
};
|
|
|
|
var twoFactorProviders = organization.GetTwoFactorProviders();
|
|
|
|
var duo = Assert.Contains(TwoFactorProviderType.OrganizationDuo, (IDictionary<TwoFactorProviderType, TwoFactorProvider>)twoFactorProviders);
|
|
Assert.True(duo.Enabled);
|
|
Assert.NotNull(duo.MetaData);
|
|
var iKey = Assert.Contains("IKey", (IDictionary<string, object>)duo.MetaData);
|
|
Assert.Equal("IKey_value", iKey);
|
|
var sKey = Assert.Contains("SKey", (IDictionary<string, object>)duo.MetaData);
|
|
Assert.Equal("SKey_value", sKey);
|
|
var host = Assert.Contains("Host", (IDictionary<string, object>)duo.MetaData);
|
|
Assert.Equal("Host_value", host);
|
|
}
|
|
|
|
[Fact]
|
|
public void GetTwoFactorProviders_SavedWithName_Success()
|
|
{
|
|
var organization = new Organization();
|
|
// This should save items with the string name of the enum and we will validate that we can read
|
|
// from that just incase some organizations have it saved that way.
|
|
organization.TwoFactorProviders = JsonSerializer.Serialize(_testConfig);
|
|
|
|
// Preliminary Asserts to make sure we are testing what we want to be testing
|
|
using var jsonDocument = JsonDocument.Parse(organization.TwoFactorProviders);
|
|
var root = jsonDocument.RootElement;
|
|
// This means it saved the enum as its string name
|
|
AssertHelper.AssertJsonProperty(root, "OrganizationDuo", JsonValueKind.Object);
|
|
|
|
// Actual checks
|
|
var twoFactorProviders = organization.GetTwoFactorProviders();
|
|
|
|
var duo = Assert.Contains(TwoFactorProviderType.OrganizationDuo, (IDictionary<TwoFactorProviderType, TwoFactorProvider>)twoFactorProviders);
|
|
Assert.True(duo.Enabled);
|
|
Assert.NotNull(duo.MetaData);
|
|
var iKey = Assert.Contains("IKey", (IDictionary<string, object>)duo.MetaData);
|
|
Assert.Equal("IKey_value", iKey);
|
|
var sKey = Assert.Contains("SKey", (IDictionary<string, object>)duo.MetaData);
|
|
Assert.Equal("SKey_value", sKey);
|
|
var host = Assert.Contains("Host", (IDictionary<string, object>)duo.MetaData);
|
|
Assert.Equal("Host_value", host);
|
|
}
|
|
}
|