| OLD | NEW |
| (Empty) | |
| 1 /* |
| 2 * Copyright (c) 2014-2015 Erik Doernenburg and contributors |
| 3 * |
| 4 * Licensed under the Apache License, Version 2.0 (the "License"); you may |
| 5 * not use these files except in compliance with the License. You may obtain |
| 6 * a copy of the License at |
| 7 * |
| 8 * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 * |
| 10 * Unless required by applicable law or agreed to in writing, software |
| 11 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| 12 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| 13 * License for the specific language governing permissions and limitations |
| 14 * under the License. |
| 15 */ |
| 16 |
| 17 #import "OCMInvocationExpectation.h" |
| 18 #import "NSInvocation+OCMAdditions.h" |
| 19 |
| 20 @implementation OCMInvocationExpectation |
| 21 |
| 22 - (void)setMatchAndReject:(BOOL)flag { |
| 23 matchAndReject = flag; |
| 24 if (matchAndReject) |
| 25 isSatisfied = YES; |
| 26 } |
| 27 |
| 28 - (BOOL)isMatchAndReject { |
| 29 return matchAndReject; |
| 30 } |
| 31 |
| 32 - (BOOL)isSatisfied { |
| 33 return isSatisfied; |
| 34 } |
| 35 |
| 36 - (void)handleInvocation:(NSInvocation*)anInvocation { |
| 37 [super handleInvocation:anInvocation]; |
| 38 |
| 39 if (matchAndReject) { |
| 40 isSatisfied = NO; |
| 41 [NSException |
| 42 raise:NSInternalInconsistencyException |
| 43 format:@"%@: explicitly disallowed method invoked: %@", |
| 44 [self description], [anInvocation invocationDescription]]; |
| 45 } else { |
| 46 isSatisfied = YES; |
| 47 } |
| 48 } |
| 49 |
| 50 @end |
| OLD | NEW |