valueForKey: can be stubbed

allending · Dec 11, 2012 · 4dcbbc3 · 4dcbbc3
1 parent 183bba2
commit 4dcbbc3
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/Kiwi/KWMock.m b/Kiwi/KWMock.m
@@ -594,7 +594,17 @@ - (id)mutableCopy {
 #pragma mark Key-Value Coding Support
 
 - (id)valueForKey:(NSString *)key {
-    return nil;
+    KWMessagePattern *messagePattern = [KWMessagePattern messagePatternWithSelector:_cmd];
+    [self expectMessagePattern:messagePattern];
+    NSInvocation *invocation = [NSInvocation invocationWithTarget:self selector:_cmd messageArguments:&key];
+
+    if ([self processReceivedInvocation:invocation]) {
+        id result = nil;
+        [invocation getReturnValue:&result];
+        return result;
+    } else {
+        return nil;
+    }
 }
 
 - (void)setValue:(id)value forKey:(NSString *)keyPath {

diff --git a/Tests/KWMockTest.m b/Tests/KWMockTest.m
@@ -305,6 +305,14 @@ - (void)testItShouldNotRaiseWhenReceivingKVCMessagesAsANullMock {
     STAssertNoThrow([mock setValue:@"baz" forKeyPath:@"foo.bar"], @"expected setValue:forKeyPath: not to raise");
 }
 
+- (void)testItShouldAllowStubbingValueForKey {
+    id mock = [Cruiser mock];
+    id otherMock = [Cruiser mock];
+    [mock stub:@selector(valueForKey:) andReturn:otherMock withArguments:@"foo"];
+    id value = [mock valueForKey:@"foo"];
+    STAssertEquals(value, otherMock, @"expected valueForKey: to be stubbed");
+}
+
 @end
 
 #endif // #if KW_TESTS_ENABLED