Add OCMock for stub method