单元测试双向EF关系

时间:2012-12-17 16:40:36

标签: entity-framework unit-testing

我正在做一个小练习项目来提高我的单元测试技能。我正在使用Entity Framework Code First。

我正在使用FakeDBSet,它适用于简单的实体列表。当返回实体树时,事情并不那么好。特别是不保持双向关系,因为这是实体框架魔术的一部分。

我有两个班级:

public class Book
{
    public virtual ICollection<Review> Reviews {get; set;}
}

public class Review
{
    public virtual Book Book { get; set;}
}

如果我将图书设置为评论,则评论不会添加到图书的评论集中。它在使用EF时会发生,但在我的虚假版本中却没有。

有没有办法嘲笑这种行为,还是我不应该依靠EF来实现双向关系?或者是在嘲笑数据上下文只是浪费时间?

2 个答案:

答案 0 :(得分:6)

这实际上是一个非常常见的问题(而且没有一个非常好的答案)。在EF内部发生了一个名为fixups的进程,它在检测更改循环内运行(在添加/删除时触发的那个以及一些其他更改)。这将评估模型中的反向链接。当你开始模拟你的上下文时,你将失去检测更改循环,从而失去修正。

在过去,我通过理解我的模拟的这个特殊限制并确保我以正确的方式在代码中理解我的设置代码(这让我们面对它非常不理想)。这里的另一个选择是在单元测试中使用某种形式的真正轻量级数据库并继续使用EF。

答案 1 :(得分:5)

我发现一个可能的解决方案是创建一个模拟EF修正代码的模拟对象。

以下是使用Mocking框架NSubstitute的示例:

private static Book CreateMockBook()
{
    var book = Substitute.For<Book>();

    // Mock EF fixup: Add a review to collection should also set book for the review
    book.Reviews.Add(Arg.Do<Review>((x) => { if(x.Book != book) x.Book = book; }));

    return book;
}

private static Review CreateMockReview()
{
    var review = Substitute.For<Review>();

    // Mock EF fixup: Set a book for the review should also should add the review to book's review collection
    review.When(x  => x.Book = Arg.Any<Book>()).Do(x => review.Book.Review.Add(review));

    return review;
}

这很好用,但我不确定是否需要模拟这种行为表明我的测试太复杂或我的代码正在利用它不应该的副作用。

我对其他人对此的看法感兴趣吗?