up
This commit is contained in:
master
@@ -0,0 +1,167 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using FluentAssertions;
|
||||
using Microsoft.Extensions.Logging.Abstractions;
|
||||
using Microsoft.Extensions.Options;
|
||||
using StellaOps.Notify.Models;
|
||||
using StellaOps.Notify.Queue;
|
||||
using StellaOps.Notify.Worker;
|
||||
using StellaOps.Notify.Worker.Handlers;
|
||||
using StellaOps.Notify.Worker.Processing;
|
||||
using Xunit;
|
||||
|
||||
namespace StellaOps.Notify.Worker.Tests;
|
||||
|
||||
public sealed class NotifyEventLeaseProcessorTests
|
||||
{
|
||||
[Fact]
|
||||
public async Task ProcessOnce_ShouldAcknowledgeSuccessfulLease()
|
||||
{
|
||||
var lease = new FakeLease();
|
||||
var queue = new FakeEventQueue(lease);
|
||||
var handler = new TestHandler();
|
||||
var options = Options.Create(new NotifyWorkerOptions { LeaseBatchSize = 1, LeaseDuration = TimeSpan.FromSeconds(5) });
|
||||
var processor = new NotifyEventLeaseProcessor(queue, handler, options, NullLogger<NotifyEventLeaseProcessor>.Instance, TimeProvider.System);
|
||||
|
||||
var processed = await processor.ProcessOnceAsync(CancellationToken.None);
|
||||
|
||||
processed.Should().Be(1);
|
||||
lease.AcknowledgeCount.Should().Be(1);
|
||||
lease.ReleaseCount.Should().Be(0);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ProcessOnce_ShouldRetryOnHandlerFailure()
|
||||
{
|
||||
var lease = new FakeLease();
|
||||
var queue = new FakeEventQueue(lease);
|
||||
var handler = new TestHandler(shouldThrow: true);
|
||||
var options = Options.Create(new NotifyWorkerOptions { LeaseBatchSize = 1, LeaseDuration = TimeSpan.FromSeconds(5) });
|
||||
var processor = new NotifyEventLeaseProcessor(queue, handler, options, NullLogger<NotifyEventLeaseProcessor>.Instance, TimeProvider.System);
|
||||
|
||||
var processed = await processor.ProcessOnceAsync(CancellationToken.None);
|
||||
|
||||
processed.Should().Be(1);
|
||||
lease.AcknowledgeCount.Should().Be(0);
|
||||
lease.ReleaseCount.Should().Be(1);
|
||||
lease.LastDisposition.Should().Be(NotifyQueueReleaseDisposition.Retry);
|
||||
}
|
||||
|
||||
private sealed class FakeEventQueue : INotifyEventQueue
|
||||
{
|
||||
private readonly Queue<INotifyQueueLease<NotifyQueueEventMessage>> _leases;
|
||||
|
||||
public FakeEventQueue(params INotifyQueueLease<NotifyQueueEventMessage>[] leases)
|
||||
{
|
||||
_leases = new Queue<INotifyQueueLease<NotifyQueueEventMessage>>(leases);
|
||||
}
|
||||
|
||||
public ValueTask<NotifyQueueEnqueueResult> PublishAsync(NotifyQueueEventMessage message, CancellationToken cancellationToken = default)
|
||||
=> throw new NotSupportedException();
|
||||
|
||||
public ValueTask<IReadOnlyList<INotifyQueueLease<NotifyQueueEventMessage>>> LeaseAsync(NotifyQueueLeaseRequest request, CancellationToken cancellationToken = default)
|
||||
{
|
||||
if (_leases.Count == 0)
|
||||
{
|
||||
return ValueTask.FromResult<IReadOnlyList<INotifyQueueLease<NotifyQueueEventMessage>>>(Array.Empty<INotifyQueueLease<NotifyQueueEventMessage>>());
|
||||
}
|
||||
|
||||
return ValueTask.FromResult<IReadOnlyList<INotifyQueueLease<NotifyQueueEventMessage>>>(new[] { _leases.Dequeue() });
|
||||
}
|
||||
|
||||
public ValueTask<IReadOnlyList<INotifyQueueLease<NotifyQueueEventMessage>>> ClaimExpiredAsync(NotifyQueueClaimOptions options, CancellationToken cancellationToken = default)
|
||||
=> ValueTask.FromResult<IReadOnlyList<INotifyQueueLease<NotifyQueueEventMessage>>>(Array.Empty<INotifyQueueLease<NotifyQueueEventMessage>>());
|
||||
}
|
||||
|
||||
private sealed class FakeLease : INotifyQueueLease<NotifyQueueEventMessage>
|
||||
{
|
||||
private readonly NotifyQueueEventMessage _message;
|
||||
|
||||
public FakeLease()
|
||||
{
|
||||
var notifyEvent = NotifyEvent.Create(
|
||||
Guid.NewGuid(),
|
||||
kind: "test.event",
|
||||
tenant: "tenant-1",
|
||||
ts: DateTimeOffset.UtcNow,
|
||||
payload: null);
|
||||
|
||||
_message = new NotifyQueueEventMessage(notifyEvent, "notify:events", traceId: "trace-123");
|
||||
}
|
||||
|
||||
public string MessageId { get; } = Guid.NewGuid().ToString("n");
|
||||
|
||||
public int Attempt { get; internal set; } = 1;
|
||||
|
||||
public DateTimeOffset EnqueuedAt { get; } = DateTimeOffset.UtcNow;
|
||||
|
||||
public DateTimeOffset LeaseExpiresAt { get; private set; } = DateTimeOffset.UtcNow.AddSeconds(30);
|
||||
|
||||
public string Consumer { get; } = "worker-1";
|
||||
|
||||
public string Stream => _message.Stream;
|
||||
|
||||
public string TenantId => _message.TenantId;
|
||||
|
||||
public string? PartitionKey => _message.PartitionKey;
|
||||
|
||||
public string IdempotencyKey => _message.IdempotencyKey;
|
||||
|
||||
public string? TraceId => _message.TraceId;
|
||||
|
||||
public IReadOnlyDictionary<string, string> Attributes => _message.Attributes;
|
||||
|
||||
public NotifyQueueEventMessage Message => _message;
|
||||
|
||||
public int AcknowledgeCount { get; private set; }
|
||||
|
||||
public int ReleaseCount { get; private set; }
|
||||
|
||||
public NotifyQueueReleaseDisposition? LastDisposition { get; private set; }
|
||||
|
||||
public Task AcknowledgeAsync(CancellationToken cancellationToken = default)
|
||||
{
|
||||
AcknowledgeCount++;
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
public Task RenewAsync(TimeSpan leaseDuration, CancellationToken cancellationToken = default)
|
||||
{
|
||||
LeaseExpiresAt = DateTimeOffset.UtcNow.Add(leaseDuration);
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
public Task ReleaseAsync(NotifyQueueReleaseDisposition disposition, CancellationToken cancellationToken = default)
|
||||
{
|
||||
LastDisposition = disposition;
|
||||
ReleaseCount++;
|
||||
Attempt++;
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
public Task DeadLetterAsync(string reason, CancellationToken cancellationToken = default)
|
||||
=> Task.CompletedTask;
|
||||
}
|
||||
|
||||
private sealed class TestHandler : INotifyEventHandler
|
||||
{
|
||||
private readonly bool _shouldThrow;
|
||||
|
||||
public TestHandler(bool shouldThrow = false)
|
||||
{
|
||||
_shouldThrow = shouldThrow;
|
||||
}
|
||||
|
||||
public Task HandleAsync(NotifyQueueEventMessage message, CancellationToken cancellationToken)
|
||||
{
|
||||
if (_shouldThrow)
|
||||
{
|
||||
throw new InvalidOperationException("handler failure");
|
||||
}
|
||||
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,26 @@
|
||||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
<PropertyGroup>
|
||||
<TargetFramework>net10.0</TargetFramework>
|
||||
<ImplicitUsings>enable</ImplicitUsings>
|
||||
<Nullable>enable</Nullable>
|
||||
<IsPackable>false</IsPackable>
|
||||
<UseConcelierTestInfra>false</UseConcelierTestInfra>
|
||||
</PropertyGroup>
|
||||
<ItemGroup>
|
||||
<PackageReference Include="FluentAssertions" Version="6.12.0" />
|
||||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.14.0" />
|
||||
<PackageReference Include="xunit" Version="2.9.2" />
|
||||
<PackageReference Include="xunit.runner.visualstudio" Version="2.8.2">
|
||||
<PrivateAssets>all</PrivateAssets>
|
||||
</PackageReference>
|
||||
<PackageReference Include="coverlet.collector" Version="6.0.4">
|
||||
<PrivateAssets>all</PrivateAssets>
|
||||
</PackageReference>
|
||||
<PackageReference Include="Microsoft.Extensions.Logging.Abstractions" Version="8.0.0" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ProjectReference Include="..\StellaOps.Notify.Worker\StellaOps.Notify.Worker.csproj" />
|
||||
<ProjectReference Include="..\StellaOps.Notify.Queue\StellaOps.Notify.Queue.csproj" />
|
||||
<ProjectReference Include="..\StellaOps.Notify.Models\StellaOps.Notify.Models.csproj" />
|
||||
</ItemGroup>
|
||||
</Project>
|
||||
Reference in New Issue
Block a user