public void Execute_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.ExecuteStoreCommands(It.IsAny<EntityCommand>(), It.IsAny<CommandBehavior>()))
.Returns(
(EntityCommand ec, CommandBehavior cb) =>
{
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 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.Execute<string>(objectContextMock.Object, objectParameterCollectionMock.Object);
Assert.Equal("Bar", result.Single());
}
private ObjectResult <T> GetResults(MergeOption?forMergeOption)
{
this.QueryState.ObjectContext.EnsureConnection();
try
{
ObjectQueryExecutionPlan execPlan = this.QueryState.GetExecutionPlan(forMergeOption);
return(execPlan.Execute <T>(this.QueryState.ObjectContext, this.QueryState.Parameters));
}
catch
{
this.QueryState.ObjectContext.ReleaseConnection();
throw;
}
}
private void Execute_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.ExecuteStoreCommands(It.IsAny<EntityCommand>(), It.IsAny<CommandBehavior>()))
.Returns(
(EntityCommand ec, CommandBehavior cb) =>
{
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 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 ObjectParameter[0]).GetEnumerator());
objectQueryExecutionPlan.Execute<string>(objectContextMock.Object, objectParameterCollectionMock.Object);
Assert.Equal(!streaming, reader.IsClosed);
}
private void Execute_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.ExecuteStoreCommands(It.IsAny<EntityCommand>(), It.IsAny<CommandBehavior>()))
.Returns(
(EntityCommand ec, CommandBehavior cb) =>
{
Assert.Equal(new[] { objectContextMock.Object }, ec.InterceptionContext.ObjectContexts);
reader = Common.Internal.Materialization.MockHelper.CreateDbDataReader(
new[] { new object[] { "Bar" } });
return reader;
});
var shaperFactory = new ShaperFactory<string>(
1,
Objects.MockHelper.CreateCoordinatorFactory(shaper => (string)shaper.Reader.GetValue(2)),
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 ObjectParameter[0]).GetEnumerator());
Assert.Throws<InvalidCastException>(
() =>
objectQueryExecutionPlan.Execute<string>(objectContextMock.Object, objectParameterCollectionMock.Object));
Assert.Equal(true, reader.IsClosed);
var readerMock = Mock.Get(reader);
readerMock.Verify(m => m.GetValues(It.IsAny<object[]>()), streaming ? Times.Never() : Times.Once());
}
