ASP.NET 5 单元测试中使用依赖注入

相关博文：《ASP.NET 5 使用 TestServer 进行单元测试》

在上一篇博文中，主要说的是，使用 TestServer 对 ASP.NET 5 WebApi 进行单元测试，依赖注入在 WebApi Startup.cs 中完成，所以 UnitTest 中只需要使用 TestServer 启动 WebApi 站点就可以了，因为整个解决方案的对象都是用 ASP.NET 5 “自带”的依赖注入进行管理，所以，在对 ASP.NET 5 类库进行单元测试的时候，我都是手动进行 new 创建的对象，比如针对 Application 的单元测试，贴一段 AdImageServiceTest 中的代码：

namespace CNBlogs.Ad.Application.Tests

{

    public class AdTextServiceTest : BaseTest

    {

        private IAdTextService _adTextService;

        public AdTextServiceTest(ITestOutputHelper output)

            : base(output)

        {

            Bootstrapper.Startup.ConfigureMapper();

            IUnitOfWork unitOfWork = new UnitOfWork(dbContext);

            IMemcached memcached = new EnyimMemcached(null);

            _adTextService = new AdTextService(new AdTextRepository(dbContext),

                new AdTextCreativeRepository(dbContext),

                new UserService(memcached),

                unitOfWork,

                memcached);

        }

        [Fact]

        public async Task GetByAdTextsTest()

        {

            var adTexts = await _adTextService.GetAdTexts();

            Assert.NotNull(adTexts);

            adTexts.ForEach(x => Console.WriteLine($"{x.Title}({x.Link})"));

        }

    }

}

AdImageServiceTest 构造函数中的代码，是不是看起来很别扭呢？我当时这样写，也是没有办法的，因为依赖注入的配置是写在 Startup 中，比如下面代码：

namespace CNBlogs.Ad.WebApi

{

    public class Startup

    {

        public Startup(IHostingEnvironment env, IApplicationEnvironment appEnv)

        {

            // Set up configuration sources.

            var builder = new ConfigurationBuilder()

                .AddJsonFile("appsettings.json")

                .AddEnvironmentVariables();

            Configuration = builder.Build();

        }

        public IConfigurationRoot Configuration { get; set; }

        // This method gets called by the runtime. Use this method to add services to the container.

        public void ConfigureServices(IServiceCollection services)

        {

            // Add framework services.

            services.AddMvc();

            services.AddSingleton<IUnitOfWork, UnitOfWork>();

            services.AddScoped<IDbContext, EFDbContext>();

            services.AddSingleton<IAdTextRepository, AdTextRepository>();

            services.AddSingleton<IAdTextService, AdTextService>();

        }

    }

}

这样设计导致的结果是，针对类库项目的单元测试，就没办法使用依赖注入获取对象了，我后来想使用针对 WebApi 单元测试的方式，来对类库进行单元测试，比如用 TestServer 来启动，但类库中没有办法获取所注入的对象，构成函数注入会报错，[FromServices] 属性注入是 MVC Controller 中的东西，并不支持，所以针对类库的单元测试，和 WebApi 的单元测试并不是一样。

IServiceCollection 的程序包是 Microsoft.Extensions.DependencyInjection.Abstractions，我原来以为它和 ASP.NET 5 Web 应用程序相关，其实它们也没啥关系，你可以脱离 ASP.NET 5 Web 应用程序，独立使用它，比如在类库的单元测试中，但如果这样设计使用，我们首先需要做一个工作，把 Startup.cs 中的 ConfigureServices 配置，独立出来，比如放在 CNBlogs.Ad.Bootstrapper 中，这样 Web 应用程序和单元测试项目，都可以使用它，减少代码的重复，比如我们可以进行下面设计：

namespace CNBlogs.Ad.Bootstrapper

{

    public static class Startup

    {

        public static void Configure(this IServiceCollection services, string connectionString)

        {

            services.AddEntityFramework()

                .AddSqlServer()

                .AddDbContext<EFDbContext>(options => options.UseSqlServer(connectionString));

            services.AddEnyimMemcached();

            ConfigureMapper();

            services.AddSingleton<IUnitOfWork, UnitOfWork>();

            services.AddScoped<IDbContext, EFDbContext>();

            services.AddSingleton<IAdTextRepository, AdTextRepository>();

            services.AddSingleton<IAdTextService, AdTextService>();

        }

        public static void ConfigureMapper()

        {

            Mapper.CreateMap<CNBlogs.Ad.Domain.Entities.AdText, AdTextDTO>();

        }

    }

}

ASP.NET 5 WebApi 项目中的 Startup.cs 配置会非常简单，只需要下面代码：

public void ConfigureServices(IServiceCollection services)

{

    services.Configure(Configuration["data:ConnectionString"]);//add using CNBlogs.Ad.Bootstrapper;

}

好了，做好上面工作后，单元测试中使用依赖注入就非常简单了，为了减少重复代码，我们可以先抽离出一个 BaseTest：

namespace CNBlogs.Ad.BaseTests

{

    public class BaseTest

    {

        protected readonly ITestOutputHelper output;

        protected IServiceProvider provider;

        public BaseTest(ITestOutputHelper output)

        {

            var connectionString = "";

            var services = new ServiceCollection();

            this.output = output;

            services.Configure(connectionString);////add using CNBlogs.Ad.Bootstrapper;

            provider = services.BuildServiceProvider();

        }

    }

}

可以看到，我们并没有使用 TestServer，而是手动创建一个 ServiceCollection，它有点类似于我们之前写依赖注入的 Unity Container，不过它们有很大不同，ServiceCollection 的功能更加强大，从 Bootstrapper.Startup 中可以看到，它可以注入 EF、MVC、Memcache 等等服务对象，BuildServiceProvider 的作用就是获取注入的服务对象，我们下面会用到：

namespace CNBlogs.Ad.Repository.Tests

{

    public class AdTextServiceTest : BaseTest

    {

        private IAdTextService _adTextService;

        public AdTextServiceTest(ITestOutputHelper output)

            : base(output)

        {

            _adTextService = provider.GetService<IAdTextService>();

        }

        [Fact]

        public async Task GetByAdTextsTest()

        {

            var adTexts = await _adTextService.GetAdTexts();

            Assert.NotNull(adTexts);

            adTexts.ForEach(x => Console.WriteLine($"{x.Title}({x.Link})"));

        }

    }

}

这段代码和一开始的那段代码，形成了鲜明对比，这就是代码设计的魅力所在！！！

巴特西

ASP.NET 5 单元测试中使用依赖注入

最新文章

热门文章