1
0
mirror of https://github.com/bitwarden/server.git synced 2025-04-06 21:48:12 -05:00

Added new TemplateProcessor and added/updated unit tests

This commit is contained in:
Brant DeBow 2025-03-06 20:29:18 -05:00
parent 862f1821c8
commit d33edbdd85
No known key found for this signature in database
GPG Key ID: 94411BB25947C72B
6 changed files with 160 additions and 17 deletions

View File

@ -46,7 +46,7 @@ public class OrganizationIntegrationConfigurationRepository(GlobalSettings globa
{ {
Url = _webhookUrl, Url = _webhookUrl,
}, },
Template = "{ \"newObject\": true }" Template = "{ \"Date\": \"#Date#\", \"Type\": \"#Type#\", \"UserId\": \"#UserId#\" }"
} as IntegrationConfiguration<T>; } as IntegrationConfiguration<T>;
default: default:
return null; return null;

View File

@ -1,4 +1,4 @@
using System.Net.Http.Json; using System.Text;
using Bit.Core.Enums; using Bit.Core.Enums;
using Bit.Core.Models.Data; using Bit.Core.Models.Data;
using Bit.Core.Models.Data.Integrations; using Bit.Core.Models.Data.Integrations;
@ -27,7 +27,11 @@ public class WebhookEventHandler(
eventMessage.Type); eventMessage.Type);
if (configuration is not null) if (configuration is not null)
{ {
var content = JsonContent.Create(configuration.Template); var content = new StringContent(
TemplateProcessor.ReplaceTokens(configuration.Template, eventMessage),
Encoding.UTF8,
"application/json"
);
var response = await _httpClient.PostAsync( var response = await _httpClient.PostAsync(
configuration.Configuration.Url, configuration.Configuration.Url,
content); content);

View File

@ -0,0 +1,20 @@
using System.Text.RegularExpressions;
public static class TemplateProcessor
{
private static readonly Regex TokenRegex = new(@"#(\w+)#", RegexOptions.Compiled);
public static string ReplaceTokens(string template, object values)
{
if (string.IsNullOrEmpty(template) || values == null)
return template;
var type = values.GetType();
return TokenRegex.Replace(template, match =>
{
var propertyName = match.Groups[1].Value;
var property = type.GetProperty(propertyName);
return property?.GetValue(values)?.ToString() ?? match.Value;
});
}
}

View File

@ -74,7 +74,6 @@
</ItemGroup> </ItemGroup>
<ItemGroup> <ItemGroup>
<Folder Include="AdminConsole\Utilities\" />
<Folder Include="Resources\" /> <Folder Include="Resources\" />
<Folder Include="Properties\" /> <Folder Include="Properties\" />
</ItemGroup> </ItemGroup>

View File

@ -1,6 +1,9 @@
using System.Net; using System.Net;
using System.Net.Http.Json; using System.Net.Http.Json;
using Bit.Core.Enums;
using Bit.Core.Models.Data; using Bit.Core.Models.Data;
using Bit.Core.Models.Data.Integrations;
using Bit.Core.Repositories;
using Bit.Core.Services; using Bit.Core.Services;
using Bit.Test.Common.AutoFixture; using Bit.Test.Common.AutoFixture;
using Bit.Test.Common.AutoFixture.Attributes; using Bit.Test.Common.AutoFixture.Attributes;
@ -8,7 +11,6 @@ using Bit.Test.Common.Helpers;
using Bit.Test.Common.MockedHttpClient; using Bit.Test.Common.MockedHttpClient;
using NSubstitute; using NSubstitute;
using Xunit; using Xunit;
using GlobalSettings = Bit.Core.Settings.GlobalSettings;
namespace Bit.Core.Test.Services; namespace Bit.Core.Test.Services;
@ -16,7 +18,7 @@ namespace Bit.Core.Test.Services;
public class WebhookEventHandlerTests public class WebhookEventHandlerTests
{ {
private readonly MockedHttpMessageHandler _handler; private readonly MockedHttpMessageHandler _handler;
private HttpClient _httpClient; private readonly HttpClient _httpClient;
private const string _webhookUrl = "http://localhost/test/event"; private const string _webhookUrl = "http://localhost/test/event";
@ -29,16 +31,26 @@ public class WebhookEventHandlerTests
_httpClient = _handler.ToHttpClient(); _httpClient = _handler.ToHttpClient();
} }
public SutProvider<WebhookEventHandler> GetSutProvider() private SutProvider<WebhookEventHandler> GetSutProvider()
{ {
var clientFactory = Substitute.For<IHttpClientFactory>(); var clientFactory = Substitute.For<IHttpClientFactory>();
clientFactory.CreateClient(WebhookEventHandler.HttpClientName).Returns(_httpClient); clientFactory.CreateClient(WebhookEventHandler.HttpClientName).Returns(_httpClient);
var globalSettings = new GlobalSettings(); var repository = Substitute.For<IOrganizationIntegrationConfigurationRepository>();
globalSettings.EventLogging.WebhookUrl = _webhookUrl; repository.GetConfigurationAsync<WebhookConfiguration>(
Arg.Any<Guid>(),
IntegrationType.Webhook,
Arg.Any<EventType>()
).Returns(
new IntegrationConfiguration<WebhookConfiguration>
{
Configuration = new WebhookConfiguration { ApiKey = "", Url = _webhookUrl },
Template = "{ \"Date\": \"#Date#\", \"Type\": \"#Type#\", \"UserId\": \"#UserId#\" }"
}
);
return new SutProvider<WebhookEventHandler>() return new SutProvider<WebhookEventHandler>()
.SetDependency(globalSettings) .SetDependency(repository)
.SetDependency(clientFactory) .SetDependency(clientFactory)
.Create(); .Create();
} }
@ -50,21 +62,22 @@ public class WebhookEventHandlerTests
await sutProvider.Sut.HandleEventAsync(eventMessage); await sutProvider.Sut.HandleEventAsync(eventMessage);
sutProvider.GetDependency<IHttpClientFactory>().Received(1).CreateClient( sutProvider.GetDependency<IHttpClientFactory>().Received(1).CreateClient(
Arg.Is(AssertHelper.AssertPropertyEqual<string>(WebhookEventHandler.HttpClientName)) Arg.Is(AssertHelper.AssertPropertyEqual(WebhookEventHandler.HttpClientName))
); );
Assert.Single(_handler.CapturedRequests); Assert.Single(_handler.CapturedRequests);
var request = _handler.CapturedRequests[0]; var request = _handler.CapturedRequests[0];
Assert.NotNull(request); Assert.NotNull(request);
var returned = await request.Content.ReadFromJsonAsync<EventMessage>(); var returned = await request.Content.ReadFromJsonAsync<MockEvent>();
var expected = MockEvent.From(eventMessage);
Assert.Equal(HttpMethod.Post, request.Method); Assert.Equal(HttpMethod.Post, request.Method);
Assert.Equal(_webhookUrl, request.RequestUri.ToString()); Assert.Equal(_webhookUrl, request.RequestUri.ToString());
AssertHelper.AssertPropertyEqual(eventMessage, returned, new[] { "IdempotencyId" }); AssertHelper.AssertPropertyEqual(expected, returned);
} }
[Theory, BitAutoData] [Theory, BitAutoData]
public async Task HandleEventManyAsync_PostsEventsToUrl(IEnumerable<EventMessage> eventMessages) public async Task HandleEventManyAsync_PostsEventsToUrl(List<EventMessage> eventMessages)
{ {
var sutProvider = GetSutProvider(); var sutProvider = GetSutProvider();
@ -73,13 +86,29 @@ public class WebhookEventHandlerTests
Arg.Is(AssertHelper.AssertPropertyEqual<string>(WebhookEventHandler.HttpClientName)) Arg.Is(AssertHelper.AssertPropertyEqual<string>(WebhookEventHandler.HttpClientName))
); );
Assert.Single(_handler.CapturedRequests);
var request = _handler.CapturedRequests[0]; var request = _handler.CapturedRequests[0];
Assert.NotNull(request); Assert.NotNull(request);
var returned = request.Content.ReadFromJsonAsAsyncEnumerable<EventMessage>(); var returned = await request.Content.ReadFromJsonAsync<MockEvent>();
var expected = MockEvent.From(eventMessages.First());
Assert.Equal(HttpMethod.Post, request.Method); Assert.Equal(HttpMethod.Post, request.Method);
Assert.Equal(_webhookUrl, request.RequestUri.ToString()); Assert.Equal(_webhookUrl, request.RequestUri.ToString());
AssertHelper.AssertPropertyEqual(eventMessages, returned, new[] { "IdempotencyId" }); AssertHelper.AssertPropertyEqual(expected, returned);
}
}
public class MockEvent(string date, string type, string userId)
{
public string Date { get; set; } = date;
public string Type { get; set; } = type;
public string UserId { get; set; } = userId;
public static MockEvent From(EventMessage eventMessage)
{
return new MockEvent(
eventMessage.Date.ToString(),
eventMessage.Type.ToString(),
eventMessage.UserId.ToString()
);
} }
} }

View File

@ -0,0 +1,91 @@
using Bit.Core.Models.Data;
using Bit.Test.Common.AutoFixture.Attributes;
using Xunit;
namespace Bit.Core.Test.Utilities;
public class TemplateProcessorTests
{
[Theory, BitAutoData]
public void ReplaceTokens_ReplacesSingleToken(EventMessage eventMessage)
{
var template = "Event #Type# occurred.";
var expected = $"Event {eventMessage.Type} occurred.";
var result = TemplateProcessor.ReplaceTokens(template, eventMessage);
Assert.Equal(expected, result);
}
[Theory, BitAutoData]
public void ReplaceTokens_ReplacesMultipleTokens(EventMessage eventMessage)
{
var template = "Event #Type#, User (id: #UserId#).";
var expected = $"Event {eventMessage.Type}, User (id: {eventMessage.UserId}).";
var result = TemplateProcessor.ReplaceTokens(template, eventMessage);
Assert.Equal(expected, result);
}
[Theory, BitAutoData]
public void ReplaceTokens_LeavesUnknownTokensUnchanged(EventMessage eventMessage)
{
var template = "Event #Type#, User (id: #UserId#), Details: #UnknownKey#.";
var expected = $"Event {eventMessage.Type}, User (id: {eventMessage.UserId}), Details: #UnknownKey#.";
var result = TemplateProcessor.ReplaceTokens(template, eventMessage);
Assert.Equal(expected, result);
}
[Theory, BitAutoData]
public void ReplaceTokens_WithNullProperty_LeavesTokenUnchanged(EventMessage eventMessage)
{
eventMessage.UserId = null; // Ensure UserId is null for this test
var template = "Event #Type#, User (id: #UserId#).";
var expected = $"Event {eventMessage.Type}, User (id: #UserId#).";
var result = TemplateProcessor.ReplaceTokens(template, eventMessage);
Assert.Equal(expected, result);
}
[Theory, BitAutoData]
public void ReplaceTokens_IgnoresCaseSensitiveTokens(EventMessage eventMessage)
{
var template = "Event #type#, User (id: #UserId#)."; // Lowercase "type"
var expected = $"Event #type#, User (id: {eventMessage.UserId})."; // Token remains unchanged
var result = TemplateProcessor.ReplaceTokens(template, eventMessage);
Assert.Equal(expected, result);
}
[Theory, BitAutoData]
public void ReplaceTokens_ReturnsOriginalString_IfNoTokensPresent(EventMessage eventMessage)
{
var template = "System is running normally.";
var expected = "System is running normally.";
var result = TemplateProcessor.ReplaceTokens(template, eventMessage);
Assert.Equal(expected, result);
}
[Theory, BitAutoData]
public void ReplaceTokens_ReturnsOriginalString_IfTemplateIsNullOrEmpty(EventMessage eventMessage)
{
var emptyTemplate = "";
var expectedEmpty = "";
Assert.Equal(expectedEmpty, TemplateProcessor.ReplaceTokens(emptyTemplate, eventMessage));
Assert.Null(TemplateProcessor.ReplaceTokens(null, eventMessage));
}
[Fact]
public void ReplaceTokens_ReturnsOriginalString_IfDataObjectIsNull()
{
var template = "Event #Type#, User (id: #UserId#).";
var expected = "Event #Type#, User (id: #UserId#).";
var result = TemplateProcessor.ReplaceTokens(template, null);
Assert.Equal(expected, result);
}
}