129 lines
4.2 KiB
C#
129 lines
4.2 KiB
C#
using System.Linq.Expressions;
|
|
using Microsoft.EntityFrameworkCore.Query;
|
|
|
|
namespace HopFrame.Tests.Core.Models;
|
|
|
|
// A mock implementation for async query provider
|
|
internal class TestAsyncQueryProvider<TEntity> : IAsyncQueryProvider {
|
|
private readonly IQueryProvider _inner;
|
|
|
|
internal TestAsyncQueryProvider(IQueryProvider inner) {
|
|
_inner = inner;
|
|
}
|
|
|
|
public IQueryable CreateQuery(Expression expression) {
|
|
return new TestAsyncEnumerable<TEntity>(expression);
|
|
}
|
|
|
|
public IQueryable<TElement> CreateQuery<TElement>(Expression expression) {
|
|
return new TestAsyncEnumerable<TElement>(expression);
|
|
}
|
|
|
|
public object Execute(Expression expression) {
|
|
return _inner.Execute(expression);
|
|
}
|
|
|
|
public TResult Execute<TResult>(Expression expression) {
|
|
return _inner.Execute<TResult>(expression);
|
|
}
|
|
|
|
public TResult ExecuteAsync<TResult>(Expression expression,
|
|
CancellationToken cancellationToken = new CancellationToken()) {
|
|
return _inner.Execute<TResult>(expression);
|
|
}
|
|
|
|
public IAsyncEnumerable<TResult> ExecuteAsync<TResult>(Expression expression) {
|
|
return new TestAsyncEnumerable<TResult>(expression);
|
|
}
|
|
}
|
|
|
|
internal class TestAsyncEnumerable<T> : EnumerableQuery<T>, IAsyncEnumerable<T>, IQueryable<T> {
|
|
public TestAsyncEnumerable(IEnumerable<T> enumerable) : base(enumerable) { }
|
|
|
|
public TestAsyncEnumerable(Expression expression) : base(expression) { }
|
|
|
|
public IAsyncEnumerator<T> GetAsyncEnumerator(CancellationToken cancellationToken = default) {
|
|
return new TestAsyncEnumerator<T>(this.AsEnumerable().GetEnumerator());
|
|
}
|
|
|
|
IQueryProvider IQueryable.Provider => new TestAsyncQueryProvider<T>(this);
|
|
}
|
|
|
|
internal class TestAsyncEnumerator<T> : IAsyncEnumerator<T> {
|
|
private readonly IEnumerator<T> _inner;
|
|
|
|
public TestAsyncEnumerator(IEnumerator<T> inner) {
|
|
_inner = inner ?? throw new ArgumentNullException(nameof(inner));
|
|
}
|
|
|
|
public T Current => _inner.Current;
|
|
|
|
public ValueTask DisposeAsync() {
|
|
_inner.Dispose();
|
|
return new ValueTask();
|
|
}
|
|
|
|
public ValueTask<bool> MoveNextAsync() {
|
|
return new ValueTask<bool>(_inner.MoveNext());
|
|
}
|
|
}
|
|
|
|
/*// A mock implementation for async query provider
|
|
internal class TestAsyncQueryProvider<TEntity> : IAsyncQueryProvider {
|
|
private readonly IQueryProvider _inner;
|
|
|
|
internal TestAsyncQueryProvider(IQueryProvider inner) {
|
|
_inner = inner;
|
|
}
|
|
|
|
public IQueryable CreateQuery(Expression expression) {
|
|
return new TestAsyncEnumerable<TEntity>(expression);
|
|
}
|
|
|
|
public IQueryable<TElement> CreateQuery<TElement>(Expression expression) {
|
|
return new TestAsyncEnumerable<TElement>(expression);
|
|
}
|
|
|
|
public object? Execute(Expression expression) {
|
|
return _inner.Execute(expression);
|
|
}
|
|
|
|
public TResult Execute<TResult>(Expression expression) {
|
|
return _inner.Execute<TResult>(expression);
|
|
}
|
|
|
|
public TResult ExecuteAsync<TResult>(Expression expression, CancellationToken cancellationToken) {
|
|
return _inner.Execute<TResult>(expression);
|
|
}
|
|
|
|
public IAsyncEnumerable<TResult> ExecuteAsync<TResult>(Expression expression) {
|
|
return new TestAsyncEnumerable<TResult>(expression);
|
|
}
|
|
}
|
|
|
|
internal class TestAsyncEnumerable<T> : EnumerableQuery<T>, IAsyncEnumerable<T>, IQueryable<T> {
|
|
public TestAsyncEnumerable(IEnumerable<T> enumerable) : base(enumerable) { }
|
|
|
|
public TestAsyncEnumerable(Expression expression) : base(expression) { }
|
|
|
|
public IAsyncEnumerator<T> GetAsyncEnumerator(CancellationToken cancellationToken = default) {
|
|
return new TestAsyncEnumerator<T>(this.AsEnumerable().GetEnumerator());
|
|
}
|
|
|
|
IQueryProvider IQueryable.Provider => new TestAsyncQueryProvider<T>(this);
|
|
}
|
|
|
|
internal class TestAsyncEnumerator<T>(IEnumerator<T> inner) : IAsyncEnumerator<T> {
|
|
private readonly IEnumerator<T> _inner = inner ?? throw new ArgumentNullException(nameof(inner));
|
|
|
|
public T Current => _inner.Current;
|
|
|
|
public ValueTask DisposeAsync() {
|
|
_inner.Dispose();
|
|
return new ValueTask();
|
|
}
|
|
|
|
public ValueTask<bool> MoveNextAsync() {
|
|
return new ValueTask<bool>(_inner.MoveNext());
|
|
}
|
|
}*/ |