使用Moq进行单元测试 - 数值不能为空

Question

我正在使用EF6。生成的代码是一样的东西：

public partial class MyDataContext : DbContext 
{ 
    public MyDataContext() : base("name=mydata") 
    { 
    } 

    public virtual DbSet<Book> Books { get; set; } 
} 

然后，我有一个通用的存储库，如：

public class GenericRepository<TObject> where TObject : class 
{ 
    protected readonly MyDataContext Context; 

    protected GenericRepository(MyDataContext context) 
    { 
     Context = context; 
    } 

    public virtual DbSet<TObject> GetAll() 
    { 
     return Context.Set<TObject>(); 
    } 
} 

然后，我有一个使用GenericRepository返回数据的服务：

public class MyDataService<TObject> where TObject : class 
{ 
    private readonly MyDataContext context; 

    public MyDataService(MyDataContext ct) 
    { 
     context = ct; 
    } 

    public ICollection<TObject> GetAll() 
    { 
     var r = new GenericRepository<TObject>(context); 
     return r.GetAll().ToList(); 
    } 
} 

所以我可以得到像这样的所有书籍：

var ds = new MyDataService<Book>(new MyDataContext()); 
var data = ds.GetAll(); 

这工作正常。接下来，我尝试使用起订量来的单元测试上面的代码的东西，如：

var books = new List<Book> 
{ 
    new Book {Id = 1, Name = "BBB"}, 
    new Book {Id = 2, Name = "ZZZ"}, 
    new Book {Id = 3, Name = "AAA"}, 
}.AsQueryable(); 

var mockSet = new Mock<DbSet<Book>>(); 
mockSet.As<IQueryable<Book>>().Setup(m => m.Provider).Returns(books.Provider); 
mockSet.As<IQueryable<Book>>().Setup(m => m.Expression).Returns(books.Expression); 
mockSet.As<IQueryable<Book>>().Setup(m => m.ElementType).Returns(books.ElementType); 
mockSet.As<IQueryable<Book>>().Setup(m => GetEnumerator()).Returns(books.GetEnumerator()); 

var mockContext = new Mock<MyDataContext>(); 
mockContext.Setup(c => c.Books).Returns(mockSet.Object); 

var service = new MyDataService<Book>(mockContext.Object); 
var data = service.GetAll(); 

不过，我得到的最后一行的"Value cannot be null.\r\nParameter name: source"错误。当我进入代码时，我看到上下文对象中的Books集合是空的。

我在做什么错？

Answer 1

这是因为在测试的数据上下文但在实际代码设置.Setup(c => c.Books)访问Context.Set<TObject>()在GetAll()方法，所以对于测试最终将作为null。

尝试更改为

mockContext.Setup(c => c.Set<Book>()).Returns(mockSet.Object);