如何模拟DbContext以在存储库的单元测试中使用？

// 定义IDbContext接口
public interface IDbContext
{
    DbSet<TEntity> Set<TEntity>() where TEntity : class;
    int SaveChanges();
    // 其他与实际DbContext相同的方法和属性
}

// 创建FakeDbContext类，实现IDbContext接口
public class FakeDbContext : IDbContext
{
    private Dictionary<Type, object> _dataSets;

    public FakeDbContext()
    {
        _dataSets = new Dictionary<Type, object>();
    }

    public DbSet<TEntity> Set<TEntity>() where TEntity : class
    {
        if (!_dataSets.ContainsKey(typeof(TEntity)))
        {
            _dataSets[typeof(TEntity)] = new List<TEntity>();
        }

        return new FakeDbSet<TEntity>((List<TEntity>)_dataSets[typeof(TEntity)]);
    }

    public int SaveChanges()
    {
        // 不需要实现具体的保存逻辑，因为这里是模拟的内存数据结构
        return 0;
    }
    // 其他与实际DbContext相同的方法和属性的实现
}

// 创建FakeDbSet类，用于模拟DbSet
public class FakeDbSet<TEntity> : DbSet<TEntity> where TEntity : class
{
    private List<TEntity> _data;

    public FakeDbSet(List<TEntity> data)
    {
        _data = data;
    }

    public override TEntity Add(TEntity entity)
    {
        _data.Add(entity);
        return entity;
    }

    public override TEntity Remove(TEntity entity)
    {
        _data.Remove(entity);
        return entity;
    }

    // 其他与实际DbSet相同的方法和属性的实现
}

// 在存储库的单元测试中使用FakeDbContext
[TestClass]
public class RepositoryTests
{
    private IRepository _repository;
    private IDbContext _dbContext;

    [TestInitialize]
    public void Initialize()
    {
        _dbContext = new FakeDbContext();
        _repository = new Repository(_dbContext);
    }

    [TestMethod]
    public void Add_Entity_Should_Add_To_DbContext()
    {
        // 创建要测试的实体
        var entity = new Entity();

        // 调用存储库的添加方法
        _repository.Add(entity);

        // 从FakeDbContext中获取实体，验证是否添加成功
        var entities = _dbContext.Set<Entity>().ToList();
        Assert.IsTrue(entities.Contains(entity));
    }

    // 其他存储库的单元测试方法
}