如何在单元测试中使用moq模拟Nhibernate .ToListAsync()？

public interface INhibernateSessionWrapper
{
    Task<List<T>> ToListAsync<T>(IQueryable<T> query);
}

public class NhibernateSessionWrapper : INhibernateSessionWrapper
{
    public async Task<List<T>> ToListAsync<T>(IQueryable<T> query)
    {
        // 在这里可以模拟ToListAsync方法的行为
        // 返回模拟的结果
        return new List<T>();
    }
}

[Test]
public async Task TestMethod()
{
    // 创建moq对象
    var mockSession = new Mock<INhibernateSessionWrapper>();

    // 设置ToListAsync方法的模拟行为
    mockSession.Setup(x => x.ToListAsync(It.IsAny<IQueryable<int>>()))
        .ReturnsAsync(new List<int> { 1, 2, 3 });

    // 调用需要测试的方法，并传入模拟的会话对象
    var result = await SomeClass.DoSomething(mockSession.Object);

    // 验证结果是否符合预期
    Assert.AreEqual(3, result.Count);
}