OCMInvocationExpectation.m 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  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. #import "OCMInvocationExpectation.h"
  17. #import "NSInvocation+OCMAdditions.h"
  18. @implementation OCMInvocationExpectation
  19. - (void)setMatchAndReject:(BOOL)flag
  20. {
  21. matchAndReject = flag;
  22. if(matchAndReject)
  23. isSatisfied = YES;
  24. }
  25. - (BOOL)isMatchAndReject
  26. {
  27. return matchAndReject;
  28. }
  29. - (BOOL)isSatisfied
  30. {
  31. return isSatisfied;
  32. }
  33. - (void)handleInvocation:(NSInvocation *)anInvocation
  34. {
  35. [super handleInvocation:anInvocation];
  36. if(matchAndReject)
  37. {
  38. isSatisfied = NO;
  39. [NSException raise:NSInternalInconsistencyException format:@"%@: explicitly disallowed method invoked: %@",
  40. [self description], [anInvocation invocationDescription]];
  41. }
  42. else
  43. {
  44. isSatisfied = YES;
  45. }
  46. }
  47. @end