Add basic descriptions for invocation actions.

Helps in debugging and error reporting.
erikdoe · Aug 13, 2020 · 5e01c36 · 5e01c36
1 parent ffeeed2
commit 5e01c36
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
@@ -53,5 +53,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