使用Moq模拟UnitOfWork/GenericRepository

public interface IUnitOfWork : IDisposable
{
    IRepository<TEntity> GetRepository<TEntity>() where TEntity : class;
    void SaveChanges();
}

public class UnitOfWork : IUnitOfWork
{
    private DbContext _context;
    
    public UnitOfWork(DbContext context)
    {
        _context = context;
    }
    
    public IRepository<TEntity> GetRepository<TEntity>() where TEntity : class
    {
        return new GenericRepository<TEntity>(_context);
    }
    
    public void SaveChanges()
    {
        _context.SaveChanges();
    }
    
    public void Dispose()
    {
        _context.Dispose();
    }
}

public interface IRepository<TEntity> where TEntity : class
{
    TEntity GetById(int id);
    void Add(TEntity entity);
    void Update(TEntity entity);
    void Delete(TEntity entity);
}

public class GenericRepository<TEntity> : IRepository<TEntity> where TEntity : class
{
    private DbContext _context;
    private DbSet<TEntity> _dbSet;
    
    public GenericRepository(DbContext context)
    {
        _context = context;
        _dbSet = context.Set<TEntity>();
    }
    
    public TEntity GetById(int id)
    {
        return _dbSet.Find(id);
    }
    
    public void Add(TEntity entity)
    {
        _dbSet.Add(entity);
    }
    
    public void Update(TEntity entity)
    {
        _dbSet.Attach(entity);
        _context.Entry(entity).State = EntityState.Modified;
    }
    
    public void Delete(TEntity entity)
    {
        _dbSet.Remove(entity);
    }
}

[Test]
public void TestUnitOfWork()
{
    // 创建Mock DbContext
    var dbContextMock = new Mock<DbContext>();
    
    // 创建UnitOfWork和GenericRepository
    var unitOfWork = new UnitOfWork(dbContextMock.Object);
    var repository = new GenericRepository<MyEntity>(dbContextMock.Object);
    
    // 设置Mock行为
    dbContextMock.Setup(d => d.Set<MyEntity>()).Returns(Mock.Of<DbSet<MyEntity>>());
    repository.Setup(r => r.GetById(It.IsAny<int>())).Returns(new MyEntity());
    
    // 执行测试代码
    var entity = repository.Object.GetById(1);
    
    // 断言结果
    Assert.IsNotNull(entity);
}