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()); }