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 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)),
                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 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.GetValue(It.IsAny <int>()), streaming ? Times.Never() : Times.Once());
        }
예제 #3
0
        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)),
                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);
        }