Еще нужно сделать проверку аргументов с которыми вызывался мок (см.доку)