public void UseCorrelation_AddsAllServices()
{
var hostBuilder = new Mock <IWebHostBuilder>();
Action <IServiceCollection> configActionPassed = null;
hostBuilder
.Setup(b => b.ConfigureServices(It.IsAny <Action <IServiceCollection> >()))
.Callback <Action <IServiceCollection> >((configAction) =>
{
configActionPassed = configAction;
});
hostBuilder.Object.PropagateIstioHeaders();
var serviceCollection = new MockServiceCollection();
configActionPassed(serviceCollection);
Assert.Equal(5, serviceCollection.Count);
Assert.Equal(1, serviceCollection.Count(d => d.ServiceType == typeof(IstioHeadersFetcherMiddleware) && d.Lifetime == ServiceLifetime.Transient));
Assert.Equal(1, serviceCollection.Count(d => d.ServiceType == typeof(HeadersPropagationDelegatingHandler) && d.Lifetime == ServiceLifetime.Transient));
Assert.Equal(1, serviceCollection.Count(d => d.ServiceType == typeof(IIstioHeadersHolder) && d.ImplementationType == typeof(IstioHeadersHolder) && d.Lifetime == ServiceLifetime.Scoped));
Assert.Equal(1, serviceCollection.Count(d => d.ServiceType == typeof(IStartupFilter) && d.ImplementationType == typeof(IstioHeadersFetcherMiddlewareStartupFilter) && d.Lifetime == ServiceLifetime.Singleton));
Assert.Equal(1, serviceCollection.Count(d => d.ServiceType == typeof(IHttpMessageHandlerBuilderFilter) && d.ImplementationType == typeof(HeadersPropagationMessageHandlerBuilderFilter) && d.Lifetime == ServiceLifetime.Singleton));
}
