DevFW-CICD/ingress-nginx-helm
16
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update template_test.go

Browse source
This commit is contained in:
bmv126 2024-02-17 22:10:03 +05:30 committed by GitHub
parent 5940fb6c6e
commit 6148bcda09
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 52 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
internal/ingress/controller/template/template_test.go
View file

@ -959,10 +959,12 @@ func TestBuildRateLimit(t *testing.T) {
loc.RateLimit.RPS.Name = "rps"
loc.RateLimit.RPS.Limit = 1
loc.RateLimit.RPS.Burst = 1
loc.RateLimit.RPS.Delay = -1
loc.RateLimit.RPM.Name = "rpm"
loc.RateLimit.RPM.Limit = 2
loc.RateLimit.RPM.Burst = 2
loc.RateLimit.RPM.Delay = -1
loc.RateLimit.LimitRateAfter = 1
loc.RateLimit.LimitRate = 1
@ -983,6 +985,56 @@ func TestBuildRateLimit(t *testing.T) {
}
}
loc = &ingress.Location{}
loc.RateLimit.RPS.Name = "rps"
loc.RateLimit.RPS.Limit = 1
loc.RateLimit.RPS.Burst = 5
loc.RateLimit.RPS.Delay = 2
loc.RateLimit.RPM.Name = "rpm"
loc.RateLimit.RPM.Limit = 2
loc.RateLimit.RPM.Burst = 5
loc.RateLimit.RPM.Delay = 3
validLimits = []string{
"limit_req zone=rps burst=5 delay=2;",
"limit_req zone=rpm burst=5 delay=3;",
}
limits = buildRateLimit(loc)
for i, limit := range limits {
if limit != validLimits[i] {
t.Errorf("Expected '%v' but returned '%v'", validLimits, limits)
}
}
loc = &ingress.Location{}
loc.RateLimit.RPS.Name = "rps"
loc.RateLimit.RPS.Limit = 1
loc.RateLimit.RPS.Burst = 0
loc.RateLimit.RPS.Delay = -1
loc.RateLimit.RPM.Name = "rpm"
loc.RateLimit.RPM.Limit = 2
loc.RateLimit.RPM.Burst = 0
loc.RateLimit.RPM.Delay = -1
validLimits = []string{
"limit_req zone=rps nodelay;",
"limit_req zone=rpm nodelay;",
}
limits = buildRateLimit(loc)
for i, limit := range limits {
if limit != validLimits[i] {
t.Errorf("Expected '%v' but returned '%v'", validLimits, limits)
}
}
// Invalid limit
limits = buildRateLimit(&ingress.Ingress{})
if !reflect.DeepEqual(expected, limits) {