61 lines
2.4 KiB
Go
61 lines
2.4 KiB
Go
package dispatchadapter
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestValidateRoutingPolicyJSON(t *testing.T) {
|
|
// 空串合法（使用全部内置默认）
|
|
assert.NoError(t, ValidateRoutingPolicyJSON(""))
|
|
|
|
// 合法的部分覆盖
|
|
assert.NoError(t, ValidateRoutingPolicyJSON(`{"tierFactors":{"secondary":0.3}}`))
|
|
assert.NoError(t, ValidateRoutingPolicyJSON(`{"breaker":{"failThreshold":5},"retry":{"inPlaceBudget":3}}`))
|
|
|
|
// 非法 JSON
|
|
err := ValidateRoutingPolicyJSON(`{"tierFactors":`)
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "JSON 解析失败")
|
|
|
|
// 合法 JSON 但 Schema 越界
|
|
assert.Error(t, ValidateRoutingPolicyJSON(`{"tierFactors":{"primary":0}}`), "primary 因子必须 > 0")
|
|
assert.Error(t, ValidateRoutingPolicyJSON(`{"health":{"alpha":-1}}`))
|
|
assert.Error(t, ValidateRoutingPolicyJSON(`{"retry":{"failoverBudget":-1}}`))
|
|
assert.Error(t, ValidateRoutingPolicyJSON(`{"tierFactors":{"vip":1}}`), "未知层级")
|
|
assert.Error(t, ValidateRoutingPolicyJSON(`{"breaker":{"modelFailThreshold":0}}`), "模型级阈值必须 > 0")
|
|
assert.NoError(t, ValidateRoutingPolicyJSON(`{"breaker":{"modelFailThreshold":6}}`))
|
|
|
|
// 存量 JSON 缺 modelFailThreshold → 内置默认 4（默认值 + 浅覆盖机制兼容）
|
|
pol, err := buildPolicy(`{"breaker":{"failThreshold":5}}`)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 5, pol.Breaker.FailThreshold)
|
|
assert.Equal(t, 4, pol.Breaker.ModelFailThreshold)
|
|
}
|
|
|
|
func TestBuildTenantPolicy(t *testing.T) {
|
|
// 三层浅合并：默认 → 全局 → 租户，后者覆盖前者
|
|
pol, err := buildTenantPolicy(
|
|
`{"tierFactors":{"secondary":0.3},"retry":{"inPlaceBudget":3}}`,
|
|
`{"retry":{"inPlaceBudget":1}}`,
|
|
)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 0.3, pol.TierFactors["secondary"], "全局覆盖保留")
|
|
assert.Equal(t, 1, pol.Retry.InPlaceBudget, "租户覆盖优先于全局")
|
|
assert.Equal(t, 2, pol.Retry.FailoverBudget, "未覆盖字段用内置默认")
|
|
|
|
// 全局串非法时忽略全局层，租户层仍生效
|
|
pol, err = buildTenantPolicy(`{invalid`, `{"retry":{"inPlaceBudget":1}}`)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 1, pol.Retry.InPlaceBudget)
|
|
|
|
// 租户串非法 → 错误
|
|
_, err = buildTenantPolicy("", `{invalid`)
|
|
assert.Error(t, err)
|
|
|
|
// 租户覆盖导致 Schema 越界 → 错误
|
|
_, err = buildTenantPolicy("", `{"tierFactors":{"primary":0}}`)
|
|
assert.Error(t, err)
|
|
}
|