Skip to content

Commit fed1845

Browse files
committed
test: AI 요청 로그 타입 추가
1 parent c83449c commit fed1845

1 file changed

Lines changed: 21 additions & 0 deletions

File tree

backend/src/test/java/com/stackup/stackup/log/ai/application/AiRequestLogServiceTest.java

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,8 @@
1919
import org.mockito.InjectMocks;
2020
import org.mockito.Mock;
2121
import org.mockito.junit.jupiter.MockitoExtension;
22+
import org.junit.jupiter.params.ParameterizedTest;
23+
import org.junit.jupiter.params.provider.ValueSource;
2224
import org.springframework.test.util.ReflectionTestUtils;
2325

2426
@ExtendWith(MockitoExtension.class)
@@ -60,4 +62,23 @@ void record_fallsBackToFailedOnUnknownStatus() {
6062
verify(logRepository).save(cap.capture());
6163
assertThat(cap.getValue().getStatus()).isEqualTo(AiRequestStatus.FAILED);
6264
}
65+
66+
@ParameterizedTest
67+
@ValueSource(strings = {
68+
"generate.questions",
69+
"generate.followup",
70+
"generate.feedback",
71+
"stt.transcribe",
72+
"tts.synthesize"
73+
})
74+
void record_keepsAiWorkflowRequestType(String requestType) {
75+
service.record(new AiRequestLogCommand(
76+
null, null, requestType, "gemini-flash",
77+
10, 20, 800, "SUCCESS", null
78+
));
79+
80+
ArgumentCaptor<AiRequestLog> cap = ArgumentCaptor.forClass(AiRequestLog.class);
81+
verify(logRepository).save(cap.capture());
82+
assertThat(cap.getValue().getRequestType()).isEqualTo(requestType);
83+
}
6384
}

0 commit comments

Comments
 (0)