MiFaZhan
61 lines
2.2 KiB
Go
61 lines
2.2 KiB
Go
package inbound
|
|
|
|
import "testing"
|
|
|
|
func TestInboundCarrierForOperatorUsesDictionaryValues(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
operator int
|
|
carrier string
|
|
ok bool
|
|
}{
|
|
{name: "mobile", operator: 1, carrier: inboundCarrierMobile, ok: true},
|
|
{name: "telecom", operator: 2, carrier: inboundCarrierTelecom, ok: true},
|
|
{name: "legacy telecom value is unsupported", operator: 3, carrier: "", ok: false},
|
|
{name: "missing", operator: 0, carrier: "", ok: false},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
carrier, ok := inboundCarrierForOperator(test.operator)
|
|
if carrier != test.carrier || ok != test.ok {
|
|
t.Fatalf("inboundCarrierForOperator(%d) = %q,%v, want %q,%v", test.operator, carrier, ok, test.carrier, test.ok)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestResolveInboundUpstreamRateLimitFromChannelMetadataUsesCarrierPolicy(t *testing.T) {
|
|
cfg := defaultInboundUpstreamRateLimitConfig()
|
|
|
|
mobile := resolveInboundUpstreamRateLimitFromChannelMetadata(cfg, "chan-mobile", 1, "sim_yd_v5")
|
|
if mobile.carrier != inboundCarrierMobile {
|
|
t.Fatalf("mobile carrier = %q, want %q", mobile.carrier, inboundCarrierMobile)
|
|
}
|
|
if mobile.enabled {
|
|
t.Fatalf("mobile enabled = %v, want false", mobile.enabled)
|
|
}
|
|
|
|
telecom := resolveInboundUpstreamRateLimitFromChannelMetadata(cfg, "chan-telecom", 2, "sim_dx_v1")
|
|
if telecom.carrier != inboundCarrierTelecom {
|
|
t.Fatalf("telecom carrier = %q, want %q", telecom.carrier, inboundCarrierTelecom)
|
|
}
|
|
if !telecom.enabled {
|
|
t.Fatalf("telecom enabled = %v, want true", telecom.enabled)
|
|
}
|
|
if telecom.qps != defaultInboundUpstreamQPS {
|
|
t.Fatalf("telecom qps = %v, want %v", telecom.qps, defaultInboundUpstreamQPS)
|
|
}
|
|
}
|
|
|
|
func TestTelecomTemplateOverridesInconsistentOperatorMetadata(t *testing.T) {
|
|
for _, key := range []string{"sim_dx_v1", "sim_dx_v3"} {
|
|
for _, operator := range []int{0, 1, 2, 3} {
|
|
setting := resolveInboundUpstreamRateLimitFromChannelMetadata(defaultInboundUpstreamRateLimitConfig(), "telecom", operator, key)
|
|
if setting.carrier != inboundCarrierTelecom || !setting.enabled || setting.skipReason != "" {
|
|
t.Fatalf("template %q operator %d: %+v", key, operator, setting)
|
|
}
|
|
}
|
|
}
|
|
}
|