test(H-359): assert direct upload completes after retry (#70)

Co-authored-by: Rogee <rogee@ipao.vip>
2026-08-21 13:13:58 +08:00
parent 181e9051ec
commit 517a6090b3
1 changed files with 3 additions and 0 deletions
@@ -311,6 +311,9 @@ func TestWidgetService_PublicCreateMessageRollsBackCaptainJobFailureAndRetries(t
 	message, _, attachments, err := svc.PublicCreateMessage(context.Background(), "public-api", "visitor-source", displayID, req)
 	require.NoError(t, err)
 	require.Len(t, attachments, 1)
+	var retriedUpload model.DirectUpload
+	require.NoError(t, db.First(&retriedUpload, upload.ID).Error)
+	assert.Equal(t, model.DirectUploadStatusCompleted, retriedUpload.Status)
 	assert.Len(t, listener.events, 1)
 	var job model.BackgroundJob
 	require.NoError(t, db.Where("job_type = ?", TaskTypeCaptainConversationResponseBuilder).First(&job).Error)