Add basic descriptions for invocation actions.

Helps in debugging and error reporting.
erikdoe · Jul 14, 2020 · 2fef3c6 · 2fef3c6
1 parent b8746e5
commit 2fef3c6
Show file tree

Hide file tree

Showing 6 changed files with 28 additions and 0 deletions.
diff --git a/Source/OCMock/OCMBlockCaller.m b/Source/OCMock/OCMBlockCaller.m
@@ -43,4 +43,9 @@ - (void)handleInvocation:(NSInvocation *)anInvocation
     }
 }
 
+- (NSString *)description
+{
+    return [NSString stringWithFormat:@"[%@ - %p]: Calls `%@`", [self class], self, block];
+}
+
 @end
diff --git a/Source/OCMock/OCMExceptionReturnValueProvider.m b/Source/OCMock/OCMExceptionReturnValueProvider.m
@@ -27,4 +27,9 @@ - (void)handleInvocation:(NSInvocation *)anInvocation
 	[[NSException exceptionWithName:OCMStubbedException reason:@"Exception stubbed in test." userInfo:@{ @"exception": returnValue }] raise];
 }
 
+- (NSString *)description
+{
+  return [NSString stringWithFormat:@"[%@ - %p]: Throws `%@`", [self class], self, returnValue];
+}
+
 @end
diff --git a/Source/OCMock/OCMIndirectReturnValueProvider.m b/Source/OCMock/OCMIndirectReturnValueProvider.m
@@ -49,4 +49,9 @@ - (void)handleInvocation:(NSInvocation *)anInvocation
     [anInvocation setSelector:originalSelector];
 }
 
+- (NSString *)description
+{
+  return [NSString stringWithFormat:@"[%@ - %p]: Calls `%@` on `%@`", [self class], self, NSStringFromSelector(selector), provider];
+}
+
 @end
diff --git a/Source/OCMock/OCMNonRetainingObjectReturnValueProvider.m b/Source/OCMock/OCMNonRetainingObjectReturnValueProvider.m
@@ -48,5 +48,10 @@ - (void)handleInvocation:(NSInvocation *)anInvocation
     }
     [anInvocation setReturnValue:&returnValue];
 }
+
+- (NSString *)description
+{
+    return [NSString stringWithFormat:@"[%@ - %p]: Returns `%@`", [self class], self, returnValue];
+}
 @end
 
diff --git a/Source/OCMock/OCMNotificationPoster.m b/Source/OCMock/OCMNotificationPoster.m
@@ -40,5 +40,9 @@ - (void)handleInvocation:(NSInvocation *)anInvocation
 	[[NSNotificationCenter defaultCenter] postNotification:notification];
 }
 
+- (NSString *)description
+{
+  return [NSString stringWithFormat:@"[%@ - %p]: Posts `%@`", [self class], self, notification];
+}
 
 @end
diff --git a/Source/OCMock/OCMRealObjectForwarder.m b/Source/OCMock/OCMRealObjectForwarder.m
@@ -44,5 +44,9 @@ - (void)handleInvocation:(NSInvocation *)anInvocation
 	[anInvocation invoke];
 }
 
+- (NSString *)description
+{
+  return [NSString stringWithFormat:@"[%@ - %p]", [self class], self];
+}
 
 @end
-Original file line number
+Diff line change
@@ Expand Up / @@ -43,4 +43,9 @@ - (void)handleInvocation:(NSInvocation *)anInvocation @@
         }
     }
+    - (NSString *)description
+    {
+        return [NSString stringWithFormat:@"[%@ - %p]: Calls `%@`", [self class], self, block];
+    }
     @end