-
Notifications
You must be signed in to change notification settings - Fork 644
/
GBTestObjectsRegistry.m
206 lines (175 loc) · 7.2 KB
/
GBTestObjectsRegistry.m
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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
//
// GBTestObjectsRegistry.m
// appledoc
//
// Created by Tomaz Kragelj on 26.7.10.
// Copyright (C) 2010, Gentle Bytes. All rights reserved.
//
#import "GBApplicationSettingsProvider.h"
#import "GBDataObjects.h"
#import "GBTestObjectsRegistry.h"
#define GBMOCKVALUE(variable) [NSValue value:&variable withObjCType:@encode(typeof(variable))]
@implementation GBTestObjectsRegistry
#pragma mark Common objects creation methods
+ (id)realSettingsProvider {
GBApplicationSettingsProvider *result = [GBApplicationSettingsProvider provider];
result.projectName = @"project";
result.projectCompany = @"company";
return result;
}
+ (OCMockObject *)mockSettingsProvider {
OCMockObject *result = [OCMockObject niceMockForClass:[GBApplicationSettingsProvider class]];
[[[result stub] andReturn:@"project"] projectName];
[[[result stub] andReturn:@"company"] projectCompany];
[[[result stub] andReturn:[GBCommentComponentsProvider provider]] commentComponents];
[[[result stub] andReturn:[GBApplicationStringsProvider provider]] stringTemplates];
return result;
}
+ (void)settingsProvider:(OCMockObject *)provider keepObjects:(BOOL)objects keepMembers:(BOOL)members {
[[[provider stub] andReturnValue:[NSNumber numberWithBool:objects]] keepUndocumentedObjects];
[[[provider stub] andReturnValue:[NSNumber numberWithBool:members]] keepUndocumentedMembers];
}
+ (void)registerComment:(id)comment forObject:(GBModelBase *)object {
object.comment = comment;
}
#pragma mark GBIvarData creation methods
+ (GBIvarData *)ivarWithComponents:(NSString *)first, ... {
va_list args;
va_start(args, first);
NSMutableArray *components = [NSMutableArray array];
for (NSString *argument=first; argument!=nil; argument=va_arg(args, NSString*)) {
[components addObject:argument];
}
va_end(args);
return [GBIvarData ivarDataWithComponents:components];
}
#pragma mark GBMethodData creation methods
+ (GBMethodData *)instanceMethodWithName:(NSString *)name comment:(id)comment {
GBMethodData *result = [self instanceMethodWithNames:name, nil];
[self registerComment:comment forObject:result];
return result;
}
+ (GBMethodData *)instanceMethodWithArguments:(GBMethodArgument *)first,... {
va_list args;
va_start(args, first);
NSMutableArray *arguments = [NSMutableArray array];
for (GBMethodArgument *argument=first; argument!=nil; argument=va_arg(args, GBMethodArgument*)) {
[arguments addObject:argument];
}
va_end(args);
return [GBMethodData methodDataWithType:GBMethodTypeInstance result:[NSArray arrayWithObject:@"void"] arguments:arguments];
}
+ (GBMethodData *)classMethodWithArguments:(GBMethodArgument *)first,... {
va_list args;
va_start(args, first);
NSMutableArray *arguments = [NSMutableArray array];
for (GBMethodArgument *argument=first; argument!=nil; argument=va_arg(args, GBMethodArgument*)) {
[arguments addObject:argument];
}
va_end(args);
return [GBMethodData methodDataWithType:GBMethodTypeClass result:[NSArray arrayWithObject:@"void"] arguments:arguments];
}
+ (GBMethodData *)instanceMethodWithNames:(NSString *)first,... {
va_list args;
va_start(args, first);
NSMutableArray *arguments = [NSMutableArray array];
for (NSString *name=first; name!=nil; name=va_arg(args, NSString*)) {
GBMethodArgument *argument = [self typedArgumentWithName:name];
[arguments addObject:argument];
}
va_end(args);
return [GBMethodData methodDataWithType:GBMethodTypeInstance result:[NSArray arrayWithObject:@"void"] arguments:arguments];
}
+ (GBMethodData *)classMethodWithNames:(NSString *)first,... {
va_list args;
va_start(args, first);
NSMutableArray *arguments = [NSMutableArray array];
for (NSString *name=first; name!=nil; name=va_arg(args, NSString*)) {
GBMethodArgument *argument = [self typedArgumentWithName:name];
[arguments addObject:argument];
}
va_end(args);
return [GBMethodData methodDataWithType:GBMethodTypeClass result:[NSArray arrayWithObject:@"void"] arguments:arguments];
}
+ (GBMethodData *)propertyMethodWithArgument:(NSString *)name {
GBMethodArgument *argument = [GBMethodArgument methodArgumentWithName:name];
return [GBMethodData methodDataWithType:GBMethodTypeProperty result:[NSArray arrayWithObject:@"int"] arguments:[NSArray arrayWithObject:argument]];
}
+ (GBMethodArgument *)typedArgumentWithName:(NSString *)name {
return [GBMethodArgument methodArgumentWithName:name types:[NSArray arrayWithObject:@"id"] var:name];
}
#pragma mark Store objects creation methods
+ (GBClassData *)classWithName:(NSString *)name methods:(GBMethodData *)first,... {
GBClassData *result = [GBClassData classDataWithName:name];
va_list args;
va_start(args, first);
for (GBMethodData *method=first; method!=nil; method=va_arg(args, GBMethodData*)) {
[result.methods registerMethod:method];
}
va_end(args);
return result;
}
+ (GBCategoryData *)categoryWithName:(NSString *)name className:(NSString *)className methods:(GBMethodData *)first,... {
GBCategoryData *result = [GBCategoryData categoryDataWithName:name className:className];
va_list args;
va_start(args, first);
for (GBMethodData *method=first; method!=nil; method=va_arg(args, GBMethodData*)) {
[result.methods registerMethod:method];
}
va_end(args);
return result;
}
+ (GBProtocolData *)protocolWithName:(NSString *)name methods:(GBMethodData *)first,... {
GBProtocolData *result = [GBProtocolData protocolDataWithName:name];
va_list args;
va_start(args, first);
for (GBMethodData *method=first; method!=nil; method=va_arg(args, GBMethodData*)) {
[result.methods registerMethod:method];
}
va_end(args);
return result;
}
#pragma mark GBStore creation methods
+ (GBStore *)store {
return [[[GBStore alloc] init] autorelease];
}
+ (GBStore *)storeWithClassWithName:(NSString *)name {
GBClassData *class = [GBClassData classDataWithName:name];
return [self storeByPerformingSelector:@selector(registerClass:) withObject:class];
}
+ (GBStore *)storeWithClassWithComment:(id)comment {
GBClassData *class = [GBClassData classDataWithName:@"Class"];
[self registerComment:comment forObject:class];
return [self storeByPerformingSelector:@selector(registerClass:) withObject:class];
}
+ (GBStore *)storeWithCategoryWithComment:(id)comment {
GBCategoryData *category = [GBCategoryData categoryDataWithName:@"Category" className:@"Class"];
[self registerComment:comment forObject:category];
return [self storeByPerformingSelector:@selector(registerCategory:) withObject:category];
}
+ (GBStore *)storeWithProtocolWithComment:(id)comment {
GBProtocolData *protocol = [GBProtocolData protocolDataWithName:@"Protocol"];
[self registerComment:comment forObject:protocol];
return [self storeByPerformingSelector:@selector(registerProtocol:) withObject:protocol];
}
+ (GBStore *)storeWithObjects:(id)first, ... {
GBStore *result = [self store];
va_list args;
va_start(args, first);
for (id object=first; object!=nil; object=va_arg(args, id)) {
if ([object isKindOfClass:[GBCategoryData class]])
[result registerCategory:object];
else if ([object isKindOfClass:[GBProtocolData class]])
[result registerProtocol:object];
else
[result registerClass:object];
}
va_end(args);
return result;
}
+ (GBStore *)storeByPerformingSelector:(SEL)selector withObject:(id)object {
GBStore *result = [self store];
[result performSelector:selector withObject:object];
return result;
}
@end