2025-02-04 08:02:43 -06:00
|
|
|
|
using System.Net;
|
|
|
|
|
|
using System.Net.Http.Json;
|
2025-04-23 10:44:43 -04:00
|
|
|
|
using System.Text.Json;
|
|
|
|
|
|
using Bit.Core.Enums;
|
2025-02-04 08:02:43 -06:00
|
|
|
|
using Bit.Core.Models.Data;
|
2025-04-23 10:44:43 -04:00
|
|
|
|
using Bit.Core.Models.Data.Organizations;
|
|
|
|
|
|
using Bit.Core.Repositories;
|
2025-02-04 08:02:43 -06:00
|
|
|
|
using Bit.Core.Services;
|
|
|
|
|
|
using Bit.Test.Common.AutoFixture;
|
|
|
|
|
|
using Bit.Test.Common.AutoFixture.Attributes;
|
|
|
|
|
|
using Bit.Test.Common.Helpers;
|
|
|
|
|
|
using Bit.Test.Common.MockedHttpClient;
|
|
|
|
|
|
using NSubstitute;
|
|
|
|
|
|
using Xunit;
|
|
|
|
|
|
|
|
|
|
|
|
namespace Bit.Core.Test.Services;
|
|
|
|
|
|
|
|
|
|
|
|
[SutProviderCustomize]
|
2025-02-11 10:20:06 -05:00
|
|
|
|
public class WebhookEventHandlerTests
|
2025-02-04 08:02:43 -06:00
|
|
|
|
{
|
|
|
|
|
|
private readonly MockedHttpMessageHandler _handler;
|
2025-04-23 10:44:43 -04:00
|
|
|
|
private readonly HttpClient _httpClient;
|
2025-02-04 08:02:43 -06:00
|
|
|
|
|
2025-04-23 10:44:43 -04:00
|
|
|
|
private const string _template =
|
|
|
|
|
|
"""
|
|
|
|
|
|
{
|
|
|
|
|
|
"Date": "#Date#",
|
|
|
|
|
|
"Type": "#Type#",
|
|
|
|
|
|
"UserId": "#UserId#"
|
|
|
|
|
|
}
|
|
|
|
|
|
""";
|
2025-02-11 10:20:06 -05:00
|
|
|
|
private const string _webhookUrl = "http://localhost/test/event";
|
2025-04-23 10:44:43 -04:00
|
|
|
|
private const string _webhookUrl2 = "http://localhost/another/event";
|
2025-02-04 08:02:43 -06:00
|
|
|
|
|
2025-02-11 10:20:06 -05:00
|
|
|
|
public WebhookEventHandlerTests()
|
2025-02-04 08:02:43 -06:00
|
|
|
|
{
|
|
|
|
|
|
_handler = new MockedHttpMessageHandler();
|
|
|
|
|
|
_handler.Fallback
|
|
|
|
|
|
.WithStatusCode(HttpStatusCode.OK)
|
|
|
|
|
|
.WithContent(new StringContent("<html><head><title>test</title></head><body>test</body></html>"));
|
|
|
|
|
|
_httpClient = _handler.ToHttpClient();
|
|
|
|
|
|
}
|
|
|
|
|
|
|
2025-04-23 10:44:43 -04:00
|
|
|
|
private SutProvider<WebhookEventHandler> GetSutProvider(
|
|
|
|
|
|
List<OrganizationIntegrationConfigurationDetails> configurations)
|
2025-02-04 08:02:43 -06:00
|
|
|
|
{
|
|
|
|
|
|
var clientFactory = Substitute.For<IHttpClientFactory>();
|
2025-02-11 10:20:06 -05:00
|
|
|
|
clientFactory.CreateClient(WebhookEventHandler.HttpClientName).Returns(_httpClient);
|
2025-02-04 08:02:43 -06:00
|
|
|
|
|
2025-04-23 10:44:43 -04:00
|
|
|
|
var repository = Substitute.For<IOrganizationIntegrationConfigurationRepository>();
|
|
|
|
|
|
repository.GetConfigurationDetailsAsync(Arg.Any<Guid>(),
|
|
|
|
|
|
IntegrationType.Webhook, Arg.Any<EventType>()).Returns(configurations);
|
2025-02-04 08:02:43 -06:00
|
|
|
|
|
2025-02-11 10:20:06 -05:00
|
|
|
|
return new SutProvider<WebhookEventHandler>()
|
2025-04-23 10:44:43 -04:00
|
|
|
|
.SetDependency(repository)
|
2025-02-04 08:02:43 -06:00
|
|
|
|
.SetDependency(clientFactory)
|
|
|
|
|
|
.Create();
|
|
|
|
|
|
}
|
|
|
|
|
|
|
2025-04-23 10:44:43 -04:00
|
|
|
|
private static List<OrganizationIntegrationConfigurationDetails> NoConfigurations()
|
|
|
|
|
|
{
|
|
|
|
|
|
return [];
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
private static List<OrganizationIntegrationConfigurationDetails> OneConfiguration()
|
|
|
|
|
|
{
|
|
|
|
|
|
var config = Substitute.For<OrganizationIntegrationConfigurationDetails>();
|
|
|
|
|
|
config.Configuration = null;
|
|
|
|
|
|
config.IntegrationConfiguration = JsonSerializer.Serialize(new { url = _webhookUrl });
|
|
|
|
|
|
config.Template = _template;
|
|
|
|
|
|
|
|
|
|
|
|
return [config];
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
private static List<OrganizationIntegrationConfigurationDetails> TwoConfigurations()
|
|
|
|
|
|
{
|
|
|
|
|
|
var config = Substitute.For<OrganizationIntegrationConfigurationDetails>();
|
|
|
|
|
|
config.Configuration = null;
|
|
|
|
|
|
config.IntegrationConfiguration = JsonSerializer.Serialize(new { url = _webhookUrl });
|
|
|
|
|
|
config.Template = _template;
|
|
|
|
|
|
var config2 = Substitute.For<OrganizationIntegrationConfigurationDetails>();
|
|
|
|
|
|
config2.Configuration = null;
|
|
|
|
|
|
config2.IntegrationConfiguration = JsonSerializer.Serialize(new { url = _webhookUrl2 });
|
|
|
|
|
|
config2.Template = _template;
|
|
|
|
|
|
|
|
|
|
|
|
return [config, config2];
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
private static List<OrganizationIntegrationConfigurationDetails> WrongConfiguration()
|
|
|
|
|
|
{
|
|
|
|
|
|
var config = Substitute.For<OrganizationIntegrationConfigurationDetails>();
|
|
|
|
|
|
config.Configuration = null;
|
|
|
|
|
|
config.IntegrationConfiguration = JsonSerializer.Serialize(new { error = string.Empty });
|
|
|
|
|
|
config.Template = _template;
|
|
|
|
|
|
|
|
|
|
|
|
return [config];
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[Theory, BitAutoData]
|
|
|
|
|
|
public async Task HandleEventAsync_NoConfigurations_DoesNothing(EventMessage eventMessage)
|
|
|
|
|
|
{
|
|
|
|
|
|
var sutProvider = GetSutProvider(NoConfigurations());
|
|
|
|
|
|
|
|
|
|
|
|
await sutProvider.Sut.HandleEventAsync(eventMessage);
|
|
|
|
|
|
sutProvider.GetDependency<IHttpClientFactory>().Received(1).CreateClient(
|
|
|
|
|
|
Arg.Is(AssertHelper.AssertPropertyEqual(WebhookEventHandler.HttpClientName))
|
|
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
|
|
Assert.Empty(_handler.CapturedRequests);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
2025-02-04 08:02:43 -06:00
|
|
|
|
[Theory, BitAutoData]
|
2025-04-23 10:44:43 -04:00
|
|
|
|
public async Task HandleEventAsync_OneConfiguration_PostsEventToUrl(EventMessage eventMessage)
|
2025-02-04 08:02:43 -06:00
|
|
|
|
{
|
2025-04-23 10:44:43 -04:00
|
|
|
|
var sutProvider = GetSutProvider(OneConfiguration());
|
2025-02-04 08:02:43 -06:00
|
|
|
|
|
|
|
|
|
|
await sutProvider.Sut.HandleEventAsync(eventMessage);
|
|
|
|
|
|
sutProvider.GetDependency<IHttpClientFactory>().Received(1).CreateClient(
|
2025-04-23 10:44:43 -04:00
|
|
|
|
Arg.Is(AssertHelper.AssertPropertyEqual(WebhookEventHandler.HttpClientName))
|
2025-02-04 08:02:43 -06:00
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
|
|
Assert.Single(_handler.CapturedRequests);
|
|
|
|
|
|
var request = _handler.CapturedRequests[0];
|
|
|
|
|
|
Assert.NotNull(request);
|
2025-04-23 10:44:43 -04:00
|
|
|
|
var returned = await request.Content.ReadFromJsonAsync<MockEvent>();
|
|
|
|
|
|
var expected = MockEvent.From(eventMessage);
|
2025-02-04 08:02:43 -06:00
|
|
|
|
|
|
|
|
|
|
Assert.Equal(HttpMethod.Post, request.Method);
|
2025-02-11 10:20:06 -05:00
|
|
|
|
Assert.Equal(_webhookUrl, request.RequestUri.ToString());
|
2025-04-23 10:44:43 -04:00
|
|
|
|
AssertHelper.AssertPropertyEqual(expected, returned);
|
2025-02-04 08:02:43 -06:00
|
|
|
|
}
|
2025-02-14 13:38:27 -05:00
|
|
|
|
|
|
|
|
|
|
[Theory, BitAutoData]
|
2025-04-23 10:44:43 -04:00
|
|
|
|
public async Task HandleEventAsync_WrongConfigurations_DoesNothing(EventMessage eventMessage)
|
2025-02-14 13:38:27 -05:00
|
|
|
|
{
|
2025-04-23 10:44:43 -04:00
|
|
|
|
var sutProvider = GetSutProvider(WrongConfiguration());
|
|
|
|
|
|
|
|
|
|
|
|
await sutProvider.Sut.HandleEventAsync(eventMessage);
|
|
|
|
|
|
sutProvider.GetDependency<IHttpClientFactory>().Received(1).CreateClient(
|
|
|
|
|
|
Arg.Is(AssertHelper.AssertPropertyEqual(WebhookEventHandler.HttpClientName))
|
|
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
|
|
Assert.Empty(_handler.CapturedRequests);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[Theory, BitAutoData]
|
|
|
|
|
|
public async Task HandleManyEventsAsync_NoConfigurations_DoesNothing(List<EventMessage> eventMessages)
|
|
|
|
|
|
{
|
|
|
|
|
|
var sutProvider = GetSutProvider(NoConfigurations());
|
2025-02-14 13:38:27 -05:00
|
|
|
|
|
|
|
|
|
|
await sutProvider.Sut.HandleManyEventsAsync(eventMessages);
|
|
|
|
|
|
sutProvider.GetDependency<IHttpClientFactory>().Received(1).CreateClient(
|
2025-04-23 10:44:43 -04:00
|
|
|
|
Arg.Is(AssertHelper.AssertPropertyEqual(WebhookEventHandler.HttpClientName))
|
2025-02-14 13:38:27 -05:00
|
|
|
|
);
|
|
|
|
|
|
|
2025-04-23 10:44:43 -04:00
|
|
|
|
Assert.Empty(_handler.CapturedRequests);
|
|
|
|
|
|
}
|
2025-02-14 13:38:27 -05:00
|
|
|
|
|
2025-04-23 10:44:43 -04:00
|
|
|
|
|
|
|
|
|
|
[Theory, BitAutoData]
|
|
|
|
|
|
public async Task HandleManyEventsAsync_OneConfiguration_PostsEventsToUrl(List<EventMessage> eventMessages)
|
|
|
|
|
|
{
|
|
|
|
|
|
var sutProvider = GetSutProvider(OneConfiguration());
|
|
|
|
|
|
|
|
|
|
|
|
await sutProvider.Sut.HandleManyEventsAsync(eventMessages);
|
|
|
|
|
|
sutProvider.GetDependency<IHttpClientFactory>().Received(1).CreateClient(
|
|
|
|
|
|
Arg.Is(AssertHelper.AssertPropertyEqual(WebhookEventHandler.HttpClientName))
|
|
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
|
|
Assert.Equal(eventMessages.Count, _handler.CapturedRequests.Count);
|
|
|
|
|
|
var index = 0;
|
|
|
|
|
|
foreach (var request in _handler.CapturedRequests)
|
|
|
|
|
|
{
|
|
|
|
|
|
Assert.NotNull(request);
|
|
|
|
|
|
var returned = await request.Content.ReadFromJsonAsync<MockEvent>();
|
|
|
|
|
|
var expected = MockEvent.From(eventMessages[index]);
|
|
|
|
|
|
|
|
|
|
|
|
Assert.Equal(HttpMethod.Post, request.Method);
|
|
|
|
|
|
Assert.Equal(_webhookUrl, request.RequestUri.ToString());
|
|
|
|
|
|
AssertHelper.AssertPropertyEqual(expected, returned);
|
|
|
|
|
|
index++;
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[Theory, BitAutoData]
|
|
|
|
|
|
public async Task HandleManyEventsAsync_TwoConfigurations_PostsEventsToMultipleUrls(List<EventMessage> eventMessages)
|
|
|
|
|
|
{
|
|
|
|
|
|
var sutProvider = GetSutProvider(TwoConfigurations());
|
|
|
|
|
|
|
|
|
|
|
|
await sutProvider.Sut.HandleManyEventsAsync(eventMessages);
|
|
|
|
|
|
sutProvider.GetDependency<IHttpClientFactory>().Received(1).CreateClient(
|
|
|
|
|
|
Arg.Is(AssertHelper.AssertPropertyEqual(WebhookEventHandler.HttpClientName))
|
|
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
|
|
using var capturedRequests = _handler.CapturedRequests.GetEnumerator();
|
|
|
|
|
|
Assert.Equal(eventMessages.Count * 2, _handler.CapturedRequests.Count);
|
|
|
|
|
|
|
|
|
|
|
|
foreach (var eventMessage in eventMessages)
|
|
|
|
|
|
{
|
|
|
|
|
|
var expected = MockEvent.From(eventMessage);
|
|
|
|
|
|
|
|
|
|
|
|
Assert.True(capturedRequests.MoveNext());
|
|
|
|
|
|
var request = capturedRequests.Current;
|
|
|
|
|
|
Assert.NotNull(request);
|
|
|
|
|
|
Assert.Equal(HttpMethod.Post, request.Method);
|
|
|
|
|
|
Assert.Equal(_webhookUrl, request.RequestUri.ToString());
|
|
|
|
|
|
var returned = await request.Content.ReadFromJsonAsync<MockEvent>();
|
|
|
|
|
|
AssertHelper.AssertPropertyEqual(expected, returned);
|
|
|
|
|
|
|
|
|
|
|
|
Assert.True(capturedRequests.MoveNext());
|
|
|
|
|
|
request = capturedRequests.Current;
|
|
|
|
|
|
Assert.NotNull(request);
|
|
|
|
|
|
Assert.Equal(HttpMethod.Post, request.Method);
|
|
|
|
|
|
Assert.Equal(_webhookUrl2, request.RequestUri.ToString());
|
|
|
|
|
|
returned = await request.Content.ReadFromJsonAsync<MockEvent>();
|
|
|
|
|
|
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()
|
|
|
|
|
|
);
|
2025-02-14 13:38:27 -05:00
|
|
|
|
}
|
2025-02-04 08:02:43 -06:00
|
|
|
|
}
|
