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 "OCMMacroState.h" |
| 18 #import "OCMStubRecorder.h" |
| 19 #import "OCMockObject.h" |
| 20 #import "OCMExpectationRecorder.h" |
| 21 #import "OCMVerifier.h" |
| 22 #import "OCMInvocationMatcher.h" |
| 23 |
| 24 |
| 25 @implementation OCMMacroState |
| 26 |
| 27 static OCMMacroState *globalState; |
| 28 |
| 29 #pragma mark Methods to begin/end macros |
| 30 |
| 31 + (void)beginStubMacro |
| 32 { |
| 33 OCMStubRecorder *recorder = [[[OCMStubRecorder alloc] init] autorelease]; |
| 34 globalState = [[[OCMMacroState alloc] initWithRecorder:recorder] autorelease
]; |
| 35 } |
| 36 |
| 37 + (OCMStubRecorder *)endStubMacro |
| 38 { |
| 39 OCMStubRecorder *recorder = (OCMStubRecorder *)[globalState recorder]; |
| 40 globalState = nil; |
| 41 return recorder; |
| 42 } |
| 43 |
| 44 |
| 45 + (void)beginExpectMacro |
| 46 { |
| 47 OCMExpectationRecorder *recorder = [[[OCMExpectationRecorder alloc] init] au
torelease]; |
| 48 globalState = [[[OCMMacroState alloc] initWithRecorder:recorder] autorelease
]; |
| 49 } |
| 50 |
| 51 + (OCMStubRecorder *)endExpectMacro |
| 52 { |
| 53 return [self endStubMacro]; |
| 54 } |
| 55 |
| 56 |
| 57 + (void)beginVerifyMacroAtLocation:(OCMLocation *)aLocation |
| 58 { |
| 59 OCMVerifier *recorder = [[[OCMVerifier alloc] init] autorelease]; |
| 60 [recorder setLocation:aLocation]; |
| 61 globalState = [[[OCMMacroState alloc] initWithRecorder:recorder] autorelease
]; |
| 62 } |
| 63 |
| 64 + (void)endVerifyMacro |
| 65 { |
| 66 globalState = nil; |
| 67 } |
| 68 |
| 69 |
| 70 #pragma mark Accessing global state |
| 71 |
| 72 + (OCMMacroState *)globalState |
| 73 { |
| 74 return globalState; |
| 75 } |
| 76 |
| 77 |
| 78 #pragma mark Init, dealloc, accessors |
| 79 |
| 80 - (id)initWithRecorder:(OCMRecorder *)aRecorder |
| 81 { |
| 82 if ((self = [super init])) |
| 83 { |
| 84 recorder = [aRecorder retain]; |
| 85 } |
| 86 |
| 87 return self; |
| 88 } |
| 89 |
| 90 - (void)dealloc |
| 91 { |
| 92 [recorder release]; |
| 93 if(globalState == self) |
| 94 globalState = nil; |
| 95 [super dealloc]; |
| 96 } |
| 97 |
| 98 - (OCMRecorder *)recorder |
| 99 { |
| 100 return recorder; |
| 101 } |
| 102 |
| 103 |
| 104 #pragma mark Changing the recorder |
| 105 |
| 106 - (void)switchToClassMethod |
| 107 { |
| 108 [recorder classMethod]; |
| 109 } |
| 110 |
| 111 |
| 112 @end |
OLD | NEW |