@@ -59,17 +59,21 @@ func TestDefaultModuleConfig(t *testing.T) {
59
59
if c .Timeout () != DefaultTimeout {
60
60
t .Errorf ("Unexpected Timeout: %v" , c .Timeout ())
61
61
}
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
+ }
64
66
}
65
67
if ! c .IsAllowCode (200 ) {
66
68
t .Errorf ("Unexpected IsAllowCode(200): %v" , c .IsAllowCode (200 ))
67
69
}
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
+ }
70
74
}
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 ))
73
77
}
74
78
if c .IsBlockCode (200 ) {
75
79
t .Errorf ("Unexpected IsBlockCode(200): %v" , c .IsBlockCode (200 ))
@@ -95,9 +99,8 @@ func TestConfiguredModuleConfig(t *testing.T) {
95
99
if c .AllowUnknownContentLength () != true {
96
100
t .Errorf ("Unexpected AllowUnknownContentLength: %v" , c .AllowUnknownContentLength ())
97
101
}
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 ())
101
104
}
102
105
if c .AnomalyDuration () != 10 * time .Second {
103
106
t .Errorf ("Unexpected AnomalyDuration: %v" , c .AnomalyDuration ())
@@ -141,17 +144,21 @@ func TestConfiguredModuleConfig(t *testing.T) {
141
144
if c .Timeout () != 10 * time .Millisecond {
142
145
t .Errorf ("Unexpected Timeout: %v" , c .Timeout ())
143
146
}
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
+ }
146
151
}
147
152
if ! c .IsAllowCode (200 ) {
148
153
t .Errorf ("Unexpected IsAllowCode(200): %v" , c .IsAllowCode (200 ))
149
154
}
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
+ }
152
159
}
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 ))
155
162
}
156
163
if c .IsBlockCode (200 ) {
157
164
t .Errorf ("Unexpected IsBlockCode(200): %v" , c .IsBlockCode (200 ))
@@ -184,9 +191,8 @@ func TestFromModuleConfig(t *testing.T) {
184
191
if c .AllowUnknownContentLength () != true {
185
192
t .Errorf ("Unexpected AllowUnknownContentLength: %v" , c .AllowUnknownContentLength ())
186
193
}
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 ())
190
196
}
191
197
if c .AnomalyDuration () != 10 * time .Second {
192
198
t .Errorf ("Unexpected AnomalyDuration: %v" , c .AnomalyDuration ())
@@ -230,17 +236,21 @@ func TestFromModuleConfig(t *testing.T) {
230
236
if c .Timeout () != 10 * time .Millisecond {
231
237
t .Errorf ("Unexpected Timeout: %v" , c .Timeout ())
232
238
}
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
+ }
235
243
}
236
244
if ! c .IsAllowCode (200 ) {
237
245
t .Errorf ("Unexpected IsAllowCode(200): %v" , c .IsAllowCode (200 ))
238
246
}
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
+ }
241
251
}
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 ))
244
254
}
245
255
if c .IsBlockCode (200 ) {
246
256
t .Errorf ("Unexpected IsBlockCode(200): %v" , c .IsBlockCode (200 ))
0 commit comments