.NET Core之單元測試（二）：使用內存數據庫處理單元測試中的數據庫依賴

目錄

• 定義一個待測試API
• 測試用例

為減少篇幅，隱藏了SampleEntity和SqliteDbContext

定義一個待測試API

如下，我們定義了一個名為Sample的API，其中有一個外部依賴項SqliteDbContext

[Route("api/[controller]")]
[ApiController]
public class SampleController : ControllerBase
{
    private readonly SqliteDbContext _sqliteDbContext;
    public SampleController(SqliteDbContext sqliteDbContext)
    {
        _sqliteDbContext = sqliteDbContext;
    }
    [HttpGet("{id}")]
    public async Task<IActionResult> Get(int id)
    {
        var sampleData = await _sqliteDbContext.SampleEntity
            .SingleOrDefaultAsync(s => s.Id == id);
        return Ok(sampleData);
    }
}

測試用例

本測試使用MSTest

Install-Package Microsoft.EntityFrameworkCore.InMemory -Version 3.1.2

private async Task<SqliteDbContext> GetSqliteDbContextAsync()
{
    var options = new DbContextOptionsBuilder<SqliteDbContext>()
        .UseInMemoryDatabase(Guid.NewGuid().ToString())
        .Options;
    var sqliteDbContext = new SqliteDbContext(options);
    sqliteDbContext.SampleEntity.Add(new Infrastructure.Entities.SampleEntity
    {
        Id = 1,
        BoolValue = false,
        DateTimeValue = DateTime.Now,
        StringValue = "sample"
    });
    await sqliteDbContext.SaveChangesAsync();
    return sqliteDbContext;
}

• 編寫測試方法

調用GetSqliteDbContextAsync()處理外部依賴項SqliteDbContext

[TestClass]
public class SampleControllerTest
{
    [TestMethod]
    public async Task Get_ReturnOK_WithExpectedParameters()
    {
        // Arrange
        var dbContext = await GetSqliteDbContextAsync();
        var controller = new SampleController(dbContext);

        // Act
        var response = await controller.Get(1);
        var responseModel = ((OkObjectResult)response).Value as SampleEntity;

        // Assert
        Assert.IsInstanceOfType(response, typeof(OkObjectResult));
        Assert.IsTrue(responseModel.Id == 1);
        Assert.IsFalse(responseModel.BoolValue);
        Assert.IsTrue(responseModel.StringValue == "sample");
    }
}

可以看到，測試已通過