private static async Task WaitForHandlerCompletion(WaitForResult waitForResult, int timeoutInMs)
{
using var timer = new Timer(new TimerCallback(TimedOutCallback), waitForResult, timeoutInMs, Timeout.Infinite);
while (!waitForResult.HasCompleted && !waitForResult.HasTimedOut)
{
await Task.Delay(100);
}
}
public static async Task <WaitForResult> WaitFor <T>(
this TestContext context,
Func <Task> func,
bool assertOnTimeout = true,
bool assertOnError = false,
int timeoutInMs = 15000)
{
var waitForResult = new WaitForResult();
var hook = context.Hooks.Find(h => h is MessageBusHook <T>) as MessageBusHook <T>;
hook.OnReceived = _ => waitForResult.HasStarted = true;
hook.OnProcessed = _ => waitForResult.HasCompleted = true;
hook.OnErrored = (ex, _) => waitForResult.SetHasErrored(ex);
try
{
await func();
}
catch (Exception ex)
{
waitForResult.SetHasErrored(ex);
}
await WaitForHandlerCompletion(waitForResult, timeoutInMs);
if (assertOnTimeout)
{
waitForResult.HasTimedOut.Should().Be(false, "handler should not have timed out");
}
if (assertOnError)
{
waitForResult.HasErrored.Should().Be(false, $"handler should not have errored with error '{waitForResult.LastException?.Message}'");
}
return(waitForResult);
}
public async Task WhenTheSendInvitationCommandIsReceived()
{
_waitResult = await _context.WaitFor <MessageContext>(async() =>
await _context.TestMessageBus.Send(_message));
}
