1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// Copyright (c) Microsoft Open Technologies, Inc. All rights reserved. See License.txt in the project root for license information.
namespace System.Data.Entity.Internal
{
using System.Collections.Generic;
using System.Data.Entity.Core;
using System.Data.Entity.Core.Metadata.Edm;
using System.Data.Entity.Core.Objects.DataClasses;
using System.Data.Entity.Internal.Linq;
using Moq;
public static class MockHelper
{
internal static InternalSqlSetQuery CreateInternalSqlSetQuery(string sql, bool isNoTracking = false, bool streaming = false, params object[] parameters)
{
return new InternalSqlSetQuery(new Mock<InternalSetForMock<FakeEntity>>().Object, sql, isNoTracking, streaming, parameters);
}
internal static InternalSqlNonSetQuery CreateInternalSqlNonSetQuery(string sql, bool streaming = false, params object[] parameters)
{
return new InternalSqlNonSetQuery(new Mock<InternalContextForMock>().Object, typeof(object), sql, streaming, parameters);
}
internal static Mock<InternalSqlNonSetQuery> CreateMockInternalSqlNonSetQuery(string sql, bool streaming = false, params object[] parameters)
{
return new Mock<InternalSqlNonSetQuery>(new Mock<InternalContextForMock>().Object, typeof(object), sql, streaming, parameters);
}
internal static Mock<IEntityStateEntry> CreateMockStateEntry<TEntity>() where TEntity : class, new()
{
var mockStateEntry = new Mock<IEntityStateEntry>();
var fakeEntity = new TEntity();
mockStateEntry.Setup(e => e.Entity).Returns(fakeEntity);
var entitySet = new EntitySet("foo set", "foo schema", "foo table", "foo query", new EntityType());
entitySet.ChangeEntityContainerWithoutCollectionFixup(new EntityContainer("foo container", DataSpace.CSpace));
mockStateEntry.Setup(e => e.EntitySet).Returns(entitySet);
mockStateEntry.Setup(e => e.EntityKey).Returns(
new EntityKey(
"foo.bar",
new[] { new KeyValuePair<string, object>("foo", "bar") }));
var modifiedProps = new List<string>
{
"Foo",
"ValueTypeProp",
"Bar"
};
mockStateEntry.Setup(e => e.GetModifiedProperties()).Returns(modifiedProps);
mockStateEntry.Setup(e => e.SetModifiedProperty(It.IsAny<string>())).Callback<string>(modifiedProps.Add);
return mockStateEntry;
}
internal static Mock<InternalEntityEntryForMock<TEntity>> CreateMockInternalEntityEntry<TEntity>()
where TEntity : class, new()
{
return CreateMockInternalEntityEntry<TEntity, object>(new TEntity());
}
internal static Mock<InternalEntityEntryForMock<TEntity>> CreateMockInternalEntityEntry<TEntity>(
TEntity entity, bool isDetached = false)
where TEntity : class, new()
{
return CreateMockInternalEntityEntry<TEntity, object>(entity, isDetached: isDetached);
}
internal static Mock<InternalEntityEntryForMock<TEntity>> CreateMockInternalEntityEntry<TEntity, TRelated>(
TEntity entity, EntityReference<TRelated> entityReference = null, bool isDetached = false)
where TEntity : class, new()
where TRelated : class
{
return CreateMockInternalEntityEntry(entity, entityReference, null, isDetached);
}
internal static Mock<InternalEntityEntryForMock<TEntity>> CreateMockInternalEntityEntry<TEntity, TRelated>(
TEntity entity, EntityCollection<TRelated> entityCollection, bool isDetached = false)
where TEntity : class, new()
where TRelated : class
{
return CreateMockInternalEntityEntry(entity, null, entityCollection, isDetached);
}
internal static Mock<InternalEntityEntryForMock<TEntity>> CreateMockInternalEntityEntry<TEntity, TRelated>(
TEntity entity, EntityReference<TRelated> entityReference, EntityCollection<TRelated> entityCollection, bool isDetached)
where TEntity : class, new()
where TRelated : class
{
var mockInternalEntityEntry = new Mock<InternalEntityEntryForMock<TEntity>>
{
CallBase = true
};
mockInternalEntityEntry.SetupGet(e => e.Entity).Returns(entity);
mockInternalEntityEntry.SetupGet(e => e.EntityType).Returns(typeof(TEntity));
mockInternalEntityEntry.SetupGet(e => e.IsDetached).Returns(isDetached);
mockInternalEntityEntry.Setup(e => e.GetRelatedEnd("Reference")).Returns(entityReference);
mockInternalEntityEntry.Setup(e => e.GetRelatedEnd("Collection")).Returns(entityCollection);
return mockInternalEntityEntry;
}
internal static Mock<InternalEntityPropertyEntry> CreateMockInternalEntityPropertyEntry(object entity)
{
var propertyEntry = new Mock<InternalEntityPropertyEntry>(
CreateMockInternalEntityEntry(
entity, new EntityReference<FakeEntity>(), new EntityCollection<FakeEntity>(), isDetached: false).Object,
new PropertyEntryMetadataForMock());
propertyEntry.SetupGet(p => p.InternalEntityEntry).Returns(new InternalEntityEntryForMock<object>());
return propertyEntry;
}
}
}
|
