aspnetcore/test/Microsoft.AspNet.Hosting.Tests/UseRequestServicesFacts.cs

// Copyright (c) Microsoft Open Technologies, Inc. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using System;
using Microsoft.AspNet.Builder;
using Microsoft.AspNet.Http.Core;
using Microsoft.AspNet.RequestContainer;
using Microsoft.Framework.DependencyInjection;
using Microsoft.Framework.Logging;
using Xunit;

namespace Microsoft.AspNet.Hosting.Tests
{
    public class RequestServicesContainerFacts
    {
        [Fact]
        public void RequestServicesAvailableOnlyAfterRequestServices()
        {
            var baseServiceProvider = HostingServices.Create().BuildServiceProvider();
            var builder = new ApplicationBuilder(baseServiceProvider);

            bool foundRequestServicesBefore = false;
            builder.Use(next => async c =>
            {
                foundRequestServicesBefore = c.RequestServices != null;
                await next.Invoke(c);
            });
            builder.UseRequestServices();
            bool foundRequestServicesAfter = false;
            builder.Use(next => async c =>
            {
                foundRequestServicesAfter = c.RequestServices != null;
                await next.Invoke(c);
            });

            var context = new DefaultHttpContext();
            builder.Build().Invoke(context);
            Assert.False(foundRequestServicesBefore);
            Assert.True(foundRequestServicesAfter);
        }

        [Theory]
        [InlineData(true)]
        [InlineData(false)]
        public void EnsureRequestServicesSetsRequestServices(bool initializeApplicationServices)
        {
            var baseServiceProvider = HostingServices.Create().BuildServiceProvider();
            var builder = new ApplicationBuilder(baseServiceProvider);

            bool foundRequestServicesBefore = false;
            builder.Use(next => async c =>
            {
                foundRequestServicesBefore = c.RequestServices != null;
                await next.Invoke(c);
            });
            builder.Use(next => async c =>
            {
                using (var container = RequestServicesContainer.EnsureRequestServices(c, baseServiceProvider))
                {
                    await next.Invoke(c);
                }
            });
            bool foundRequestServicesAfter = false;
            builder.Use(next => async c =>
            {
                foundRequestServicesAfter = c.RequestServices != null;
                await next.Invoke(c);
            });

            var context = new DefaultHttpContext();
            if (initializeApplicationServices)
            {
                context.ApplicationServices = baseServiceProvider;
            }
            builder.Build().Invoke(context);
            Assert.False(foundRequestServicesBefore);
            Assert.True(foundRequestServicesAfter);
        }

        [Theory]
        [InlineData(typeof(ITypeActivator))]
        [InlineData(typeof(IHostingEnvironment))]
        [InlineData(typeof(ILoggerFactory))]
        [InlineData(typeof(IHttpContextAccessor))]
        [InlineData(typeof(IApplicationLifetime))]
        public void UseRequestServicesHostingImportedServicesAreDefined(Type service)
        {
            var baseServiceProvider = HostingServices.Create().BuildServiceProvider();
            var builder = new ApplicationBuilder(baseServiceProvider);

            builder.UseRequestServices();

            var fromAppServices = builder.ApplicationServices.GetRequiredService(service);

            Assert.NotNull(fromAppServices);
            Assert.Equal(baseServiceProvider.GetRequiredService(service), fromAppServices);
        }
    }
}