[Refactor] 페이지네이션 1-based로 수정 및 시나리오 initãdata생성

johnbosco0414 · johnbosco0414 · commit f69981765392 · 2025-10-01T17:04:57.000+09:00
diff --git a/back/src/main/java/com/back/domain/scenario/controller/ScenarioController.java b/back/src/main/java/com/back/domain/scenario/controller/ScenarioController.java
@@ -2,13 +2,13 @@
 
 import com.back.domain.scenario.dto.*;
 import com.back.domain.scenario.service.ScenarioService;
+import com.back.global.common.PageResponse;
 import com.back.global.security.CustomUserDetails;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.Parameter;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import jakarta.validation.Valid;
 import lombok.RequiredArgsConstructor;
-import org.springframework.data.domain.Page;
 import org.springframework.data.domain.Pageable;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
@@ -92,14 +92,14 @@ public ResponseEntity<TimelineResponse> getScenarioTimeline(
     }
 
     @GetMapping("/baselines")
-    @Operation(summary = "베이스라인 목록 조회", description = "사용자의 베이스라인 목록을 페이지네이션으로 조회합니다.")
-    public ResponseEntity<Page<BaselineListResponse>> getBaselines(
+    @Operation(summary = "베이스라인 목록 조회", description = "사용자의 베이스라인 목록을 페이지네이션으로 조회합니다. (1-based 페이지네이션)")
+    public ResponseEntity<PageResponse<BaselineListResponse>> getBaselines(
             @AuthenticationPrincipal CustomUserDetails userDetails,
             Pageable pageable
     ) {
         Long userId = getUserId(userDetails);
 
-        Page<BaselineListResponse> baselines = scenarioService.getBaselines(userId, pageable);
+        PageResponse<BaselineListResponse> baselines = scenarioService.getBaselines(userId, pageable);
 
         return ResponseEntity.ok(baselines);
     }
diff --git a/back/src/main/java/com/back/domain/scenario/entity/Scenario.java b/back/src/main/java/com/back/domain/scenario/entity/Scenario.java
@@ -34,9 +34,9 @@ public class Scenario extends BaseEntity {
     @JoinColumn(name = "decision_line_id", unique = true)
     private DecisionLine decisionLine;
 
-    // 시나리오 비교 분석 대상 베이스 시나리오 (선택 경로의 베이스라인과 연결)
-    @OneToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "base_line_id", unique = true)
+    // 시나리오가 속한 베이스라인 (하나의 BaseLine에 여러 Scenario 가능)
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "base_line_id", nullable = false)
     private BaseLine baseLine;
 
     // 시나리오 처리 상태 (PENDING, PROCESSING, COMPLETED, FAILED)
diff --git a/back/src/main/java/com/back/domain/scenario/service/ScenarioService.java b/back/src/main/java/com/back/domain/scenario/service/ScenarioService.java
@@ -15,6 +15,7 @@
 import com.back.global.ai.dto.result.BaseScenarioResult;
 import com.back.global.ai.dto.result.DecisionScenarioResult;
 import com.back.global.ai.service.AiService;
+import com.back.global.common.PageResponse;
 import com.back.global.exception.ApiException;
 import com.back.global.exception.ErrorCode;
 import com.fasterxml.jackson.core.type.TypeReference;
@@ -104,9 +105,13 @@ public ScenarioStatusResponse createScenario(Long userId, ScenarioCreateRequest
 
         // 새 시나리오 생성 (DataIntegrityViolationException 처리)
         try {
+            // DecisionLine에서 BaseLine 가져오기
+            BaseLine baseLine = decisionLine.getBaseLine();
+
             Scenario scenario = Scenario.builder()
                     .user(decisionLine.getUser())
                     .decisionLine(decisionLine)
+                    .baseLine(baseLine)  // DecisionLine의 BaseLine 연결
                     .status(ScenarioStatus.PENDING)
                     .build();
 
@@ -327,12 +332,15 @@ public ScenarioCompareResponse compareScenarios(Long baseId, Long compareId, Lon
 
     // 베이스라인 목록 조회 (페이지네이션 지원)
     @Transactional(readOnly = true)
-    public Page<BaselineListResponse> getBaselines(Long userId, Pageable pageable) {
+    public PageResponse<BaselineListResponse> getBaselines(Long userId, Pageable pageable) {
         // 사용자별 베이스라인 조회 (BaseNode들과 함께 fetch)
         Page<BaseLine> baseLines = baseLineRepository.findAllByUserIdWithBaseNodes(userId, pageable);
 
         // BaseLine -> BaselineListResponse 변환
-        return baseLines.map(this::convertToBaselineListResponse);
+        Page<BaselineListResponse> responsePage = baseLines.map(this::convertToBaselineListResponse);
+
+        // PageResponse로 변환 (1-based 페이지네이션)
+        return PageResponse.of(responsePage);
     }
 
     /**
diff --git a/back/src/main/java/com/back/global/initdata/InitData.java b/back/src/main/java/com/back/global/initdata/InitData.java
@@ -8,11 +8,21 @@
 import com.back.domain.node.dto.decision.DecNodeDto;
 import com.back.domain.node.dto.decision.DecisionNodeFromBaseRequest;
 import com.back.domain.node.dto.decision.DecisionNodeNextRequest;
+import com.back.domain.node.entity.BaseLine;
+import com.back.domain.node.entity.DecisionLine;
 import com.back.domain.node.entity.NodeCategory;
+import com.back.domain.node.repository.BaseLineRepository;
+import com.back.domain.node.repository.DecisionLineRepository;
 import com.back.domain.node.service.NodeService;
 import com.back.domain.post.entity.Post;
 import com.back.domain.post.enums.PostCategory;
 import com.back.domain.post.repository.PostRepository;
+import com.back.domain.scenario.entity.SceneType;
+import com.back.domain.scenario.entity.Scenario;
+import com.back.domain.scenario.entity.ScenarioStatus;
+import com.back.domain.scenario.entity.Type;
+import com.back.domain.scenario.repository.SceneTypeRepository;
+import com.back.domain.scenario.repository.ScenarioRepository;
 import com.back.domain.user.entity.Gender;
 import com.back.domain.user.entity.Mbti;
 import com.back.domain.user.entity.Role;
@@ -29,8 +39,9 @@
 import java.util.UUID;
 
 /**
- * [요약] 기동 시 admin·user1 생성 → user1에 베이스라인(총7: 헤더+피벗5+테일) 1개와 결정라인(총5 노드) 1개 시드 주입.
+ * [요약] 기동 시 admin·user1 생성 → user1에 베이스라인(총7: 헤더+피벗5+테일) 1개와 결정라인 2개(첫 번째 6노드, 두 번째 2노드) 시드 주입.
  *        게시글 30개(일반20 + 투표10)과 댓글 14개(마지막 2개 글에 각 7개) 생성.
+ *        시나리오 3개(베이스 1개 + 완성 1개 + 처리중 1개)와 지표 10개(완성된 시나리오 2개에 각 5개씩) 생성.
  */
 @Component
 @RequiredArgsConstructor
@@ -43,6 +54,11 @@ public class InitData implements CommandLineRunner {
     private final PostRepository postRepository;
     private final CommentRepository commentRepository;
 
+    private final BaseLineRepository baseLineRepository;
+    private final DecisionLineRepository decisionLineRepository;
+    private final ScenarioRepository scenarioRepository;
+    private final SceneTypeRepository sceneTypeRepository;
+
     // user1을 만들고 베이스라인(7)과 결정라인(5)을 시드로 주입한다
     @Override
     public void run(String... args) {
@@ -161,7 +177,7 @@ public void run(String... args) {
                 )
         );
 
-        nodeService.createDecisionNodeNext(
+        DecNodeDto d4 = nodeService.createDecisionNodeNext(
                 new DecisionNodeNextRequest(
                         user1.getId(),
                         d3.id(),
@@ -175,6 +191,36 @@ public void run(String... args) {
                 )
         );
 
+        // 두 번째 DecisionLine 생성 (processingScenario 테스트용)
+        DecNodeDto d5 = nodeService.createDecisionNodeFromBase(
+                new DecisionNodeFromBaseRequest(
+                        user1.getId(),
+                        baseLineId,
+                        1,  // 두 번째 피벗
+                        null,
+                        0,
+                        NodeCategory.CAREER,
+                        "스타트업 창업",
+                        List.of("단독 창업", "공동 창업"),
+                        0,
+                        "기술 스타트업 설립 선택지"
+                )
+        );
+
+        DecNodeDto d6 = nodeService.createDecisionNodeNext(
+                new DecisionNodeNextRequest(
+                        user1.getId(),
+                        d5.id(),
+                        NodeCategory.FINANCE,
+                        "초기 투자 유치",
+                        null,
+                        List.of("엔젤 투자", "시드 투자"),
+                        0,
+                        0,
+                        "초기 자금 확보 전략"
+                )
+        );
+
         if (postRepository.count() > 0) {
             return;
         }
@@ -238,5 +284,148 @@ public void run(String... args) {
             }
         }
         commentRepository.saveAll(comments);
+
+        // ========== Scenario InitData 생성 ==========
+
+        // 시나리오 데이터가 이미 있으면 스킵
+        if (scenarioRepository.count() > 0) {
+            return;
+        }
+
+        // BaseLine 조회 (위에서 생성한 baseLineId 사용)
+        BaseLine baseLine = baseLineRepository.findById(baseLineId)
+                .orElseThrow(() -> new IllegalStateException("BaseLine not found"));
+
+        // DecisionLine 조회 (user1의 DecisionLine - BaseLine에 연결된 것)
+        List<DecisionLine> decisionLines = decisionLineRepository.findByBaseLine_Id(baseLineId);
+        if (decisionLines.size() < 2) {
+            throw new IllegalStateException("Expected 2 DecisionLines but found: " + decisionLines.size());
+        }
+
+        DecisionLine decisionLine1 = decisionLines.get(0);  // 첫 번째 DecisionLine (완성 시나리오용)
+        DecisionLine decisionLine2 = decisionLines.get(1);  // 두 번째 DecisionLine (처리중 시나리오용)
+
+        // 1. 베이스 시나리오 생성 (현재 삶 기준)
+        Scenario baseScenario = Scenario.builder()
+                .user(user1)
+                .baseLine(baseLine)
+                .decisionLine(null)  // 베이스 시나리오는 DecisionLine 없음
+                .status(ScenarioStatus.COMPLETED)
+                .job("주니어 백엔드 개발자")
+                .total(350)
+                .summary("현재 커리어 경로를 유지하며 안정적으로 성장하는 시나리오입니다.")
+                .description("""
+                        대학 졸업 후 백엔드 개발자로 시작하여 주니어에서 시니어로 성장하는 과정입니다.
+                        안정적인 중견기업에서 경력을 쌓으며, 스프링 부트와 API 설계 전문성을 키워나갑니다.
+                        팀 내에서 신뢰받는 개발자로 성장하며, 워라밸을 유지하면서 건강한 개발 생활을 이어갑니다.
+                        """)
+                .timelineTitles("""
+                        [
+                          {"year":2025,"title":"시니어 개발자 승진"},
+                          {"year":2027,"title":"테크 리드 역할 수행"},
+                          {"year":2030,"title":"개발팀 리더"}
+                        ]
+                        """)
+                .img("https://picsum.photos/seed/base-scenario/400/300")
+                .build();
+        scenarioRepository.save(baseScenario);
+
+        // 베이스 시나리오 지표 생성 (5개)
+        createSceneTypes(baseScenario, 70, 75, 70, 65, 70);
+
+        // 2. 완성된 시나리오 생성 (첫 번째 DecisionLine 기반)
+        Scenario completedScenario = Scenario.builder()
+                .user(user1)
+                .baseLine(baseLine)  // 베이스 시나리오와 같은 BaseLine 참조
+                .decisionLine(decisionLine1)
+                .status(ScenarioStatus.COMPLETED)
+                .job("클라우드 아키텍트")
+                .total(430)
+                .summary("클라우드와 보안 전문성을 갖춘 시니어 아키텍트로 성장하는 시나리오입니다.")
+                .description("""
+                        AWS/GCP 클라우드 플랫폼 심화 학습과 보안 인증을 통해 시스템 아키텍트로 성장합니다.
+                        대용량 트래픽 처리 경험과 인프라 자동화 능력을 갖추어 기술 리더로 인정받습니다.
+                        컨설팅 프로젝트와 기술 강연을 통해 업계 전문가로 자리매김하며,
+                        궁극적으로 스타트업 CTO 또는 대기업 기술 이사로 성장합니다.
+                        """)
+                .timelineTitles("""
+                        [
+                          {"year":2025,"title":"AWS Solutions Architect 자격증 취득"},
+                          {"year":2026,"title":"보안 전문가 인증 (CISSP)"},
+                          {"year":2028,"title":"솔루션 아키텍트 승진"},
+                          {"year":2031,"title":"기술 이사 (CTO)"}
+                        ]
+                        """)
+                .img("https://picsum.photos/seed/decision-scenario/400/300")
+                .build();
+        scenarioRepository.save(completedScenario);
+
+        // 완성 시나리오 지표 생성 (더 높은 점수)
+        createSceneTypes(completedScenario, 90, 85, 80, 88, 87);
+
+        // 3. 처리 중 시나리오 생성 (두 번째 DecisionLine 기반, 폴링 테스트용)
+        Scenario processingScenario = Scenario.builder()
+                .user(user1)
+                .baseLine(baseLine)
+                .decisionLine(decisionLine2)  // 두 번째 DecisionLine 사용
+                .status(ScenarioStatus.PROCESSING)
+                .build();
+        scenarioRepository.save(processingScenario);
+    }
+
+    /**
+     * 시나리오에 대한 5개 지표(SceneType) 데이터를 생성합니다.
+     * @param scenario 대상 시나리오
+     * @param eco 경제 점수
+     * @param happy 행복 점수
+     * @param rel 관계 점수
+     * @param career 직업 점수
+     * @param health 건강 점수
+     */
+    private void createSceneTypes(Scenario scenario, int eco, int happy, int rel, int career, int health) {
+        List<SceneType> sceneTypes = List.of(
+                SceneType.builder()
+                        .scenario(scenario)
+                        .type(Type.경제)
+                        .point(eco)
+                        .analysis(eco >= 85
+                                ? "클라우드 전문가로 높은 연봉과 컨설팅 수입을 통해 경제적 자유를 확보했습니다."
+                                : "안정적인 중견기업 재직으로 평균 이상의 경제력을 유지하고 있습니다.")
+                        .build(),
+                SceneType.builder()
+                        .scenario(scenario)
+                        .type(Type.행복)
+                        .point(happy)
+                        .analysis(happy >= 80
+                                ? "전문성 인정과 도전적인 업무를 통해 높은 직무 만족도를 느낍니다."
+                                : "업무 만족도가 높고 워라밸이 좋은 환경에서 일하고 있습니다.")
+                        .build(),
+                SceneType.builder()
+                        .scenario(scenario)
+                        .type(Type.관계)
+                        .point(rel)
+                        .analysis(rel >= 80
+                                ? "리더십 역할을 통해 업계 네트워크를 넓히고 멘토 관계를 형성했습니다."
+                                : "팀원들과 원만한 관계를 유지하며 개인 시간도 충분히 확보하고 있습니다.")
+                        .build(),
+                SceneType.builder()
+                        .scenario(scenario)
+                        .type(Type.직업)
+                        .point(career)
+                        .analysis(career >= 85
+                                ? "클라우드 및 보안 분야 최고 전문가로 인정받고 있습니다."
+                                : "백엔드 개발 전문성은 확보했으나 리더십 경험이 다소 부족합니다.")
+                        .build(),
+                SceneType.builder()
+                        .scenario(scenario)
+                        .type(Type.건강)
+                        .point(health)
+                        .analysis(health >= 85
+                                ? "체계적인 건강 관리와 규칙적인 운동 루틴을 유지하고 있습니다."
+                                : "규칙적인 생활과 적당한 운동으로 건강을 유지하고 있습니다.")
+                        .build()
+        );
+
+        sceneTypeRepository.saveAll(sceneTypes);
     }
 }
diff --git a/back/src/test/java/com/back/domain/scenario/controller/ScenarioControllerTest.java b/back/src/test/java/com/back/domain/scenario/controller/ScenarioControllerTest.java
@@ -15,6 +15,7 @@
 import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
 import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.boot.test.mock.mockito.MockBean;
+import com.back.global.common.PageResponse;
 import org.springframework.data.domain.Page;
 import org.springframework.data.domain.PageImpl;
 import org.springframework.data.domain.PageRequest;
@@ -263,7 +264,8 @@ class GetBaselines {
                     )
             );
 
-            Page<BaselineListResponse> mockPageResponse = new PageImpl<>(content, PageRequest.of(0, 10), content.size());
+            Page<BaselineListResponse> page = new PageImpl<>(content, PageRequest.of(0, 10), content.size());
+            PageResponse<BaselineListResponse> mockPageResponse = PageResponse.of(page);
 
             given(scenarioService.getBaselines(eq(1L), any()))
                     .willReturn(mockPageResponse);
@@ -274,11 +276,13 @@ class GetBaselines {
                             .param("size", "10"))
                     .andDo(print())
                     .andExpect(status().isOk())
-                    .andExpect(jsonPath("$.content").isArray())
-                    .andExpect(jsonPath("$.content[0].baselineId").value(200))
-                    .andExpect(jsonPath("$.content[0].title").value("대학 졸업 이후"))
+                    .andExpect(jsonPath("$.items").isArray())
+                    .andExpect(jsonPath("$.page").value(1))  // 1-based pagination
+                    .andExpect(jsonPath("$.size").value(10))
                     .andExpect(jsonPath("$.totalElements").value(2))
-                    .andExpect(jsonPath("$.totalPages").value(1));
+                    .andExpect(jsonPath("$.totalPages").value(1))
+                    .andExpect(jsonPath("$.items[0].baselineId").value(200))
+                    .andExpect(jsonPath("$.items[0].title").value("대학 졸업 이후"));
         }
     }
 
diff --git a/back/src/test/java/com/back/domain/user/controller/UserInfoControllerTest.java b/back/src/test/java/com/back/domain/user/controller/UserInfoControllerTest.java
@@ -113,6 +113,7 @@ void t1() throws Exception {
 
         Scenario scenario = Scenario.builder()
                 .user(testUser)
+                .baseLine(testBaseLine)  // BaseLine 추가
                 .decisionLine(testDecisionLine)
                 .status(ScenarioStatus.COMPLETED)
                 .job("Software Engineer")
