No need for this when using MagicMock. Just set return value directly.