EF InMemory测试DbContext：初始化数据无法被测试访问的问题

这个问题我之前帮好几个做EF测试的开发者捋顺过，核心问题很明确：你的SetupData是void方法，初始化时创建的实体只存在于方法内部的局部作用域里，测试方法根本拿不到这些对象的引用。给你几个实用的解决方案，按需选就行：

1. 让SetupData返回初始化的实体 #

最直接的方式是把SetupData从void改成返回你需要的实体（或实体集合），这样测试方法调用后就能拿到引用。如果需要返回多个不同类型的实体，可以用元组（C# 7+支持）或者自定义一个初始化结果类。

示例代码：

// 改造SetupData，返回需要的实体引用
private (User testUser, Product testProduct) SetupTestData(MyDbContext context)
{
    var testUser = new User { Id = 1, UserName = "TestUser123" };
    var testProduct = new Product { Id = 100, ProductName = "Laptop" };
    
    context.Users.Add(testUser);
    context.Products.Add(testProduct);
    context.SaveChanges();
    
    return (testUser, testProduct);
}

// 在测试方法中调用
[Fact]
public void TestUserRetrieval()
{
    // 每个测试新建独立的DbContext，保证原子性
    using var context = CreateNewInMemoryDbContext();
    var (targetUser, _) = SetupTestData(context);
    
    // 直接用返回的实体引用做断言
    var retrievedUser = context.Users.Find(targetUser.Id);
    Assert.Equal(targetUser.UserName, retrievedUser.UserName);
}

// 辅助方法：创建全新的内存DbContext，每个测试用唯一库名避免污染
private MyDbContext CreateNewInMemoryDbContext()
{
    var options = new DbContextOptionsBuilder<MyDbContext>()
        .UseInMemoryDatabase(databaseName: Guid.NewGuid().ToString())
        .Options;
    return new MyDbContext(options);
}

2. 用测试类的字段存储初始化实体 #

如果多个测试方法需要用到同类型的初始化实体（但每个测试依然是独立的DbContext），可以在测试类中定义私有字段，让SetupData把创建好的实体赋值给这些字段。注意：每个测试前必须重新初始化字段和DbContext，避免测试间的状态污染。

示例代码：

public class MyDbContextIntegrationTests
{
    // 存储初始化后的实体引用
    private User _testUser;
    private Product _testProduct;

    [Fact]
    public void TestProductExists()
    {
        using var context = CreateNewInMemoryDbContext();
        SetupTestData(context);
        
        // 直接使用类字段里的实体引用
        var retrievedProduct = context.Products.Find(_testProduct.Id);
        Assert.NotNull(retrievedProduct);
    }

    private void SetupTestData(MyDbContext context)
    {
        // 每次调用都重新创建实体，避免测试污染
        _testUser = new User { Id = 2, UserName = "AnotherTestUser" };
        _testProduct = new Product { Id = 200, ProductName = "Phone" };
        
        context.Users.Add(_testUser);
        context.Products.Add(_testProduct);
        context.SaveChanges();
    }

    private MyDbContext CreateNewInMemoryDbContext()
    {
        var options = new DbContextOptionsBuilder<MyDbContext>()
            .UseInMemoryDatabase(databaseName: Guid.NewGuid().ToString())
            .Options;
        return new MyDbContext(options);
    }
}

3. 封装泛型辅助方法（进阶复用） #

如果你的测试需要频繁初始化不同类型的实体，可以写一个泛型辅助方法，负责添加实体到上下文并返回引用，让代码更简洁复用。

示例代码：

// 泛型辅助方法：添加实体并保存，返回实体引用
private T AddAndSaveEntity<T>(MyDbContext context, T entity) where T : class
{
    context.Set<T>().Add(entity);
    context.SaveChanges();
    return entity;
}

// 测试方法中使用
[Fact]
public void TestOrderCreation()
{
    using var context = CreateNewInMemoryDbContext();
    // 直接创建并获取实体引用
    var testOrder = AddAndSaveEntity(context, new Order { OrderNumber = "ORD-001" });
    
    var retrievedOrder = context.Orders.Find(testOrder.Id);
    Assert.Equal(testOrder.OrderNumber, retrievedOrder.OrderNumber);
}

关键注意点 #

不管用哪种方案，一定要给每个测试的DbContext分配唯一的内存数据库名称（比如用Guid.NewGuid().ToString()），否则多个测试会共享同一个内存数据库，破坏测试的原子性，导致奇怪的测试失败。

内容的提问来源于stack exchange，提问作者SuperJMN