实体框架模拟需要全局上下文

时间:2018-08-26 10:31:13

标签: c# entity-framework unit-testing entity-framework-6 dbcontext

我最近开始研究使用Entity Framework 6模拟进行的Entity Framework单元测试。

我注意到以下情况:

实体框架模拟迫使我在BL类中创建全局上下文,例如:

public class RefundRepayment : IDisposable
{
    protected DbContext _dbContext = new DbContext();

    /* more properties and class code */

    public void Dispose()
    {
        _dbContext.Dispose();
    }
}

我不太清楚,因为我宁愿在每种方法中实现using语句以处理DbContext,我的代码如下:

public class RefundRepayment
{
    /* more properties and class code */
    public void AccessDb() 
    {
        using(DbContext dbContext = new DbContext())
        {
            /* db code here */
        }
    }
}

有什么特殊的原因为什么我们应该初始化全局上下文而不是执行using语句?

1 个答案:

答案 0 :(得分:1)

首先,您需要使用DI(通过ninject,Unity,Core等)将其实现。

让我向您展示一个测试我的MVC控制器的EF GetAll()的简单示例。

[Fact]
public void GetAllOk()
{
    // Arrange

    // Act
    var result = _controller.GetAll() as OkObjectResult;

    // Assert
    Assert.NotNull(result);
    var recordList = result.Value as List<DTO.Account>;
    Assert.NotNull(recordList);
    Assert.Equal(4, recordList.Count);
}

它依赖于此启动代码...

public class AccountsControllerTests
{
    DatabaseFixture _fixture;
    AccountsControllerV1 _controller;

    public AccountsControllerTests(DatabaseFixture fixture)
    {
        _fixture = fixture;
        _controller = new AccountsControllerV1(_fixture._uow);
    }

什么是DatabaseFixture?很高兴你问...

public class DatabaseFixture : IDisposable
{
    public ApplicationDbContext _context;
    public DbContextOptions<ApplicationDbContext> _options;
    public IUoW _uow;

    public DatabaseFixture()
    {
        var x = Directory.GetCurrentDirectory();
        var config = new ConfigurationBuilder()
            .AddJsonFile("appsettings.Tests.json", optional : true)
            .Build();

        _options = new DbContextOptionsBuilder<ApplicationDbContext>()
            .UseInMemoryDatabase(databaseName: "ProviderTests")
            .Options;

        _context = new ApplicationDbContext(_options);
        _context.Database.EnsureCreated();
        Initialize();

        _uow = new UoW(_context);
    }

    private void Initialize()
    {
        _context.Accounts.Add(new Entities.Account() { AccountNumber = "Number 1", AccountID = "", AccountUniqueID = "" });
        _context.Accounts.Add(new Entities.Account() { AccountNumber = "Number 2", AccountID = "", AccountUniqueID = "" });
        _context.Accounts.Add(new Entities.Account() { AccountNumber = "Number 3", AccountID = "", AccountUniqueID = "" });
        _context.Accounts.Add(new Entities.Account() { AccountNumber = "Number 4", AccountID = "", AccountUniqueID = "" });
        _context.SaveChanges();
    }

    public void Dispose()
    {
        // Clean Up
        _context.Database.EnsureDeleted();
    }
}

[CollectionDefinition("Database Collection")]
public class DatabaseCollection : ICollectionFixture<DatabaseFixture>
{
}

上面的代码中使用了一些定义。我使用了一个工作单元模式,其中包含对我所有EF存储库的引用。我将实体(数据库)类和DTO(数据传输对象)类分开。我在每次运行和/或测试开始时都对内存数据库进行了内存替换,并在每次运行和/或测试开始时对其进行了初始化,以便我的数据始终是 known 。我将数据库夹具注入到我的测试类中(而不是每个测试中),因此我不会不断创建/销毁文件。然后创建传入数据库UoW定义的控制器。

您是 real 控制器,需要注入使用 real 数据库创建的UoW容器。您只是在为测试替代受控数据库环境。

public AccountsControllerV1(IUoW uow)
{
    _uow = uow;
}

是的,我对敏锐的眼睛使用版本控制。是的,这是一个Core 2示例。仍然适用于EF 6,只需要第三者DI;)

我正在测试的控制器方法是什么?

[HttpGet("accounts", Name ="GetAccounts")]
public IActionResult GetAll()
{
    try
    {
        var recordList = _uow.Accounts.GetAll();

        List<DTO.Account> results = new List<DTO.Account>();
        if (recordList != null)
        {
            results = recordList.Select(r => Map(r)).ToList();
        }

        log.Info($"Providers: GetAccounts: Success: {results.Count} records returned");
        return Ok(results);
    }
    catch (Exception ex)
    {
        log.Error($"Providers: GetAccounts: Failed: {ex.Message}");
        return BadRequest($"Providers: GetAccounts: Failed: {ex.Message}");
    }
}