zfc/sub2api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: 修复内容审计日志异步断言

Browse Source
This commit is contained in:
shaw 2026-05-29 09:57:02 +08:00
parent 514ac5c6a1
commit 6010c3cca9
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backend/internal/handler/openai_gateway_handler_test.go
View File

@ -867,8 +867,11 @@ func TestOpenAIResponsesWebSocket_ContentModerationBlocksFirstFrame(t *testing.T
require.Equal(t, coderws.StatusPolicyViolation, closeErr.Code) require.Equal(t, coderws.StatusPolicyViolation, closeErr.Code)
require.Contains(t, closeErr.Reason, "内容审计测试阻断") require.Contains(t, closeErr.Reason, "内容审计测试阻断")
} }
logs := repo.logSnapshot() var logs []service.ContentModerationLog
require.Len(t, logs, 1) require.Eventually(t, func() bool {
logs = repo.logSnapshot()
return len(logs) == 1
}, time.Second, 10*time.Millisecond)
require.True(t, logs[0].Flagged) require.True(t, logs[0].Flagged)
require.Equal(t, service.ContentModerationActionBlock, logs[0].Action) require.Equal(t, service.ContentModerationActionBlock, logs[0].Action)
require.Equal(t, "bad prompt", logs[0].InputExcerpt) require.Equal(t, "bad prompt", logs[0].InputExcerpt)