如何存根DbSet.Find

时间:2014-07-26 16:42:46

标签: c# entity-framework unit-testing

// acknowledgement: http://stackoverflow.com/a/5022512/1500199
public class FakeDbSet<T> : IDbSet<T> where T : class
{
    private readonly HashSet<T> data;
    private readonly IQueryable query;

    public FakeDbSet()
    {
        data = new HashSet<T>();
        query = data.AsQueryable();
    }

    public virtual T Find(params object[] keyValues)
    {
        throw new NotImplementedException();
    }
}

如何实施Find

我需要确定T的主键值才能在Find方法中执行键值比较,但我不知道如何。

3 个答案:

答案 0 :(得分:2)

试试这个:

public class FakeDbSet<T> : IDbSet<T> where T : class
{
    private Func<T, object[], bool> _findSelector
    private readonly HashSet<T> data;
    private readonly IQueryable query;

    public FakeDbSet(Func<T, object[], bool> findSelector)
    {
        _findSelector = findSelector;
        data = new HashSet<T>();
        query = data.AsQueryable();
    }

    public virtual T Find(params object[] keyValues)
    {
        return _data.SingleOrDefault(item => _findSelector(item, keyValues));
    }
}

答案 1 :(得分:1)

如果我理解正确您需要存储DbSet对象的Find()方法以进行单元测试。

如果你正在使用版本为6或更高版本的EntityFramework并使用Mocking Frameworks,如Moq或NSubstitute或FakeItEasy,你可以轻松实现这一点,而无需像FakeDbSet那样编写类。

本文提供了对存根/模拟过程的完整说明:http://msdn.microsoft.com/en-us/data/dn314429.aspx

使用本文中的类的示例。

    [Test]
    public void Foo_WhenCalled_CallsDb()
    {
        //Arrange
        var fakeDbSet = new Mock<DbSet<Course>>();
        fakeDbSet.Setup(dbs => dbs.Find(It.IsAny<object>())).Returns(new Course { CourseID = 125 });

        var fakeContext = new Mock<SchoolEntities>();
        fakeContext.Setup(c => c.Courses).Returns(fakeDbSet.Object);


        var foo = new Foo(fakeContext.Object);

        //Act
        string result = foo.MyMethod();

        //Assert
        Assert.AreEqual("125", result);
    }

对于其他版本的EntityFramework,请考虑使用“Repository”模式 - 它提供了出色的测试功能。 我希望我的回答对你有帮助。

答案 2 :(得分:0)

覆盖方法

public class FakePostsDbSet : FakeDbSet<Post>
{
    public override Post Find(params object[] keyValues)
    {
        return this.SingleOrDefault(
            post => post.Slug == (string) keyValues.Single());
    }
}