ASP.NET 在本地方法中测试 DbContext

Question

我有一个 ASP.NET MVC 项目，并且正在按照以下内容编写我的服务：

public class UserService : IUserService
{
    public void AddUser(UserModel userModel)
    {
        if (ModelState.IsValid){
            var user = new User
            {
                FirstName = userModel.FirstName,
                LastName = userModel.LastName,
                Email = userModel.Email,
                Age = userModel.Age
            };
            using (var context = new MyDbContext())
            {
                context.Users.Add(user);
            }
        }
    }
    //... More code for service
}

还有我的 DbContext 类：

public class MyDbContext : DbContext 
{
    public MyDbContext(): base("name=mydb"){}

    public DbSet<User> Users {get; set;}

    //.. More domain objects.
}

这在大多数情况下运行良好，但是当我想尝试测试它时就会出现问题。我遵循一般准则here 将其模拟出来，但前提是 DbContext 被声明为类实例，而不是函数变量。我想这样做是因为它使我能够控制事务和上下文处理。有人对如何做到这一点有任何建议吗？

我的模拟是这样的：

public class UserServiceTest 
{
    [Fact]
    public void TestAddUser()
    {
        var userSet = GetQueryableMockDbSet(userData.ToArray());
        var context = new MyDbContext
        {
            Users = userSet
        };
        var userService = new UserService();
        userService.AddUser(); // HOW DO I GET CONTEXT IN HERE?
    }   

    private static DbSet<T> GetQueryableMockDbSet<T>(params T[] sourceList) where T : class
        {
            var queryable = sourceList.AsQueryable();

            var dbSet = new Mock<DbSet<T>>();
            dbSet.As<IQueryable<T>>().Setup(m => m.Provider).Returns(queryable.Provider);
            dbSet.As<IQueryable<T>>().Setup(m => m.Expression).Returns(queryable.Expression);
            dbSet.As<IQueryable<T>>().Setup(m => m.ElementType).Returns(queryable.ElementType);
            dbSet.As<IQueryable<T>>().Setup(m => m.GetEnumerator()).Returns(queryable.GetEnumerator());

            return dbSet.Object;
        }
}

Answer 1

实现目标的一种方法

我想这样做是因为它使我能够控制事务和上下文处理。

会做一些类似的事情

public class UserService : IUserService {
   private ContextFactory _ContextFactory;

   public UserService(ContextFactory contextFactory) {
       _ContextFactory = contextFactory;
   }

   public void AddUser(UserModel userModel) {
       ...
       using (var context = _ContextFactory.CreateInstance()) {
           context.Users.Add(user);
       }
   }

}

这将允许您注入 ContextFactory 的不同实例，这将使用 Mock 创建 DbContext

Answer 2

我肯定更喜欢 IOC 方法，但如果您想要一个快速解决方案而无需重写太多代码，您可以执行以下操作：

public class UserService : IUserService 
{
    public static Func<MyDbContext> CreateContext = () => new MyDbContext();

    public UserService ()
    {
    } 

    public void AddUser(UserModel userModel)
    {
        ...
        using (var context = CreateContext())
        {
            context.Users.Add(user);
        }
    }
}

您不必更改生产代码，但可以在您的 TestMethod 中更改

    var context = new MyDbContext
    {
        Users = userSet
    };
    UserService.CreateContext = () => context;

Answer 3

您需要在服务的构造函数中传递DbContext。最简单的方法是开始使用任何 IoC 框架（AutoFac、Unity、Ninject 等）。要模拟数据库，最简单的方法是开始使用任何模拟库，例如 Moq 和 RhinoMocks。

public class UserService : IUserService 
{
    private MyDbContext _context;

    public UserService (MyDbContext dbContext)
    {
        _context = dbContext;
    } 
}