TestInitialize上集成测试中多个上下文的一个事务

Question

我正在编写集成测试，我想使用事务范围。我们使用EF和带有上下文的存储库。

[TestInitialize]
public void RuleEngineTestsStart() {
    customContext = new CustomContext();
    transaction = customContext.Database.BeginTransaction();
    repo = new CustomRepository(customContext);

    // I need to make this context to work in the same transaction as above
    anotherContext = new AnotherContext();
    anotherRepo = new AnotherRepository(anotherContext);
}

编辑在评论中我被要求包括更多的代码，然而，我认为没有必要回答我的问题。

customContext = new CustomContext();
repo = new CustomRepository(customContext);

customContext2 = new CustomContext2();
otherRepository = new CustomRepository2(customContext2);

// class to be tested needs both repositories
ToBeTestedClass cl = new ToBeTestedClass(customRepository, otherRepository);

// "BASE" interface
public interface IRepository<TEntity> where TEntity : class
{
    TEntity GetById(long id);
    IEnumerable<TEntity> GetByFilter(Expression<Func<TEntity, bool>> predicate);
    TEntity GetSingleByFilter(Expression<Func<TEntity, bool>> filter);
    void Insert(TEntity entity);
    void Delete(long id);
    void Update(TEntity entity);
    ...
}

 // BASE CLASS
 public class Repository<TEntity> : IRepository<TEntity> where TEntity : class
{
    protected readonly DbContext _context;
    protected readonly DbSet<TEntity> _dbSet;

    public Repository(ColldeskDbContext context)
    {
        _context = context;
        _dbSet = context.Set<TEntity>();
    }

    // GetSingle, GetAll, Insert, Update etc.

  }

  // CustomRepository (other Repositories are similar, with custom methods)
  public interface ICustomRepository : IRepository<CusotmData>
   {
         // some specific methods that are not in Base class   
   }

  public class CustomRepository: Repository<CustomData>, ICustomRepository
  {
    public CustomRepository(CustomContext context) : base(context)
    {
    }
    // custom methods that are specific for given context
  }

  // Contexts - each context consists of its one DbSets

Answer 1

不要在存储库中使用dbcontext.saveChanges()。在创建存储库时使用一个dbContext。示例：

        using ( var db = new YourDbContext() )
        {
            // Create and begin transaction
            using ( var transaction = db.Database.BeginTransaction() )
            {
                try
                {
                    // ONE dbContext for all repositories
                    var firstRepo = new Custom1Repository(db);
                    var secondRepo = new Custom2Repository(db);

                    City city = new City { Description = "My city" };
                    Street street = new Street { Description = "My street", City = city};
                    firstRepo.Insert(city);
                    secondRepo.Insert(street);

                    // Save all your changes and after that commit transaction
                    db.SaveChanges();
                    transaction.Commit();
                }
                catch ( Exception ec)
                {
                    transaction.Rollback();
                }
            }
        }

这样做，您的存储库就变成了dbset 上的包装器