如何模拟返回的模拟对象的方法?

时间:2014-11-20 14:09:24

标签: c# entity-framework unit-testing moq

我是Moq的新手。我按照here提供的指导,对实体框架6项目进行单元测试。所以我有一个模板化的方法来创建假表:

    protected Mock<DbSet<TheType>> MockDBSet<TheType>(List<TheType> data) where TheType : class
    {
        var mockSet = new Mock<DbSet<TheType>>();
        var dataSet = data.AsQueryable();
        mockSet.As<IQueryable<TheType>>().Setup(m => m.Provider).Returns(dataSet.Provider);
        mockSet.As<IQueryable<TheType>>().Setup(m => m.Expression).Returns(dataSet.Expression);
        mockSet.As<IQueryable<TheType>>().Setup(m => m.ElementType).Returns(dataSet.ElementType);
        mockSet.As<IQueryable<TheType>>().Setup(m => m.GetEnumerator()).Returns(dataSet.GetEnumerator());
        mockSet.Setup(x => x.Add(It.IsAny<TheType>()))
            .Returns(new Func<TheType, TheType>(x =>
            {
                data.Add(x);
                return data.Last();
            }));

        return mockSet;
    }

使用上述方法可以很好地添加和查询假数据库:

var db = new Mock<BloggingContext>();
db.Setup(m => m.Blog s).Returns(MockDBSet<Blog >(
    new List<Blog>() 
).Object);

BloggingContext context = db.Object;

Blog blog= new Blog();
context.Blogs.Add(blog); //fine
Assert.IsTrue(context.Blogs.Count() == 1); //fine

EF还在实体集合上提供了“本地”属性,可以访问未保存的实体。因此,在不进行模拟时,BloggingContext.Blogs.Local会传回一个未保存实体的ObservableCollection集合。调用BloggingContext.SaveChanges()时,对象从BloggingContext.Blogs.Local移动到BloggingContext.Blogs。

我想模仿这种行为,所以我创建了一个新类:

    public class FakeBlogs : List<Blog>
    {
        ObservableCollection<Blog> _local = new ObservableCollection<Blog>();
        ObservableCollection<Blog> Local { get { return _local; } }

        public void Add (Blog item)
        {
            _local.Add(item);
        }
    }

单元测试时,以下代码有效:

var db = new Mock<BloggingContext>();
db.Setup(m => m.Blog s).Returns(MockDBSet<Blog>(
    new FakeBlogs() //<===== Changed to use FakeBlogs
).Object);

BloggingContext context = db.Object;

Blog blog= new Blog();    
context.Blogs.Add(blog); //fine
Assert.IsTrue(context.Blogs.Count() == 1); //fine

但是,使用Local属性会抛出NPE,因为Local属性为null。

var blog = (from i in context.Blogs.Local select i).FirstOrDefault();//throws NPE

如何成功模拟Local属性?

1 个答案:

答案 0 :(得分:1)

固定。我将以下行添加到MockDBSet模板函数中:

        mockSet.Setup(m => m.Local).Returns(data.Local);

宣布&#34; Local&#34; FakeTable的财产作为公共财产。感谢所有感兴趣的人。