OCMExpectationRecorder.m 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. /*
  2. * Copyright (c) 2004-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. #import "OCMExpectationRecorder.h"
  17. #import "OCMInvocationExpectation.h"
  18. @implementation OCMExpectationRecorder
  19. #pragma mark Initialisers, description, accessors, etc.
  20. - (id)init
  21. {
  22. self = [super init];
  23. [invocationMatcher release];
  24. invocationMatcher = [[OCMInvocationExpectation alloc] init];
  25. return self;
  26. }
  27. - (OCMInvocationExpectation *)expectation
  28. {
  29. return (OCMInvocationExpectation *)invocationMatcher;
  30. }
  31. #pragma mark Modifying the expectation
  32. - (id)never
  33. {
  34. [[self expectation] setMatchAndReject:YES];
  35. return self;
  36. }
  37. #pragma mark Finishing recording
  38. - (void)forwardInvocation:(NSInvocation *)anInvocation
  39. {
  40. [super forwardInvocation:anInvocation];
  41. [mockObject addExpectation:[self expectation]];
  42. }
  43. @end