2021-01-06 19:49:28 +01:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using Bit.Core.Utilities;
|
|
|
|
|
using Xunit;
|
2021-10-05 18:12:05 +02:00
|
|
|
|
using Bit.Core.Test.AutoFixture.UserFixtures;
|
|
|
|
|
using IdentityModel;
|
|
|
|
|
using Bit.Core.Enums.Provider;
|
|
|
|
|
using Bit.Core.Models.Table;
|
|
|
|
|
using Bit.Core.Context;
|
|
|
|
|
using AutoFixture;
|
|
|
|
|
using Bit.Core.Enums;
|
2021-10-29 21:05:45 +02:00
|
|
|
|
using Bit.Test.Common.AutoFixture.Attributes;
|
|
|
|
|
using Bit.Test.Common.AutoFixture;
|
2021-01-06 19:49:28 +01:00
|
|
|
|
|
|
|
|
|
namespace Bit.Core.Test.Utilities
|
|
|
|
|
{
|
|
|
|
|
public class CoreHelpersTests
|
|
|
|
|
{
|
|
|
|
|
public static IEnumerable<object[]> _epochTestCases = new[]
|
|
|
|
|
{
|
|
|
|
|
new object[] {new DateTime(2020, 12, 30, 11, 49, 12, DateTimeKind.Utc), 1609328952000L},
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
[Fact]
|
|
|
|
|
public void GenerateComb_Success()
|
|
|
|
|
{
|
|
|
|
|
// Arrange & Act
|
|
|
|
|
var comb = CoreHelpers.GenerateComb();
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.NotEqual(Guid.Empty, comb);
|
|
|
|
|
// TODO: Add more asserts to make sure important aspects of
|
|
|
|
|
// the comb are working properly
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Theory]
|
|
|
|
|
[InlineData(2, 5, new[] { 1, 2, 3, 4, 5, 6, 7, 8 , 9, 0 })]
|
|
|
|
|
[InlineData(2, 3, new[] { 1, 2, 3, 4, 5 })]
|
|
|
|
|
[InlineData(2, 1, new[] { 1, 2 })]
|
|
|
|
|
[InlineData(1, 1, new[] { 1 })]
|
|
|
|
|
[InlineData(2, 2, new[] { 1, 2, 3 })]
|
|
|
|
|
public void Batch_Success(int batchSize, int totalBatches, int[] collection)
|
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
var remainder = collection.Length % batchSize;
|
|
|
|
|
|
|
|
|
|
// Act
|
|
|
|
|
var batches = collection.Batch(batchSize);
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.Equal(totalBatches, batches.Count());
|
|
|
|
|
|
|
|
|
|
foreach (var batch in batches.Take(totalBatches - 1))
|
|
|
|
|
{
|
|
|
|
|
Assert.Equal(batchSize, batch.Count());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Assert.Equal(batches.Last().Count(), remainder == 0 ? batchSize : remainder);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Fact]
|
|
|
|
|
public void ToGuidIdArrayTVP_Success()
|
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
var item0 = Guid.NewGuid();
|
|
|
|
|
var item1 = Guid.NewGuid();
|
|
|
|
|
|
|
|
|
|
var ids = new[] { item0, item1 };
|
|
|
|
|
|
|
|
|
|
// Act
|
|
|
|
|
var dt = ids.ToGuidIdArrayTVP();
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.Single(dt.Columns);
|
|
|
|
|
Assert.Equal("GuidId", dt.Columns[0].ColumnName);
|
|
|
|
|
Assert.Equal(2, dt.Rows.Count);
|
|
|
|
|
Assert.Equal(item0, dt.Rows[0][0]);
|
|
|
|
|
Assert.Equal(item1, dt.Rows[1][0]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// TODO: Test the other ToArrayTVP Methods
|
|
|
|
|
|
|
|
|
|
[Theory]
|
|
|
|
|
[InlineData("12345&6789", "123456789")]
|
|
|
|
|
[InlineData("abcdef", "ABCDEF")]
|
|
|
|
|
[InlineData("1!@#$%&*()_+", "1")]
|
|
|
|
|
[InlineData("\u00C6123abc\u00C7", "123ABC")]
|
|
|
|
|
[InlineData("123\u00C6ABC", "123ABC")]
|
|
|
|
|
[InlineData("\r\nHello", "E")]
|
|
|
|
|
[InlineData("\tdef", "DEF")]
|
|
|
|
|
[InlineData("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUV1234567890", "ABCDEFABCDEF1234567890")]
|
|
|
|
|
public void CleanCertificateThumbprint_Success(string input, string output)
|
|
|
|
|
{
|
|
|
|
|
// Arrange & Act
|
|
|
|
|
var sanitizedInput = CoreHelpers.CleanCertificateThumbprint(input);
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.Equal(output, sanitizedInput);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// TODO: Add more tests
|
|
|
|
|
[Theory]
|
|
|
|
|
[MemberData(nameof(_epochTestCases))]
|
|
|
|
|
public void ToEpocMilliseconds_Success(DateTime date, long milliseconds)
|
|
|
|
|
{
|
|
|
|
|
// Act & Assert
|
|
|
|
|
Assert.Equal(milliseconds, CoreHelpers.ToEpocMilliseconds(date));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Theory]
|
|
|
|
|
[MemberData(nameof(_epochTestCases))]
|
|
|
|
|
public void FromEpocMilliseconds(DateTime date, long milliseconds)
|
|
|
|
|
{
|
|
|
|
|
// Act & Assert
|
|
|
|
|
Assert.Equal(date, CoreHelpers.FromEpocMilliseconds(milliseconds));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Fact]
|
|
|
|
|
public void SecureRandomString_Success()
|
|
|
|
|
{
|
|
|
|
|
// Arrange & Act
|
|
|
|
|
var @string = CoreHelpers.SecureRandomString(8);
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
// TODO: Should probably add more Asserts down the line
|
|
|
|
|
Assert.Equal(8, @string.Length);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Theory]
|
|
|
|
|
[InlineData(1, "1 Bytes")]
|
|
|
|
|
[InlineData(-5L, "-5 Bytes")]
|
|
|
|
|
[InlineData(1023L, "1023 Bytes")]
|
|
|
|
|
[InlineData(1024L, "1 KB")]
|
|
|
|
|
[InlineData(1025L, "1 KB")]
|
|
|
|
|
[InlineData(-1023L, "-1023 Bytes")]
|
|
|
|
|
[InlineData(-1024L, "-1 KB")]
|
|
|
|
|
[InlineData(-1025L, "-1 KB")]
|
|
|
|
|
[InlineData(1048575L, "1024 KB")]
|
|
|
|
|
[InlineData(1048576L, "1 MB")]
|
|
|
|
|
[InlineData(1048577L, "1 MB")]
|
|
|
|
|
[InlineData(-1048575L, "-1024 KB")]
|
|
|
|
|
[InlineData(-1048576L, "-1 MB")]
|
|
|
|
|
[InlineData(-1048577L, "-1 MB")]
|
|
|
|
|
[InlineData(1073741823L, "1024 MB")]
|
|
|
|
|
[InlineData(1073741824L, "1 GB")]
|
|
|
|
|
[InlineData(1073741825L, "1 GB")]
|
|
|
|
|
[InlineData(-1073741823L, "-1024 MB")]
|
|
|
|
|
[InlineData(-1073741824L, "-1 GB")]
|
|
|
|
|
[InlineData(-1073741825L, "-1 GB")]
|
|
|
|
|
[InlineData(long.MaxValue, "8589934592 GB")]
|
|
|
|
|
public void ReadableBytesSize_Success(long size, string readable)
|
|
|
|
|
{
|
|
|
|
|
// Act & Assert
|
|
|
|
|
Assert.Equal(readable, CoreHelpers.ReadableBytesSize(size));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Fact]
|
|
|
|
|
public void CloneObject_Success()
|
|
|
|
|
{
|
2021-10-05 18:12:05 +02:00
|
|
|
|
var original = new { Message = "Message" };
|
2021-01-06 19:49:28 +01:00
|
|
|
|
|
2021-10-05 18:12:05 +02:00
|
|
|
|
var copy = CoreHelpers.CloneObject(original);
|
2021-01-06 19:49:28 +01:00
|
|
|
|
|
2021-10-05 18:12:05 +02:00
|
|
|
|
Assert.Equal(original.Message, copy.Message);
|
2021-01-06 19:49:28 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Fact]
|
|
|
|
|
public void ExtendQuery_AddNewParameter_Success()
|
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
var uri = new Uri("https://bitwarden.com/?param1=value1");
|
|
|
|
|
|
|
|
|
|
// Act
|
|
|
|
|
var newUri = CoreHelpers.ExtendQuery(uri,
|
|
|
|
|
new Dictionary<string, string> { { "param2", "value2" } });
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.Equal("https://bitwarden.com/?param1=value1¶m2=value2", newUri.ToString());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Fact]
|
|
|
|
|
public void ExtendQuery_AddTwoNewParameters_Success()
|
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
var uri = new Uri("https://bitwarden.com/?param1=value1");
|
|
|
|
|
|
|
|
|
|
// Act
|
|
|
|
|
var newUri = CoreHelpers.ExtendQuery(uri,
|
|
|
|
|
new Dictionary<string, string>
|
|
|
|
|
{
|
|
|
|
|
{ "param2", "value2" },
|
|
|
|
|
{ "param3", "value3" }
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.Equal("https://bitwarden.com/?param1=value1¶m2=value2¶m3=value3", newUri.ToString());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Fact]
|
|
|
|
|
public void ExtendQuery_AddExistingParameter_Success()
|
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
var uri = new Uri("https://bitwarden.com/?param1=value1¶m2=value2");
|
|
|
|
|
|
|
|
|
|
// Act
|
|
|
|
|
var newUri = CoreHelpers.ExtendQuery(uri,
|
|
|
|
|
new Dictionary<string, string> { { "param1", "test_value" } });
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.Equal("https://bitwarden.com/?param1=test_value¶m2=value2", newUri.ToString());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Fact]
|
|
|
|
|
public void ExtendQuery_AddNoParameters_Success()
|
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
const string startingUri = "https://bitwarden.com/?param1=value1";
|
|
|
|
|
|
|
|
|
|
var uri = new Uri(startingUri);
|
|
|
|
|
|
|
|
|
|
// Act
|
|
|
|
|
var newUri = CoreHelpers.ExtendQuery(uri, new Dictionary<string, string>());
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.Equal(startingUri, newUri.ToString());
|
|
|
|
|
}
|
2021-08-31 05:49:11 +02:00
|
|
|
|
|
|
|
|
|
[Theory]
|
|
|
|
|
[InlineData("bücher.com", "xn--bcher-kva.com")]
|
|
|
|
|
[InlineData("bücher.cömé", "xn--bcher-kva.xn--cm-cja4c")]
|
|
|
|
|
[InlineData("hello@bücher.com", "hello@xn--bcher-kva.com")]
|
|
|
|
|
[InlineData("hello@world.cömé", "hello@world.xn--cm-cja4c")]
|
|
|
|
|
[InlineData("hello@bücher.cömé", "hello@xn--bcher-kva.xn--cm-cja4c")]
|
|
|
|
|
[InlineData("ascii.com", "ascii.com")]
|
|
|
|
|
[InlineData("", "")]
|
|
|
|
|
[InlineData(null, null)]
|
|
|
|
|
public void PunyEncode_Success(string text, string expected)
|
|
|
|
|
{
|
|
|
|
|
var actual = CoreHelpers.PunyEncode(text);
|
|
|
|
|
Assert.Equal(expected, actual);
|
|
|
|
|
}
|
2021-10-05 18:12:05 +02:00
|
|
|
|
|
|
|
|
|
[Fact]
|
|
|
|
|
public void GetEmbeddedResourceContentsAsync_Success()
|
|
|
|
|
{
|
|
|
|
|
var fileContents = CoreHelpers.GetEmbeddedResourceContentsAsync("data.embeddedResource.txt");
|
|
|
|
|
Assert.Equal("Contents of embeddedResource.txt\n", fileContents.Replace("\r\n", "\n"));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Theory, CustomAutoData(typeof(UserFixture))]
|
|
|
|
|
public void BuildIdentityClaims_BaseClaims_Success(User user, bool isPremium)
|
|
|
|
|
{
|
|
|
|
|
var expected = new Dictionary<string, string>
|
|
|
|
|
{
|
|
|
|
|
{ "premium", isPremium ? "true" : "false" },
|
|
|
|
|
{ JwtClaimTypes.Email, user.Email },
|
|
|
|
|
{ JwtClaimTypes.EmailVerified, user.EmailVerified ? "true" : "false" },
|
|
|
|
|
{ JwtClaimTypes.Name, user.Name },
|
|
|
|
|
{ "sstamp", user.SecurityStamp },
|
|
|
|
|
}.ToList();
|
|
|
|
|
|
|
|
|
|
var actual = CoreHelpers.BuildIdentityClaims(user, Array.Empty<CurrentContentOrganization>(),
|
|
|
|
|
Array.Empty<CurrentContentProvider>(), isPremium);
|
|
|
|
|
|
|
|
|
|
foreach (var claim in expected)
|
|
|
|
|
{
|
|
|
|
|
Assert.Contains(claim, actual);
|
|
|
|
|
}
|
|
|
|
|
Assert.Equal(expected.Count, actual.Count);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Theory, CustomAutoData(typeof(UserFixture))]
|
|
|
|
|
public void BuildIdentityClaims_NonCustomOrganizationUserType_Success(User user)
|
|
|
|
|
{
|
|
|
|
|
var fixture = new Fixture().WithAutoNSubstitutions();
|
|
|
|
|
foreach (var organizationUserType in Enum.GetValues<OrganizationUserType>().Except(new[] { OrganizationUserType.Custom }))
|
|
|
|
|
{
|
|
|
|
|
var org = fixture.Create<CurrentContentOrganization>();
|
|
|
|
|
org.Type = organizationUserType;
|
|
|
|
|
|
|
|
|
|
var expected = new KeyValuePair<string, string>($"org{organizationUserType.ToString().ToLower()}", org.Id.ToString());
|
|
|
|
|
var actual = CoreHelpers.BuildIdentityClaims(user, new[] { org }, Array.Empty<CurrentContentProvider>(), false);
|
|
|
|
|
|
|
|
|
|
Assert.Contains(expected, actual);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Theory, CustomAutoData(typeof(UserFixture))]
|
|
|
|
|
public void BuildIdentityClaims_CustomOrganizationUserClaims_Success(User user, CurrentContentOrganization org)
|
|
|
|
|
{
|
|
|
|
|
var fixture = new Fixture().WithAutoNSubstitutions();
|
|
|
|
|
org.Type = OrganizationUserType.Custom;
|
|
|
|
|
|
|
|
|
|
var actual = CoreHelpers.BuildIdentityClaims(user, new[] { org }, Array.Empty<CurrentContentProvider>(), false);
|
|
|
|
|
foreach (var (permitted, claimName) in org.Permissions.ClaimsMap)
|
|
|
|
|
{
|
|
|
|
|
var claim = new KeyValuePair<string, string>(claimName, org.Id.ToString());
|
|
|
|
|
if (permitted)
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
Assert.Contains(claim, actual);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
Assert.DoesNotContain(claim, actual);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Theory, CustomAutoData(typeof(UserFixture))]
|
|
|
|
|
public void BuildIdentityClaims_ProviderClaims_Success(User user)
|
|
|
|
|
{
|
|
|
|
|
var fixture = new Fixture().WithAutoNSubstitutions();
|
|
|
|
|
var providers = new List<CurrentContentProvider>();
|
|
|
|
|
foreach (var providerUserType in Enum.GetValues<ProviderUserType>())
|
|
|
|
|
{
|
|
|
|
|
var provider = fixture.Create<CurrentContentProvider>();
|
|
|
|
|
provider.Type = providerUserType;
|
|
|
|
|
providers.Add(provider);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var claims = new List<KeyValuePair<string, string>>();
|
|
|
|
|
|
|
|
|
|
if (providers.Any())
|
|
|
|
|
{
|
|
|
|
|
foreach (var group in providers.GroupBy(o => o.Type))
|
|
|
|
|
{
|
|
|
|
|
switch (group.Key)
|
|
|
|
|
{
|
|
|
|
|
case ProviderUserType.ProviderAdmin:
|
|
|
|
|
foreach (var provider in group)
|
|
|
|
|
{
|
|
|
|
|
claims.Add(new KeyValuePair<string, string>("providerprovideradmin", provider.Id.ToString()));
|
|
|
|
|
}
|
|
|
|
|
break;
|
|
|
|
|
case ProviderUserType.ServiceUser:
|
|
|
|
|
foreach (var provider in group)
|
|
|
|
|
{
|
|
|
|
|
claims.Add(new KeyValuePair<string, string>("providerserviceuser", provider.Id.ToString()));
|
|
|
|
|
}
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var actual = CoreHelpers.BuildIdentityClaims(user, Array.Empty<CurrentContentOrganization>(), providers, false);
|
|
|
|
|
foreach (var claim in claims)
|
|
|
|
|
{
|
|
|
|
|
Assert.Contains(claim, actual);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2021-01-06 19:49:28 +01:00
|
|
|
|
}
|
|
|
|
|
}
|