2014-01-11 50 views
5

我想模拟实体框架6.0.2使用JustMock的异步方法。我下面testing with async queries但使用起订量我想这个转换成JustMock与Mock Multiple Interfaces帮助,但得到一个异常写:如何使用JustMock模拟EF 6异步方法?

的提供商源的IQueryable没有实现 IDbAsyncQueryProvider。只有实现 IDbAsyncQueryProvider的提供者才能用于实体框架异步 操作。欲了解更多详情,请参阅 http://go.microsoft.com/fwlink/?LinkId=287068

这里是我的代码:

var dummyData = GetEmployeeSkills(); 
var mockSet = Mock.Create<DbSet<EmployeeSkill>>(); 
(mockSet as IDbAsyncEnumerable<EmployeeSkill>).Arrange(x => x.GetAsyncEnumerator()) 
     .Returns(new TestDbAsyncEnumerator<EmployeeSkill>(dummyData.GetEnumerator())); 

(mockSet as IQueryable<EmployeeSkill>).Arrange(x => x.Provider).Returns(new TestDbAsyncQueryProvider<EmployeeSkill>(dummyData.Provider)); 

(mockSet as IQueryable<EmployeeSkill>).Arrange(x => x.Expression).Returns(dummyData.Expression); 
(mockSet as IQueryable<EmployeeSkill>).Arrange(x => x.ElementType).Returns(dummyData.ElementType); 
(mockSet as IQueryable<EmployeeSkill>).Arrange(x => x.GetEnumerator()).Returns(dummyData.GetEnumerator()); 

var mockContext = Mock.Create<TimeSketchContext>(); 
mockContext.Arrange(x => x.Set<EmployeeSkill>()).Returns(mockSet); 

baseRepository = new BaseRepository<EmployeeSkill>(mockContext); 

private EmployeeSkill GetEmployeeSkill() 
    { 
     return new EmployeeSkill 
     { 
      SkillDescription = "SkillDescription", 
      SkillName = "SkillName", 
      Id = 1 
     }; 
    } 

    private IQueryable<EmployeeSkill> GetEmployeeSkills() 
    { 
     return new List<EmployeeSkill> 
     { 
      GetEmployeeSkill(), 
      GetEmployeeSkill(), 
      GetEmployeeSkill(), 
     }.AsQueryable(); 
    } 

测试:

[Fact] 
public async Task DbTest() 
{ 
    var data = await baseRepository.FindAsync(1); 
    Assert.NotEqual(null, data); 
} 

库:

public class BaseRepository<T> : IRepositoryBase<T> where T : class, IEntity, new() 
{ 
    protected readonly DbContext InnerDbContext; 
    protected DbSet<T> InnerDbSet; 

    public BaseRepository(IDbContext innerDbContext) 
    { 
     InnerDbContext = innerDbContext as DbContext; 
     InnerDbSet = innerDbContext.Set<T>(); 
    } 

    public virtual Task<T> FindAsync(long id) 
    { 
     return InnerDbSet.FirstOrDefaultAsync(x=>x.Id == id); 
    } 
} 

接口:

public interface IDbContext 
{ 
    DbSet<T> Set<T>() where T : class; 
} 

语境:

public class TimeSketchContext : DbContext, IDbContext 
{ 
    public virtual DbSet<EmployeeSkill> EmployeeSkill { get; set; } 
} 
+0

感谢您的示例代码。我现在正在测试工作.. tq。 –

回答

2

因为JustMock可以当你写

var mockContext = Mock.Create<TimeSketchContext>(); 
mockContext.Arrange(x => x.Set<EmployeeSkill>()).Returns(mockSet); 

嘲笑非虚方法会嘲笑DbContext.Set<>而不是你IDbContext.Set<>,所以你得到的异常。

至少有2解决这个:

  • 模拟你的IDbContext接口

    var mockContext = Mock.Create<IDbContext>(); 
    
  • 或更改回你的BaseRepository使用DbContext,而不是你的接口:

    public class BaseRepository<T> : IRepositoryBase<T> where T : class, IEntity, new() 
    { 
        protected readonly DbContext InnerDbContext; 
        protected DbSet<T> InnerDbSet; 
    
        public BaseRepository(DbContext innerDbContext) 
        { 
         InnerDbContext = innerDbContext; 
         InnerDbSet = InnerDbContext.Set<T>(); 
        } 
    
        public virtual Task<T> FindAsync(long id) 
        { 
         return InnerDbSet.FirstOrDefaultAsync(x => x.Id == id); 
        } 
    }