2014-03-24 39 views
2

我有一个从表返回的记录列表中的泛型方法:单元测试泛型方法与最小起订量

public List<T> GetValidRecords<T>() where T: class, IGetListOfTables 
{ 
    try 
    { 
     return _context.Set<T>().Where(x => x.Valid == 1).ToList(); 
    } 
    catch (Exception ex) 
    { 
     throw new Exception(ex.Message); 
    } 
} 

,我有此方法的单元测试:

[TestMethod] 
public void GetValidRecords() 
{ 
    var data = new List<tableName> 
    { 
     new tableName() {Valid= 1}, 
     new tableName() {Valid= 1} 
    }.AsQueryable(); 

    var mockSet = new Mock<DbSet<tableName>>(); 
    mockSet.As<IQueryable<tableName>>().Setup(m => m.Provider).Returns(data.Provider); 
    mockSet.As<IQueryable<tableName>>().Setup(m => m.Expression).Returns(data.Expression); 
    mockSet.As<IQueryable<tableName>>().Setup(m => m.ElementType).Returns(data.ElementType); 
    mockSet.As<IQueryable<tableName>>().Setup(m => m.GetEnumerator()).Returns(data.GetEnumerator()); 

    var mockContext = new Mock<ALMONEntitiesNew>(); 
    mockContext.Setup(x => x.tableName).Returns(mockSet.Object); 
    var database = new Database(mockContext.Object); 
    var numberOfRecords = database.GetValidRecords<tableName>(); 
    Assert.AreEqual(2, numberOfRecords.Count, "Wrong number of valid records."); 
} 

的问题是,我从表中获得实际的记录数,而不是数量。 我该如何解决它?

+0

你没有'Setup'为'设置<>'方法 –

+0

@JordyLangen你能提供样品吗? – Sasha

回答

1

您需要从GetValidRecords方法中获取EF实现的所有依赖项,特别是_context,否则EF特定实现将不断流入您的单元测试。为了将GetValidRecords作为一个单元来测试,你需要使它能够独立运行。如果你想按照原样进行测试,我建议使用集成测试,它实际上是从数据库中检索记录并断言它们恢复正常 - 这不需要使用任何模拟框架,并且是一种非常好的测试方法这个功能。

在作出GetValidRecords的话题独立,我看到DbSet实现IEnumerable,所以也许你想要的是这样的:

public static List<T> GetValidRecords<T>(this IEnumerable<T> source) where T: class, IGetListOfTables 
{ 
    if (null == source) 
    { 
     throw new ArgumentNullException("source"); 
    } 

    return source.Where(x => x.Valid == 1).ToList(); 
} 
+0

好的答案,但在您的示例代码中,try/catch吞噬了原始异常的堆栈跟踪,应该删除。 –

+0

@MikeStockdale是的,我知道 - 只是想保持话题。感谢您注意到,我认为解决这个问题最好。会做。 –