ReturnMethod

ReturnMethod

- (void)testReturnMethod_UsingClassMockAndStub {
    //GIVEN
    NSString* inputedValue = @"Simulated value";
    id objectMock = OCMClassMock([ClassB class]);
    OCMStub([objectMock executeReturnMethod]).andReturn(inputedValue);
    
    //WHEN /* run code under test */
    NSString* returnedValue = [objectMock executeReturnMethod];
    
    //THEN
    XCTAssertEqualObjects(inputedValue, returnedValue);
}
- (void)testReturnMethod_UsingPartialMockAndStub {
    //GIVEN
    NSString* inputedValue = @"Simulated value";
    ClassB* object = [[ClassB alloc] init];
    //You can comment these codes and run again.
    id objectMock = OCMPartialMock(object); //Real object
    OCMStub([objectMock executeReturnMethod]).andReturn(inputedValue);
    //End comment
    
    //WHEN /* run code under test */
    NSString* returnedValue = [object executeReturnMethod];
    
    //THEN
    XCTAssertEqualObjects(inputedValue, returnedValue);
}