DBSet的mock

DBSet的mock类. 从老外那边找来的.

算是为测试解决了一个头痛的问题.

 

需要注意一点,在EF4.1的环境中,你不能直接用继承DbContext的类来进行调试. : 为他抽象出一个接口,然后在单元测试那边继承这个接口.

由于现在的开发环境是EF4.1 我没有测试更低的版本是否可以直接使用.

 

public class DbSetMock<T> : IDbSet<T> where T : class { readonly HashSet<T> _data; readonly IQueryable _query; public InMemoryDbSet() { _data = new HashSet<T>(); _query = _data.AsQueryable(); } public T Add(T entity) { _data.Add(entity); return entity; } public T Attach(T entity) { _data.Add(entity); return entity; } public TDerivedEntity Create<TDerivedEntity>() where TDerivedEntity : class, T { throw new NotImplementedException(); } public T Create() { return Activator.CreateInstance<T>(); } public virtual T Find(params object[] keyValues) { throw new NotImplementedException("Derive from FakeDbSet and override Find"); } public System.Collections.ObjectModel.ObservableCollection<T> Local { get { return new System.Collections.ObjectModel.ObservableCollection<T>(_data); } } public T Remove(T entity) { _data.Remove(entity); return entity; } public IEnumerator<T> GetEnumerator() { return _data.GetEnumerator(); } IEnumerator IEnumerable.GetEnumerator() { return _data.GetEnumerator(); } public Type ElementType { get { return _query.ElementType; } } public Expression Expression { get { return _query.Expression; } } public IQueryProvider Provider { get { return _query.Provider; } } }