Tests for primitive argument matching of spied methods

BennettSmith · Feb 23, 2013 · a10a769 · a10a769
1 parent 220f01b
commit a10a769
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/Source/Tests/MKTObjectMockSpyTest.m b/Source/Tests/MKTObjectMockSpyTest.m
@@ -47,6 +47,11 @@ - (NSString *)methodWithArg:(NSString *)arg
     return @"unstubbed";
 }
 
+- (NSString *)methodWithInt:(int)arg
+{
+    return @"unstubbed";
+}
+
 @end
 
 
@@ -140,4 +145,14 @@ - (void)testWillReturnValuesMatchingAnyArgument
     assertThat([sut methodWithArg:@"bar"], is(@"foo"));
 }
 
+- (void)testWillReturnValuesMatchingPrimitiveArgument
+{
+    // given
+    [[willReturn(@"foo") when:sut] methodWithInt:0];
+
+    // then
+    assertThat([sut methodWithInt:0], is(@"foo"));
+    assertThat([sut methodWithInt:1], is(@"unstubbed"));
+}
+
 @end