feat(phase1): 完成 JPA 实体类和 Repository 层实现
- 创建 3 个 JPA 实体类：DiagnosisRecord、CaseLibrary、ApiDocument - 创建 3 个 Repository 接口，实现基础 CRUD 和自定义查询方法 - 实现 DiagnosisRecordRepository 单元测试（6 个测试全部通过） - 修复 Hibernate schema 验证问题（枚举类型使用 VARCHAR） - 更新 application.yml，添加完整的数据库和 Redis 配置 - 更新 OpenSpec tasks.md，标记已完成任务（12/33） 测试结果： - DiagnosisRecordRepositoryTest: 6/6 通过 - 编译成功，无错误 Progress: 12/33 tasks completed
This commit is contained in:
zhuyongxin
@@ -0,0 +1,174 @@
|
||||
package org.example.repository;
|
||||
|
||||
import com.superbiz.agent.domain.enums.DiagnosisStatus;
|
||||
import com.superbiz.agent.domain.enums.FaultCategory;
|
||||
import org.example.domain.entity.DiagnosisRecord;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.test.autoconfigure.jdbc.AutoConfigureTestDatabase;
|
||||
import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
|
||||
import org.springframework.test.context.TestPropertySource;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.List;
|
||||
import java.util.Optional;
|
||||
import java.util.UUID;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.*;
|
||||
|
||||
/**
|
||||
* DiagnosisRecordRepository 单元测试
|
||||
*/
|
||||
@DataJpaTest
|
||||
@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.NONE)
|
||||
@TestPropertySource(properties = {
|
||||
"spring.flyway.enabled=true",
|
||||
"spring.jpa.hibernate.ddl-auto=validate",
|
||||
"spring.jpa.show-sql=true"
|
||||
})
|
||||
class DiagnosisRecordRepositoryTest {
|
||||
|
||||
@Autowired
|
||||
private DiagnosisRecordRepository repository;
|
||||
|
||||
@Test
|
||||
void testSaveAndFindById() {
|
||||
// 创建测试数据
|
||||
DiagnosisRecord record = DiagnosisRecord.builder()
|
||||
.diagnosisId(UUID.randomUUID().toString())
|
||||
.sessionId("session-001")
|
||||
.businessId("order-12345")
|
||||
.traceId("trace-abc123")
|
||||
.faultCategory(FaultCategory.EXTERNAL_API)
|
||||
.faultSource("广东")
|
||||
.faultTarget("http://api.example.com/query")
|
||||
.errorCode("40003")
|
||||
.errorMessage("接口超时")
|
||||
.status(DiagnosisStatus.SUCCESS)
|
||||
.confidence(85)
|
||||
.duration(1500)
|
||||
.build();
|
||||
|
||||
// 保存
|
||||
DiagnosisRecord saved = repository.save(record);
|
||||
assertNotNull(saved.getId());
|
||||
assertNotNull(saved.getCreatedAt());
|
||||
System.out.println("✓ 保存成功，ID: " + saved.getId());
|
||||
|
||||
// 查询
|
||||
Optional<DiagnosisRecord> found = repository.findById(saved.getId());
|
||||
assertTrue(found.isPresent());
|
||||
assertEquals("order-12345", found.get().getBusinessId());
|
||||
System.out.println("✓ 根据 ID 查询成功");
|
||||
}
|
||||
|
||||
@Test
|
||||
void testFindByDiagnosisId() {
|
||||
String diagnosisId = UUID.randomUUID().toString();
|
||||
DiagnosisRecord record = DiagnosisRecord.builder()
|
||||
.diagnosisId(diagnosisId)
|
||||
.businessId("order-test-001")
|
||||
.faultCategory(FaultCategory.DATABASE)
|
||||
.status(DiagnosisStatus.PENDING)
|
||||
.build();
|
||||
|
||||
repository.save(record);
|
||||
|
||||
Optional<DiagnosisRecord> found = repository.findByDiagnosisId(diagnosisId);
|
||||
assertTrue(found.isPresent());
|
||||
assertEquals(diagnosisId, found.get().getDiagnosisId());
|
||||
System.out.println("✓ 根据 diagnosisId 查询成功");
|
||||
}
|
||||
|
||||
@Test
|
||||
void testFindByFaultCategoryAndErrorCode() {
|
||||
// 创建测试数据
|
||||
DiagnosisRecord record1 = DiagnosisRecord.builder()
|
||||
.diagnosisId(UUID.randomUUID().toString())
|
||||
.faultCategory(FaultCategory.EXTERNAL_API)
|
||||
.errorCode("40003")
|
||||
.status(DiagnosisStatus.SUCCESS)
|
||||
.build();
|
||||
|
||||
DiagnosisRecord record2 = DiagnosisRecord.builder()
|
||||
.diagnosisId(UUID.randomUUID().toString())
|
||||
.faultCategory(FaultCategory.EXTERNAL_API)
|
||||
.errorCode("40003")
|
||||
.status(DiagnosisStatus.FAILED)
|
||||
.build();
|
||||
|
||||
repository.save(record1);
|
||||
repository.save(record2);
|
||||
|
||||
// 查询
|
||||
List<DiagnosisRecord> results = repository.findByFaultCategoryAndErrorCode(
|
||||
FaultCategory.EXTERNAL_API, "40003");
|
||||
|
||||
assertFalse(results.isEmpty());
|
||||
assertTrue(results.size() >= 2);
|
||||
System.out.println("✓ 根据故障类别和错误码查询成功，找到 " + results.size() + " 条记录");
|
||||
}
|
||||
|
||||
@Test
|
||||
void testFindByStatus() {
|
||||
DiagnosisRecord record = DiagnosisRecord.builder()
|
||||
.diagnosisId(UUID.randomUUID().toString())
|
||||
.status(DiagnosisStatus.RUNNING)
|
||||
.faultCategory(FaultCategory.CACHE)
|
||||
.build();
|
||||
|
||||
repository.save(record);
|
||||
|
||||
List<DiagnosisRecord> results = repository.findByStatus(DiagnosisStatus.RUNNING);
|
||||
assertFalse(results.isEmpty());
|
||||
System.out.println("✓ 根据状态查询成功，找到 " + results.size() + " 条 RUNNING 记录");
|
||||
}
|
||||
|
||||
@Test
|
||||
void testUpdateRecord() {
|
||||
// 创建并保存
|
||||
DiagnosisRecord record = DiagnosisRecord.builder()
|
||||
.diagnosisId(UUID.randomUUID().toString())
|
||||
.status(DiagnosisStatus.PENDING)
|
||||
.confidence(0)
|
||||
.build();
|
||||
|
||||
DiagnosisRecord saved = repository.save(record);
|
||||
Long id = saved.getId();
|
||||
|
||||
// 更新
|
||||
saved.setStatus(DiagnosisStatus.SUCCESS);
|
||||
saved.setConfidence(90);
|
||||
saved.setRootCause("接口超时导致");
|
||||
saved.setSolution("增加重试机制");
|
||||
|
||||
repository.save(saved);
|
||||
|
||||
// 验证更新
|
||||
Optional<DiagnosisRecord> updated = repository.findById(id);
|
||||
assertTrue(updated.isPresent());
|
||||
assertEquals(DiagnosisStatus.SUCCESS, updated.get().getStatus());
|
||||
assertEquals(90, updated.get().getConfidence());
|
||||
assertNotNull(updated.get().getUpdatedAt());
|
||||
System.out.println("✓ 更新记录成功");
|
||||
}
|
||||
|
||||
@Test
|
||||
void testDeleteRecord() {
|
||||
DiagnosisRecord record = DiagnosisRecord.builder()
|
||||
.diagnosisId(UUID.randomUUID().toString())
|
||||
.status(DiagnosisStatus.PENDING)
|
||||
.build();
|
||||
|
||||
DiagnosisRecord saved = repository.save(record);
|
||||
Long id = saved.getId();
|
||||
|
||||
// 删除
|
||||
repository.deleteById(id);
|
||||
|
||||
// 验证删除
|
||||
Optional<DiagnosisRecord> deleted = repository.findById(id);
|
||||
assertFalse(deleted.isPresent());
|
||||
System.out.println("✓ 删除记录成功");
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user