如何将DbContext实现为接口？最终目标是使用Moq进行单元测试

public interface IDbContext : IDisposable
{
    DbSet<TEntity> Set<TEntity>() where TEntity : class;
    int SaveChanges();
    Task<int> SaveChangesAsync();
    // 其他需要实现的方法和属性
}

public class MyDbContext : IDbContext
{
    private readonly DbContext _dbContext;

    public MyDbContext(DbContext dbContext)
    {
        _dbContext = dbContext;
    }

    public DbSet<TEntity> Set<TEntity>() where TEntity : class
    {
        return _dbContext.Set<TEntity>();
    }

    public int SaveChanges()
    {
        return _dbContext.SaveChanges();
    }

    public Task<int> SaveChangesAsync()
    {
        return _dbContext.SaveChangesAsync();
    }

    // 其他需要实现的方法和属性，委托给实际的DbContext对象
}

public class MyService
{
    private readonly IDbContext _dbContext;

    public MyService(IDbContext dbContext)
    {
        _dbContext = dbContext;
    }

    public void DoSomething()
    {
        // 使用IDbContext进行数据库操作
        var entities = _dbContext.Set<MyEntity>().ToList();
        // 其他操作
    }
}

[Test]
public void TestDoSomething()
{
    // 创建Mock对象
    var dbContextMock = new Mock<IDbContext>();
    var myService = new MyService(dbContextMock.Object);

    // 设置Mock对象的行为
    dbContextMock.Setup(m => m.Set<MyEntity>()).Returns(MockDbSet<MyEntity>(new List<MyEntity>()));

    // 执行测试
    myService.DoSomething();

    // 验证行为
    dbContextMock.Verify(m => m.Set<MyEntity>(), Times.Once);
    // 其他验证
}

// 辅助方法，用于创建模拟的DbSet对象
public static DbSet<T> MockDbSet<T>(List<T> data) where T : class
{
    var queryable = data.AsQueryable();
    var dbSetMock = new Mock<DbSet<T>>();
    dbSetMock.As<IQueryable<T>>().Setup(m => m.Provider).Returns(queryable.Provider);
    dbSetMock.As<IQueryable<T>>().Setup(m => m.Expression).Returns(queryable.Expression);
    dbSetMock.As<IQueryable<T>>().Setup(m => m.ElementType).Returns(queryable.ElementType);
    dbSetMock.As<IQueryable<T>>().Setup(m => m.GetEnumerator()).Returns(() => queryable.GetEnumerator());
    return dbSetMock.Object;
}