Spring Boot单元测试中StakeLimitService异常抛出问题排查

我正在开发一个包含Device、StakeLimit（关联Device外键）等实体的Spring Boot应用。编写StakeLimitServiceTest单元测试时，测试getStakeLimit等方法本不应抛出DeviceNotFoundException，但实际却触发了该自定义异常。尝试多种方式均未解决，仅ChatGPT提供的代码修改方案有效，但无法理解原因。此外，手动创建Device实例时，ID未自动生成UUID，需手动指定ID。

相关代码 #

Device实体 #

@Entity
public class Device {
    @Id
    @GeneratedValue(strategy = GenerationType.UUID)
    private String id;
    // 其他字段、构造方法及Getter/Setter
}

StakeLimit实体 #

@Entity
public class StakeLimit {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    
    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "device_id", nullable = false)
    private Device device;
    // 其他字段、构造方法及Getter/Setter
}

DeviceRepository #

public interface DeviceRepository extends JpaRepository<Device, String> {
}

StakeLimitRepository #

public interface StakeLimitRepository extends JpaRepository<StakeLimit, Long> {
    Optional<StakeLimit> findByDeviceId(String deviceId);
}

StakeLimitService #

@Service
public class StakeLimitService {
    private final StakeLimitRepository stakeLimitRepository;
    private final DeviceRepository deviceRepository;

    public StakeLimitService(StakeLimitRepository stakeLimitRepository, DeviceRepository deviceRepository) {
        this.stakeLimitRepository = stakeLimitRepository;
        this.deviceRepository = deviceRepository;
    }

    public StakeLimit getStakeLimit(String deviceId) {
        Device device = deviceRepository.findById(deviceId)
                .orElseThrow(() -> new DeviceNotFoundException("Device not found with id: " + deviceId));
        return stakeLimitRepository.findByDeviceId(deviceId)
                .orElseThrow(() -> new StakeLimitNotFoundException("Stake limit not found for device: " + deviceId));
    }
}

原测试代码（触发异常） #

@SpringBootTest
class StakeLimitServiceTest {
    @Autowired
    private StakeLimitService stakeLimitService;
    @Autowired
    private DeviceRepository deviceRepository;
    @Autowired
    private StakeLimitRepository stakeLimitRepository;

    @Test
    void getStakeLimit_ShouldReturnStakeLimit() {
        Device device = new Device();
        device.setId(UUID.randomUUID().toString()); // 手动指定ID
        deviceRepository.save(device);

        StakeLimit stakeLimit = new StakeLimit();
        stakeLimit.setDevice(device);
        stakeLimitRepository.save(stakeLimit);

        // 此处触发DeviceNotFoundException
        StakeLimit result = stakeLimitService.getStakeLimit(device.getId());
        assertNotNull(result);
    }
}

修改后测试代码（有效） #

@SpringBootTest
class StakeLimitServiceTest {
    @Autowired
    private StakeLimitService stakeLimitService;
    @Autowired
    private DeviceRepository deviceRepository;
    @Autowired
    private StakeLimitRepository stakeLimitRepository;

    @Test
    void getStakeLimit_ShouldReturnStakeLimit() {
        Device device = new Device();
        Device savedDevice = deviceRepository.save(device); // 接收持久化后的实例

        StakeLimit stakeLimit = new StakeLimit();
        stakeLimit.setDevice(savedDevice);
        stakeLimitRepository.save(stakeLimit);

        StakeLimit result = stakeLimitService.getStakeLimit(savedDevice.getId());
        assertNotNull(result);
    }
}

1. 单元测试触发DeviceNotFoundException的原因 #

原测试代码中，调用deviceRepository.save(device)后直接使用原device实例的ID，但此时原实例处于临时状态：JPA执行save操作时，会生成新的持久化实例并返回，原实例的ID字段并不会被自动更新。如果手动指定的ID存在格式问题，或者JPA未正确识别该ID的有效性，后续deviceRepository.findById就无法匹配到数据库中的记录，从而抛出DeviceNotFoundException。

修改后的代码通过接收save方法返回的持久化实例，确保使用的是数据库中实际存在的有效ID，因此不会触发异常。

2. Device ID未自动生成UUID的原因 #

• 实体注解问题：需确保Device的ID字段同时标注@Id和@GeneratedValue(strategy = GenerationType.UUID)，且字段类型为String。
• 实例状态问题：手动创建的Device实例是临时状态，只有调用repository.save()后，JPA才会生成UUID并赋值给返回的持久化实例，原临时实例的ID不会自动更新。
• 测试配置问题：若使用测试切片（如@DataJpaTest），需确保测试上下文正确加载JPA自动配置，否则UUID生成逻辑无法生效。

额外建议 #

• 始终使用repository.save()的返回值获取持久化实体，避免依赖原临时实例的字段。
• 测试中可添加@Transactional注解，确保测试结束后数据自动回滚，避免污染测试环境。
• 为Device的ID字段添加@Column(updatable = false)，防止后续操作意外修改ID值。

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