feat: 테스트 코드 작성

Author: phdcoco

src/main/java/com/codereferee/codereferee_server/infrastructure/redis/ResultQueueConsumer.java

@@ -2,6 +2,7 @@
 
 import com.codereferee.codereferee_server.domain.validation.AgentStep;
 import com.codereferee.codereferee_server.domain.validation.TaskStatus;
+import com.codereferee.codereferee_server.domain.validation.TaskStatusHistoryRepository;
 import com.codereferee.codereferee_server.domain.validation.TaskStatusRepository;
 import com.codereferee.codereferee_server.infrastructure.metrics.PipelineMetrics;
 import com.codereferee.codereferee_server.infrastructure.persistence.TaskStatusPgRepository;
@@ -37,7 +38,7 @@ public class ResultQueueConsumer {
 
     private final RedisTemplate<String, Object> redisTemplate;
     private final TaskStatusRepository taskStatusRepository;
-    private final TaskStatusPgRepository pgRepository;
+    private final TaskStatusHistoryRepository historyRepository;
     private final PipelineMetrics pipelineMetrics;
     private final ObjectMapper objectMapper;
 
@@ -91,18 +92,21 @@ private void handleProgress(ProgressEventMessage event) {
         Optional<AgentStep> stepOpt = event.resolveStep();
         if (taskId == null || stepOpt.isEmpty()) {
             log.warn("[ResultQueue] 잘못된 이벤트 상태는 무시됩니다. taskId = {}, step = {}", taskId, event.step());
+            return;
         }
 
         Optional<TaskStatus> currentOpt = taskStatusRepository.findById(taskId);
         if (currentOpt.isEmpty()) {
             log.warn("[ResultQueue] 알 수 없는 taskId는 무시됩니다. taskId = {}", taskId);
+            return;
         }
 
         TaskStatus current = currentOpt.get();
 
         // 종결 후 늦게 도착한 progress는 무시한다.
         if (current.currentAgent().isTerminal()) {
             log.info("[ResultQueue] taskID = {} 는 이미 {} 상태로 확정되었습니다. {} 상태는 무시됩니다.", taskId, current.currentAgent(), event.step());
+            return;
         }
 
         AgentStep step = stepOpt.get();
@@ -114,7 +118,7 @@ private void handleProgress(ProgressEventMessage event) {
         }
 
         taskStatusRepository.save(updated);
-        pgRepository.upsert(updated);
+        historyRepository.upsert(updated);
 
         log.info("ResultQueue taskId = {} 상태 -> {}{}", taskId, step, event.round() != null ?
                 " (round " + event.round() + " / " + event.maxRounds() + ")" : "");
@@ -139,7 +143,7 @@ private void handleResult(SandboxResultMessage msg) {
         pipelineMetrics.recordTransition(current.currentAgent(), verdict);
         pipelineMetrics.recordVerdict(verdict);
         taskStatusRepository.save(updated);
-        pgRepository.upsert(updated);
+        historyRepository.upsert(updated);
 
         log.info("[ResultQueue] taskId = {} -> {} reports = {}", taskId, verdict, aiReports.keySet());
 }

src/test/java/com/codereferee/codereferee_server/MockE2ePipelineTest.java

@@ -77,6 +77,7 @@ void failRepositoryEndsFailed() {
         assertThat(status.get("currentAgent")).isEqualTo("FAILED");
         assertThat(status.get("isExecutable")).isEqualTo(false);
         assertThat(reports(status)).containsKey("critic_feedback");
+        assertThat(status.get("iterationCount")).isEqualTo(3);
     }
 
     @Test

src/test/java/com/codereferee/codereferee_server/infrastructure/redis/ResultQueueConsumerTest.java

@@ -16,20 +16,18 @@
 import java.util.Optional;
 
 import static org.assertj.core.api.Assertions.assertThat;
-import static org.mockito.Mockito.mock;
-import static org.mockito.Mockito.verify;
-import static org.mockito.Mockito.when;
+import static org.mockito.Mockito.*;
 
 class ResultQueueConsumerTest {
 
     private final TaskStatusRepository taskStatusRepository = mock(TaskStatusRepository.class);
-    private final TaskStatusHistoryRepository pgRepository = mock(TaskStatusHistoryRepository.class);
+    private final TaskStatusHistoryRepository historyRepository = mock(TaskStatusHistoryRepository.class);
     private final PipelineMetrics pipelineMetrics = new PipelineMetrics(new SimpleMeterRegistry());
     private final ObjectMapper objectMapper = new ObjectMapper()
             .configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
     private final ResultQueueConsumer consumer = new ResultQueueConsumer(
             mock(org.springframework.data.redis.core.RedisTemplate.class),
-            taskStatusRepository, pgRepository, pipelineMetrics, objectMapper);
+            taskStatusRepository, historyRepository, pipelineMetrics, objectMapper);
 
     private TaskStatus queued(String taskId) {
         return new TaskStatus(taskId, AgentStep.JUDGING, false, 0, null,
@@ -54,7 +52,7 @@ void successMessageMapsToPassed() {
         assertThat(saved.executable()).isTrue();
         assertThat(saved.errorMessage()).isNull();
         assertThat(saved.aiReports()).containsKey("judge_report");
-        verify(pgRepository).upsert(saved);
+        verify(historyRepository).upsert(saved);
     }
 
     @Test
@@ -94,4 +92,55 @@ void infraErrorMapsToErrorNotFailed() {
         assertThat(saved.currentAgent()).isEqualTo(AgentStep.ERROR);
         assertThat(saved.errorMessage()).contains("판정 불가");
     }
+    @Test
+    void progressEventUpdatesStepAndRound() {
+        when(taskStatusRepository.findById("t4")).thenReturn(Optional.of(queued("t4")));
+
+        consumer.process(Map.of(
+                "type", "progress",
+                "request_id", "t4",
+                "step", "REFINING",
+                "round", 2,
+                "max_rounds", 3
+        ));
+
+        ArgumentCaptor<TaskStatus> captor = ArgumentCaptor.forClass(TaskStatus.class);
+        verify(taskStatusRepository).save(captor.capture());
+        TaskStatus saved = captor.getValue();
+
+        assertThat(saved.currentAgent()).isEqualTo(AgentStep.REFINING);
+        assertThat(saved.iterationCount()).isEqualTo(2);
+        assertThat(saved.errorMessage()).isNull();
+        verify(historyRepository).upsert(saved);
+    }
+
+    @Test
+    void progressAfterTerminalIsIgnored() {
+        TaskStatus done = new TaskStatus("t5", AgentStep.PASSED, true, 1, null,
+                LocalDateTime.of(2026, 8, 3, 12, 0), "https://github.com/phdcoco/QuickByte_Demo", "main", "d1c5c5e", null);
+        when(taskStatusRepository.findById("t5")).thenReturn(Optional.of(done));
+
+        consumer.process(Map.of("type", "progress", "request_id", "t5", "step", "BASELINE"));
+
+        // 확정된 결과는 늦게 온 progress로 되돌아가지 않는다
+        verify(taskStatusRepository, never()).save(any());
+        verify(historyRepository, never()).upsert(any());
+    }
+
+    @Test
+    void unknownMessageTypeIsIgnored() {
+        consumer.process(Map.of("type", "heartbeat", "request_id", "t6"));
+
+        verify(taskStatusRepository, never()).save(any());
+        verify(historyRepository, never()).upsert(any());
+    }
+
+    @Test
+    void unknownStepProgressIsIgnored() {
+        when(taskStatusRepository.findById("t7")).thenReturn(Optional.of(queued("t7")));
+
+        consumer.process(Map.of("type", "progress", "request_id", "t7", "step", "DEPLOYING"));
+
+        verify(taskStatusRepository, never()).save(any());
+    }
 }