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
|
/*
* Copyright 2017 Google
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#import "OCMStubRecorder+FIRAuthUnitTests.h"
#import "FIRAuthGlobalWorkQueue.h"
/** @fn argumentOf
@brief Retrieves a specific argument from a method invocation.
@param invocation The Objective-C method invocation.
@param position The position of the argument to retrieve, starting from 0.
@return The argument at the given position that the method has been invoked with.
@remarks The argument type must be compatible with @c id .
*/
static id argumentOf(NSInvocation *invocation, int position) {
__unsafe_unretained id unretainedArgument;
// Indices 0 and 1 indicate the hidden arguments self and _cmd. Actual arguments starts from 2.
[invocation getArgument:&unretainedArgument atIndex:position + 2];
// The argument needs to be retained, or it will be released along with the invocation object.
id argument = unretainedArgument;
return argument;
}
/** @fn doubleArgumentOf
@brief Retrieves a specific argument of type 'double' from a method invocation.
@param invocation The Objective-C method invocation.
@param position The position of the argument to retrieve, starting from 0.
@return The argument at the given position that the method has been invoked with.
@remarks The argument type must be @c double .
*/
static double doubleArgumentOf(NSInvocation *invocation, int position) {
double argument;
// Indices 0 and 1 indicate the hidden arguments self and _cmd. Actual arguments starts from 2.
[invocation getArgument:&argument atIndex:position + 2];
return argument;
}
@implementation OCMStubRecorder (FIRAuthUnitTests)
- (id)andCallBlock1:(FIRAuthGeneralBlock1)block1 {
return [self andDo:^(NSInvocation *invocation) {
block1(argumentOf(invocation, 0));
}];
}
- (id)andCallBlock2:(FIRAuthGeneralBlock2)block2 {
return [self andDo:^(NSInvocation *invocation) {
block2(argumentOf(invocation, 0), argumentOf(invocation, 1));
}];
}
- (id)andDispatchError2:(NSError *)error {
return [self andCallBlock2:^(id request, FIRAuthGeneralBlock2 callback) {
dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
callback(nil, error);
});
}];
}
- (id)andCallIdDoubleIdBlock:(FIRAuthIdDoubleIdBlock)block {
return [self andDo:^(NSInvocation *invocation) {
block(argumentOf(invocation, 0), doubleArgumentOf(invocation, 2), argumentOf(invocation, 2));
}];
}
- (OCMStubRecorder *(^)(FIRAuthGeneralBlock1))_andCallBlock1 {
return ^(FIRAuthGeneralBlock1 block1) {
return [self andCallBlock1:block1];
};
}
- (OCMStubRecorder *(^)(FIRAuthGeneralBlock2))_andCallBlock2 {
return ^(FIRAuthGeneralBlock2 block2) {
return [self andCallBlock2:block2];
};
}
- (OCMStubRecorder *(^)(NSError *))_andDispatchError2 {
return ^(NSError *error) {
return [self andDispatchError2:error];
};
}
- (OCMStubRecorder *(^)(FIRAuthIdDoubleIdBlock))_andCallIdDoubleIdBlock {
return ^(FIRAuthIdDoubleIdBlock block) {
return [self andCallIdDoubleIdBlock:block];
};
}
@end
|