@@ -59,17 +59,21 @@ func TestDefaultModuleConfig(t *testing.T) {
5959 if c .Timeout () != DefaultTimeout {
6060 t .Errorf ("Unexpected Timeout: %v" , c .Timeout ())
6161 }
62- if c .IsAllowCode (406 ) {
63- t .Errorf ("Unexpected IsAllowCode(406): %v" , c .IsAllowCode (406 ))
62+ for code := 300 ; code < 600 ; code ++ {
63+ if c .IsAllowCode (code ) {
64+ t .Errorf ("Unexpected IsAllowCode(%d): %v" , code , c .IsAllowCode (code ))
65+ }
6466 }
6567 if ! c .IsAllowCode (200 ) {
6668 t .Errorf ("Unexpected IsAllowCode(200): %v" , c .IsAllowCode (200 ))
6769 }
68- if ! c .IsBlockCode (406 ) {
69- t .Errorf ("Unexpected IsBlockCode(406): %v" , c .IsBlockCode (406 ))
70+ for code := 300 ; code < 600 ; code ++ {
71+ if ! c .IsBlockCode (code ) {
72+ t .Errorf ("Unexpected IsBlockCode(%d): %v" , code , c .IsBlockCode (code ))
73+ }
7074 }
71- if c .IsBlockCode (403 ) {
72- t .Errorf ("Unexpected IsBlockCode(403 ): %v" , c .IsBlockCode (403 ))
75+ if c .IsBlockCode (600 ) {
76+ t .Errorf ("Unexpected IsBlockCode(600 ): %v" , c .IsBlockCode (600 ))
7377 }
7478 if c .IsBlockCode (200 ) {
7579 t .Errorf ("Unexpected IsBlockCode(200): %v" , c .IsBlockCode (200 ))
@@ -95,9 +99,8 @@ func TestConfiguredModuleConfig(t *testing.T) {
9599 if c .AllowUnknownContentLength () != true {
96100 t .Errorf ("Unexpected AllowUnknownContentLength: %v" , c .AllowUnknownContentLength ())
97101 }
98- altResponseCodes := c .AltResponseCodes ()
99- if len (altResponseCodes ) != 1 || altResponseCodes [0 ] != 403 {
100- t .Errorf ("Unexpected AltResponseCodes: %v" , altResponseCodes )
102+ if c .AltResponseCodes () != nil {
103+ t .Errorf ("Unexpected AltResponseCodes from deprecated option (should be nil): %v" , c .AltResponseCodes ())
101104 }
102105 if c .AnomalyDuration () != 10 * time .Second {
103106 t .Errorf ("Unexpected AnomalyDuration: %v" , c .AnomalyDuration ())
@@ -141,17 +144,21 @@ func TestConfiguredModuleConfig(t *testing.T) {
141144 if c .Timeout () != 10 * time .Millisecond {
142145 t .Errorf ("Unexpected Timeout: %v" , c .Timeout ())
143146 }
144- if c .IsAllowCode (406 ) {
145- t .Errorf ("Unexpected IsAllowCode(406): %v" , c .IsAllowCode (406 ))
147+ for code := 300 ; code < 600 ; code ++ {
148+ if c .IsAllowCode (code ) {
149+ t .Errorf ("Unexpected IsAllowCode(%d): %v" , code , c .IsAllowCode (code ))
150+ }
146151 }
147152 if ! c .IsAllowCode (200 ) {
148153 t .Errorf ("Unexpected IsAllowCode(200): %v" , c .IsAllowCode (200 ))
149154 }
150- if ! c .IsBlockCode (406 ) {
151- t .Errorf ("Unexpected IsBlockCode(406): %v" , c .IsBlockCode (406 ))
155+ for code := 300 ; code < 600 ; code ++ {
156+ if ! c .IsBlockCode (code ) {
157+ t .Errorf ("Unexpected IsBlockCode(%d): %v" , code , c .IsBlockCode (code ))
158+ }
152159 }
153- if ! c .IsBlockCode (403 ) {
154- t .Errorf ("Unexpected IsBlockCode(403 ): %v" , c .IsBlockCode (403 ))
160+ if c .IsBlockCode (600 ) {
161+ t .Errorf ("Unexpected IsBlockCode(600 ): %v" , c .IsBlockCode (600 ))
155162 }
156163 if c .IsBlockCode (200 ) {
157164 t .Errorf ("Unexpected IsBlockCode(200): %v" , c .IsBlockCode (200 ))
@@ -184,9 +191,8 @@ func TestFromModuleConfig(t *testing.T) {
184191 if c .AllowUnknownContentLength () != true {
185192 t .Errorf ("Unexpected AllowUnknownContentLength: %v" , c .AllowUnknownContentLength ())
186193 }
187- altResponseCodes := c .AltResponseCodes ()
188- if len (altResponseCodes ) != 1 || altResponseCodes [0 ] != 403 {
189- t .Errorf ("Unexpected AltResponseCodes: %v" , altResponseCodes )
194+ if c .AltResponseCodes () != nil {
195+ t .Errorf ("Unexpected AltResponseCodes from deprecated option (should be nil): %v" , c .AltResponseCodes ())
190196 }
191197 if c .AnomalyDuration () != 10 * time .Second {
192198 t .Errorf ("Unexpected AnomalyDuration: %v" , c .AnomalyDuration ())
@@ -230,17 +236,21 @@ func TestFromModuleConfig(t *testing.T) {
230236 if c .Timeout () != 10 * time .Millisecond {
231237 t .Errorf ("Unexpected Timeout: %v" , c .Timeout ())
232238 }
233- if c .IsAllowCode (406 ) {
234- t .Errorf ("Unexpected IsAllowCode(406): %v" , c .IsAllowCode (406 ))
239+ for code := 300 ; code < 600 ; code ++ {
240+ if c .IsAllowCode (code ) {
241+ t .Errorf ("Unexpected IsAllowCode(%d): %v" , code , c .IsAllowCode (code ))
242+ }
235243 }
236244 if ! c .IsAllowCode (200 ) {
237245 t .Errorf ("Unexpected IsAllowCode(200): %v" , c .IsAllowCode (200 ))
238246 }
239- if ! c .IsBlockCode (406 ) {
240- t .Errorf ("Unexpected IsBlockCode(406): %v" , c .IsBlockCode (406 ))
247+ for code := 300 ; code < 600 ; code ++ {
248+ if ! c .IsBlockCode (code ) {
249+ t .Errorf ("Unexpected IsBlockCode(%d): %v" , code , c .IsBlockCode (code ))
250+ }
241251 }
242- if ! c .IsBlockCode (403 ) {
243- t .Errorf ("Unexpected IsBlockCode(403 ): %v" , c .IsBlockCode (403 ))
252+ if c .IsBlockCode (600 ) {
253+ t .Errorf ("Unexpected IsBlockCode(600 ): %v" , c .IsBlockCode (600 ))
244254 }
245255 if c .IsBlockCode (200 ) {
246256 t .Errorf ("Unexpected IsBlockCode(200): %v" , c .IsBlockCode (200 ))
0 commit comments