尝试对使用模拟 Task<TResult> 的方法进行单元测试时出现问题

我正在尝试对使用 FCM 发送一些通知的方法进行单元测试。此方法采用批次列表，并为每个批次创建一个 IBatchResponse 任务，其中 IBatchResponse 是我制作的接口，因此我可以模拟实际的返回类型发送通知的服务。然后使用 Task.WhenAny() 我正在处理完成的任务。

 public static async Task<NotificationRes> SendPushNotifications(List<Batch> batches, INotificationService notificationService)
    {
        var notificationTasks = new List<Task<IBatchResponse>>();

        try
        {

            foreach (var batch in batches) 
            {
               notificationTasks.Add(notificationService.SendNotification(guardTokens, "android"));

            }
            while (notificationTasks.Count > 0)
            {
                var finishedTask = await Task.WhenAny(notificationTasks);
                var taskIndex = notificationTasks.FindIndex(task => task == finishedTask);
                notificationTasks.Remove(finishedTask);
                var finishedTaskResult = await finishedTask;
                for (int i = 0; i < finishedTaskResult.Responses.Count; i++)
                {
                                      ....
                }
               
        }
        catch (Exception)
        {

            throw;
        }

    }

这是我尝试进行的测试的代码。我模拟了发送通知的包装服务和notificationService.SendNotification (IBatchResponse) 的返回类型。

  [Test]
    public async Task SendPushNotifications_test()
    {
        ...
        var batchSendResponses = new List<ISendResponse>();
        
        ....
        var mockNotifService = new Mock<INotificationService>();
        var mockBatchRes = new Mock<IBatchResponse>();
        mockBatchRes.Setup(res => res.Responses).Returns(batchSendResponses);
        var task = Task.FromResult(mockBatchRes.Object);
        mockNotifService.Setup(ns => ns.SendNotification(new string[5] { "token1", "token2", "", "", "" }, "android")).Returns(task);


        var notRes = await Utils.SendPushNotifications(batches, mockNotifService.Object);

    }

问题是，如果存在 2 个批次，则完成的任务结果始终为空，但如果我有 1 个批次，则它可以工作。我在这里缺少什么。