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

refactor(test): improve type assertions in ops endpoint context tests

Browse Source
This commit is contained in:
Ethan0x0000
2026-03-24 09:52:56 +08:00
parent 2f8e10db46
commit f10e56be7e
2 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
backend/internal/handler/ops_error_logger_test.go
View File

@@ -285,11 +285,15 @@ func TestSetOpsEndpointContext_SetsContextKeys(t *testing.T) {
v, ok := c.Get(opsUpstreamModelKey)
require.True(t, ok)
require.Equal(t, "claude-3-5-sonnet-20241022", v.(string))
vStr, ok := v.(string)
require.True(t, ok)
require.Equal(t, "claude-3-5-sonnet-20241022", vStr)
rt, ok := c.Get(opsRequestTypeKey)
require.True(t, ok)
require.Equal(t, int16(2), rt.(int16))
rtVal, ok := rt.(int16)
require.True(t, ok)
require.Equal(t, int16(2), rtVal)
}
func TestSetOpsEndpointContext_EmptyModelNotStored(t *testing.T) {
@@ -305,7 +309,9 @@ func TestSetOpsEndpointContext_EmptyModelNotStored(t *testing.T) {
rt, ok := c.Get(opsRequestTypeKey)
require.True(t, ok)
require.Equal(t, int16(1), rt.(int16))
rtVal, ok := rt.(int16)
require.True(t, ok)
require.Equal(t, int16(1), rtVal)
}
func TestSetOpsEndpointContext_NilContext(t *testing.T) {