public void ExecuteAsync_throws_OperationCanceledException_if_task_is_cancelled()
{
var executionPlan = new ObjectQueryExecutionPlan(null, null, null, MergeOption.AppendOnly, false, null, null);
Assert.Throws <OperationCanceledException>(
() => executionPlan.ExecuteAsync <object>(null, null, new CancellationToken(canceled: true))
.GetAwaiter().GetResult());
}
public void ExecuteAsync_sets_the_parameter_values_and_returns_the_result()
{
var entityCommandDefinitionMock = Mock.Get(EntityClient.MockHelper.CreateEntityCommandDefinition());
entityCommandDefinitionMock.Setup(m => m.Parameters).Returns(
new[]
{
new EntityParameter
{
ParameterName = "Par1"
}
});
var objectContextMock = Mock.Get(Objects.MockHelper.CreateMockObjectContext <string>());
objectContextMock.Setup(m => m.CommandTimeout).Returns(3);
entityCommandDefinitionMock.Setup(
m => m.ExecuteStoreCommandsAsync(
It.IsAny <EntityCommand>(),
It.IsAny <CommandBehavior>(), It.IsAny <CancellationToken>()))
.Returns(
(EntityCommand ec, CommandBehavior cb, CancellationToken ct) =>
{
Assert.Equal(1, ec.Parameters.Count);
Assert.Equal(2, ec.Parameters[0].Value);
Assert.Equal(3, ec.CommandTimeout);
Assert.Equal(new[] { objectContextMock.Object }, ec.InterceptionContext.ObjectContexts);
return(Task.FromResult(
Common.Internal.Materialization.MockHelper.CreateDbDataReader(
new[] { new object[] { "Bar" } })));
});
var shaperFactory = new ShaperFactory <string>(
1,
Objects.MockHelper.CreateCoordinatorFactory(shaper => (string)shaper.Reader.GetValue(0)),
new[] { typeof(string) }, new[] { true }, MergeOption.AppendOnly);
var edmTypeMock = new Mock <EdmType>();
edmTypeMock.Setup(m => m.BuiltInTypeKind).Returns(BuiltInTypeKind.SimpleType);
var objectQueryExecutionPlan = new ObjectQueryExecutionPlan(
entityCommandDefinitionMock.Object,
shaperFactory, TypeUsage.Create(edmTypeMock.Object), MergeOption.AppendOnly, false, null, null);
var objectParameterCollectionMock = new Mock <ObjectParameterCollection>(new ClrPerspective(new MetadataWorkspace()));
objectParameterCollectionMock
.Setup(m => m.GetEnumerator())
.Returns(((IEnumerable <ObjectParameter>) new[] { new ObjectParameter("Par1", 2) }).GetEnumerator());
var result = objectQueryExecutionPlan.ExecuteAsync <string>(
objectContextMock.Object,
objectParameterCollectionMock.Object, CancellationToken.None).Result;
Assert.Equal("Bar", result.SingleAsync().Result);
}
private void ExecuteAsync_disposes_the_reader_on_exception(bool streaming)
{
var entityCommandDefinitionMock = Mock.Get(EntityClient.MockHelper.CreateEntityCommandDefinition());
var objectContextMock = Mock.Get(Objects.MockHelper.CreateMockObjectContext <string>());
objectContextMock.Setup(m => m.CommandTimeout).Returns(3);
DbDataReader reader = null;
entityCommandDefinitionMock.Setup(
m => m.ExecuteStoreCommandsAsync(
It.IsAny <EntityCommand>(),
It.IsAny <CommandBehavior>(), It.IsAny <CancellationToken>()))
.Returns(
(EntityCommand ec, CommandBehavior cb, CancellationToken ct) =>
{
Assert.Equal(new[] { objectContextMock.Object }, ec.InterceptionContext.ObjectContexts);
reader = Common.Internal.Materialization.MockHelper.CreateDbDataReader(
new[] { new object[] { "Bar" } });
return(Task.FromResult(reader));
});
var shaperFactory = new ShaperFactory <string>(
1,
Objects.MockHelper.CreateCoordinatorFactory(shaper => (string)shaper.Reader.GetValue(0)),
new[] { typeof(string) }, new[] { true }, MergeOption.AppendOnly);
var edmTypeMock = new Mock <EdmType>();
edmTypeMock.Setup(m => m.BuiltInTypeKind).Returns(BuiltInTypeKind.CollectionType);
var objectQueryExecutionPlan = new ObjectQueryExecutionPlan(
entityCommandDefinitionMock.Object,
shaperFactory, TypeUsage.Create(edmTypeMock.Object), MergeOption.AppendOnly, streaming, null, null);
var objectParameterCollectionMock = new Mock <ObjectParameterCollection>(new ClrPerspective(new MetadataWorkspace()));
objectParameterCollectionMock
.Setup(m => m.GetEnumerator())
.Returns(((IEnumerable <ObjectParameter>) new[] { new ObjectParameter("Par1", 2) }).GetEnumerator());
Assert.Throws <InvalidCastException>(
() =>
ExceptionHelpers.UnwrapAggregateExceptions(
() => objectQueryExecutionPlan.ExecuteAsync <string>(
objectContextMock.Object,
objectParameterCollectionMock.Object, CancellationToken.None).Result));
Assert.Equal(true, reader.IsClosed);
var readerMock = Mock.Get(reader);
readerMock.Verify(
m => m.GetFieldValueAsync <object>(It.IsAny <int>(), It.IsAny <CancellationToken>()), streaming ? Times.Never() : Times.Once());
}
private void ExecuteAsync_with_streaming(bool streaming)
{
var entityCommandDefinitionMock = Mock.Get(EntityClient.MockHelper.CreateEntityCommandDefinition());
var objectContextMock = Mock.Get(Objects.MockHelper.CreateMockObjectContext <string>());
objectContextMock.Setup(m => m.CommandTimeout).Returns(3);
DbDataReader reader = null;
entityCommandDefinitionMock.Setup(
m => m.ExecuteStoreCommandsAsync(
It.IsAny <EntityCommand>(),
It.IsAny <CommandBehavior>(), It.IsAny <CancellationToken>()))
.Returns(
(EntityCommand ec, CommandBehavior cb, CancellationToken ct) =>
{
Assert.Equal(new[] { objectContextMock.Object }, ec.InterceptionContext.ObjectContexts);
reader = Common.Internal.Materialization.MockHelper.CreateDbDataReader(
new[] { new object[] { "Bar" } });
Assert.Equal(streaming ? CommandBehavior.Default : CommandBehavior.SequentialAccess, cb);
return(Task.FromResult(reader));
});
var shaperFactory = new ShaperFactory <string>(
1,
Objects.MockHelper.CreateCoordinatorFactory(shaper => (string)shaper.Reader.GetValue(0)),
new[] { typeof(string) }, new[] { true }, MergeOption.AppendOnly);
var edmTypeMock = new Mock <EdmType>();
edmTypeMock.Setup(m => m.BuiltInTypeKind).Returns(BuiltInTypeKind.SimpleType);
var objectQueryExecutionPlan = new ObjectQueryExecutionPlan(
entityCommandDefinitionMock.Object,
shaperFactory, TypeUsage.Create(edmTypeMock.Object), MergeOption.AppendOnly, streaming, null, null);
var objectParameterCollectionMock = new Mock <ObjectParameterCollection>(new ClrPerspective(new MetadataWorkspace()));
objectParameterCollectionMock
.Setup(m => m.GetEnumerator())
.Returns(((IEnumerable <ObjectParameter>) new[] { new ObjectParameter("Par1", 2) }).GetEnumerator());
Assert.NotNull(objectQueryExecutionPlan.ExecuteAsync <string>(
objectContextMock.Object,
objectParameterCollectionMock.Object, CancellationToken.None).Result);
Assert.Equal(!streaming, reader.IsClosed);
}