将IQueryable <t>转换为DbSet <t> </t> </t>

时间:2013-09-09 04:25:48

标签: c# unit-testing iqueryable dbset

我不确定这是否可行,但我正在尝试对使用DbSet的存储库进行单元测试。我认为最简单的解决方案就是创建一个Enumerable,然后用它替换DbSet,这是我的尝试。

我正在使用C#,EntityFramework,XUnit和Moq

[Fact]
public void SomeTest() 
{
    //arrange
    var mockContext = new Mock<MyDbContext>();
    var mockData = new List<Person>
        {
            new Person { Name = "Jim", Age = 47 }
        };
    mockContext.Setup(db => db.Persons).Returns((DbSet<Person>)mockData.AsQueryable());

    var repo = PersonRepository(mockContext.Object); 

    //act
    var result = repo.GetByFirstName("Jim");

    //assert
    //do some assertion
}

抛出的错误是它无法将类型EnumerableQuery转换为mockContext.Returns语句中的DbSet。

这与接口的外观类似。

PersonRepository.cs

public class PersonRepository: EFRepository<Person>, IPersonRepository
{
    public PersonRepository(DbContext dbContext) : base(dbContext)
    {
    }

    public IQueryable<Link> GetByFirstName(string name)
    {
        return DbSet.Where(p => p.FirstName == name);
    }
}

IPersonRepository.cs

public interface IPersonRepository: IRepository<Person>
{
    IQueryable<Person> GetByFirstName(string name);
}

IRepository.cs

public interface IRepository<T> where T : class 
{
    IQueryable<T> GetAll();
    T GetById(int id);
    void Add(T entity);
    void Delete(T entity);
    void Delete(int id);
    void Update(T entity);
}

EFRepository.cs

public class EFRepository<T> : IRepository<T> where T : class
{
    protected DbContext DbContext { get; set; }
    public IDbSet<T> DbSet { get; set; }

    public EFRepository(DbContext dbContext)
    {
        if (dbContext == null)
            throw new ArgumentNullException("dbContext");

        DbContext = dbContext;
        DbSet = DbContext.Set<T>();
    }
    ...
}

1 个答案:

答案 0 :(得分:1)

您可以做的是将Linq表达式从存储库移动到业​​务逻辑中,然后模拟存储库。

public interface IPersonRepository : IRepository<Person>
{
    IQueryable<Person> GetAll { get; }
}

public class PersonRepository : EFRepository<Person>, IPersonRepository
{
    // ...

    public IQueryable<Person> GetAll
    {
        get { return DbSet; }
    }
}

public class SomeBusinessLogicClass
{
    private readonly IPersonRepository _people;

    public SomeBusinessLogicClass(IPersonRepository people)
    {
        _people = people;
    }

    public IEnumerable<Person> GetByFirstName(string name)
    {
        return _people.GetAll.Where(p => p.FirstName == name);
    }
}

现在重写测试以验证业务逻辑。

[Fact]
public void SomeTest() 
{
    //arrange
    var mockRepository = new Mock<IPersonRepository>();
    var mockData = new List<Person>
        {
            new Person { Name = "Jim", Age = 47 }
        };
    mockRepository.Setup(x => x.GetAll).Returns(mockData);

    var bl = new SomeBusinessLogicClass(mockRepository.Object); 

    //act
    var result = bl.GetByFirstName("Jim");

    //assert
    //do some assertion
}