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
115
116
117
118
119
120
121
122
123
124
|
// Copyright (c) Microsoft Open Technologies, Inc. All rights reserved. See License.txt in the project root for license information.
namespace System.Data.Entity
{
using System.Data.Entity.Core.Metadata.Edm;
using System.Data.Entity.ModelConfiguration.Edm;
internal class TestModelBuilder
{
public static implicit operator EdmModel(TestModelBuilder testModelBuilder)
{
return testModelBuilder._model;
}
private readonly EdmModel _model;
private EntityType _entityType;
private int _counter;
public TestModelBuilder()
{
_model = new EdmModel(DataSpace.CSpace);
}
public TestModelBuilder Entities(params string[] names)
{
foreach (var name in names)
{
Entity(name);
}
return this;
}
public TestModelBuilder Entity(string name, bool addSet = true)
{
_entityType = _model.AddEntityType(name);
Type type = new MockType(name);
_entityType.Annotations.SetClrType(type);
if (addSet)
{
_model.AddEntitySet(name + "Set", _entityType);
}
return this;
}
public TestModelBuilder Property(string propertyName)
{
var property1 = EdmProperty.Primitive(propertyName, PrimitiveType.GetEdmPrimitiveType(PrimitiveTypeKind.String));
_entityType.AddMember(property1);
var property = property1;
property.SetClrPropertyInfo(new MockPropertyInfo(typeof(string), propertyName));
return this;
}
public TestModelBuilder Key(string key)
{
var property1 = EdmProperty.Primitive(key, PrimitiveType.GetEdmPrimitiveType(PrimitiveTypeKind.String));
_entityType.AddMember(property1);
var property = property1;
property.SetClrPropertyInfo(new MockPropertyInfo(typeof(int), key));
_entityType.AddKeyMember(property);
return this;
}
public TestModelBuilder Subclass(string name, string baseName = null)
{
var previous = baseName == null ? _entityType : _model.GetEntityType(baseName);
Entity(name, false);
_entityType.BaseType = previous;
return this;
}
public TestModelBuilder Association(
string sourceEntity, RelationshipMultiplicity sourceEndKind,
string targetEntity, RelationshipMultiplicity targetEndKind)
{
_model.AddAssociationSet(
"AssociationSet" + _counter++,
_model.AddAssociationType(
"Association",
_model.GetEntityType(sourceEntity), sourceEndKind,
_model.GetEntityType(targetEntity), targetEndKind));
return this;
}
public TestModelBuilder Association(
string name,
string sourceEntity, RelationshipMultiplicity sourceEndKind, string sourceNavigationProperty,
string targetEntity, RelationshipMultiplicity targetEndKind, string targetNavigationProperty)
{
var sourceEntityType = _model.GetEntityType(sourceEntity);
var targetEntityType = _model.GetEntityType(targetEntity);
var association
= _model.AddAssociationType(name, sourceEntityType, sourceEndKind, targetEntityType, targetEndKind);
_model.AddAssociationSet(name + "Set", association);
if (sourceNavigationProperty != null)
{
sourceEntityType.AddNavigationProperty(sourceNavigationProperty, association);
}
if (targetNavigationProperty != null)
{
targetEntityType.AddNavigationProperty(targetNavigationProperty, association);
}
return this;
}
}
}
|
