request/src/request.dispatcher.tests/_fixtures/StreamOpenBehavior.cs

// Copyright (c) The Geekeey Authors
// SPDX-License-Identifier: EUPL-1.2

using System.Runtime.CompilerServices;

namespace Geekeey.Request.Dispatcher.Tests;

public class StreamOpenBehavior<TRequest, TResponse>(StreamTestTracker tracker) : IStreamRequestBehavior<TRequest, TResponse>
where TRequest : IStreamRequest<TResponse>
{
	public async IAsyncEnumerable<TResponse> HandleAsync(TRequest request, StreamHandlerDelegate<TResponse> next, [EnumeratorCancellation] CancellationToken cancellationToken)
	{
		tracker.Executed = true;
		await foreach (var response in next(request, cancellationToken))
		{
			yield return response;
		}
	}
}