public DbSetMock(IEnumerable <TEntity> initialEntities, Func <TEntity, KeyContext, object> keyFactory, bool asyncQuerySupport = true)
{
_store = new DbSetBackingStore <TEntity>(initialEntities, keyFactory);
var data = _store.GetDataAsQueryable();
Object = Substitute.For <DbSet <TEntity>, IQueryable <TEntity>, IAsyncEnumerable <TEntity> >();
((IQueryable <TEntity>)Object).Provider.Returns(asyncQuerySupport ? new DbAsyncQueryProvider <TEntity>(data.Provider) : data.Provider);
Object.AsQueryable().Provider.Returns(asyncQuerySupport ? new DbAsyncQueryProvider <TEntity>(data.Provider) : data.Provider);
Object.AsQueryable().Expression.Returns(data.Expression);
Object.AsQueryable().ElementType.Returns(data.ElementType);
((IQueryable <TEntity>)Object).GetEnumerator().Returns(a => data.GetEnumerator());
((IEnumerable)Object).GetEnumerator().Returns(a => data.GetEnumerator());
if (asyncQuerySupport)
{
((IAsyncEnumerable <TEntity>)Object).GetEnumerator().Returns(a => new DbAsyncEnumerator <TEntity>(data.GetEnumerator()));
}
Object.When(a => a.Add(Arg.Any <TEntity>())).Do(b => _store.Add(b.ArgAt <TEntity>(0)));
Object.When(a => a.AddRange(Arg.Any <IEnumerable <TEntity> >())).Do(b => _store.Add(b.ArgAt <IEnumerable <TEntity> >(0)));
Object.When(a => a.Remove(Arg.Any <TEntity>())).Do(b => _store.Remove(b.ArgAt <TEntity>(0)));
Object.When(a => a.RemoveRange(Arg.Any <IEnumerable <TEntity> >())).Do(b => _store.Remove(b.ArgAt <IEnumerable <TEntity> >(0)));
Object.Find(Arg.Any <object[]>()).Returns(info => _store.Find(info.Args()[0] as object[]));
Object.FindAsync(Arg.Any <object[]>()).Returns(info => Task.FromResult(_store.Find(info.Args()[0] as object[])));
Object.FindAsync(Arg.Any <object[]>(), Arg.Any <CancellationToken>()).Returns(info => Task.FromResult(_store.Find(info.Args()[0] as object[])));
_store.UpdateSnapshot();
}
public DbSetMock(IEnumerable <TEntity> initialEntities, Func <TEntity, KeyContext, object> keyFactory, bool asyncQuerySupport = true)
{
_store = new DbSetBackingStore <TEntity>(initialEntities, keyFactory);
var data = _store.GetDataAsQueryable();
Func <DbAsyncEnumerable <TEntity> > getQuery =
() => new DbAsyncEnumerable <TEntity>(data.AsQueryable());
DbSet = Substitute.For <DbSet <TEntity>, IQueryable <TEntity>, IDbAsyncEnumerable <TEntity> >();
((IQueryable <TEntity>)DbSet).Provider.Returns(asyncQuerySupport ? new DbAsyncQueryProvider <TEntity>(data.Provider) : data.Provider);
DbSet.AsQueryable().Provider.Returns(asyncQuerySupport ? new DbAsyncQueryProvider <TEntity>(data.Provider) : data.Provider);
DbSet.AsQueryable().Expression.Returns(data.Expression);
DbSet.AsQueryable().ElementType.Returns(data.ElementType);
((IQueryable <TEntity>)DbSet).GetEnumerator().Returns(a => _store.GetDataEnumerator());
if (asyncQuerySupport)
{
((IDbAsyncEnumerable <TEntity>)DbSet).GetAsyncEnumerator()
.Returns(a => getQuery().GetAsyncEnumerator());
}
DbSet.AsNoTracking().Returns(DbSet);
DbSet.Include(Arg.Any <string>()).Returns(DbSet);
DbSet.AsNoTracking().Include(Arg.Any <string>()).Returns(DbSet);
DbSet.When(a => a.Add(Arg.Any <TEntity>())).Do(b => _store.Add(b.ArgAt <TEntity>(0)));
DbSet.When(a => a.AddRange(Arg.Any <IEnumerable <TEntity> >())).Do(b => _store.Add(b.ArgAt <IEnumerable <TEntity> >(0)));
DbSet.When(a => a.Remove(Arg.Any <TEntity>())).Do(b => _store.Remove(b.ArgAt <TEntity>(0)));
DbSet.When(a => a.RemoveRange(Arg.Any <IEnumerable <TEntity> >())).Do(b => _store.Remove(b.ArgAt <IEnumerable <TEntity> >(0)));
_store.UpdateSnapshot();
}