Skip to content

Commit def121b

Browse files
committed
Fix failing test and further align code
1 parent 1682253 commit def121b

2 files changed

Lines changed: 11 additions & 14 deletions

File tree

test/org/apache/catalina/filters/TestRateLimitFilter.java

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,6 @@
1717
package org.apache.catalina.filters;
1818

1919
import javax.servlet.FilterChain;
20-
import javax.servlet.FilterConfig;
21-
import javax.servlet.ServletException;
2220
import javax.servlet.http.HttpServletResponse;
2321

2422
import org.junit.Assert;
@@ -47,11 +45,12 @@ private void testRateLimitWith4Clients(boolean exposeHeaders, boolean enforce) t
4745

4846
Tomcat tomcat = getTomcatInstance();
4947
Context root = tomcat.addContext("", TEMP_DIR);
50-
tomcat.start();
5148

5249
MockFilterChain filterChain = new MockFilterChain();
5350
RateLimitFilter rateLimitFilter = testRateLimitFilter(filterDef, root);
5451

52+
tomcat.start();
53+
5554
FastRateLimiter fastRateLimiter = (FastRateLimiter) rateLimitFilter.rateLimiter;
5655

5756
int allowedRequests = fastRateLimiter.getRequests();
@@ -127,7 +126,7 @@ public void testUnexposeHeaderAndEnforcedRateLimitWith4Clients() throws Exceptio
127126
testRateLimitWith4Clients(false, true);
128127
}
129128

130-
private RateLimitFilter testRateLimitFilter(FilterDef filterDef, Context root) throws ServletException {
129+
private RateLimitFilter testRateLimitFilter(FilterDef filterDef, Context root) {
131130

132131
RateLimitFilter rateLimitFilter = new RateLimitFilter();
133132
filterDef.setFilterClass(RateLimitFilter.class.getName());
@@ -140,10 +139,6 @@ private RateLimitFilter testRateLimitFilter(FilterDef filterDef, Context root) t
140139
filterMap.addURLPatternDecoded("*");
141140
root.addFilterMap(filterMap);
142141

143-
FilterConfig filterConfig = TesterFilterConfigs.generateFilterConfig(filterDef);
144-
145-
rateLimitFilter.init(filterConfig);
146-
147142
return rateLimitFilter;
148143
}
149144

test/org/apache/catalina/filters/TestRateLimitFilterWithExactRateLimiter.java

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,8 @@
3131
import org.apache.tomcat.util.descriptor.web.FilterMap;
3232

3333
public class TestRateLimitFilterWithExactRateLimiter extends TomcatBaseTest {
34-
private void testRateLimitWith1Clients(boolean exposeHeaders, boolean enforce) throws Exception {
34+
35+
private void testRateLimitWith4Clients(boolean exposeHeaders, boolean enforce) throws Exception {
3536

3637
int bucketRequests = 40;
3738
int bucketDuration = 4;
@@ -45,11 +46,12 @@ private void testRateLimitWith1Clients(boolean exposeHeaders, boolean enforce) t
4546

4647
Tomcat tomcat = getTomcatInstance();
4748
Context root = tomcat.addContext("", TEMP_DIR);
48-
tomcat.start();
4949

5050
MockFilterChain filterChain = new MockFilterChain();
5151
RateLimitFilter rateLimitFilter = testRateLimitFilter(filterDef, root);
5252

53+
tomcat.start();
54+
5355
ExactRateLimiter exactRateLimiter = (ExactRateLimiter) rateLimitFilter.rateLimiter;
5456

5557
int allowedRequests = exactRateLimiter.getRequests();
@@ -111,22 +113,22 @@ private void testRateLimitWith1Clients(boolean exposeHeaders, boolean enforce) t
111113

112114
@Test
113115
public void testExposeHeaderAndReferenceRateLimitWith4Clients() throws Exception {
114-
testRateLimitWith1Clients(true, false);
116+
testRateLimitWith4Clients(true, false);
115117
}
116118

117119
@Test
118120
public void testUnexposeHeaderAndReferenceRateLimitWith4Clients() throws Exception {
119-
testRateLimitWith1Clients(false, false);
121+
testRateLimitWith4Clients(false, false);
120122
}
121123

122124
@Test
123125
public void testExposeHeaderAndEnforceRateLimitWith4Clients() throws Exception {
124-
testRateLimitWith1Clients(true, true);
126+
testRateLimitWith4Clients(true, true);
125127
}
126128

127129
@Test
128130
public void testUnexposeHeaderAndEnforceRateLimitWith4Clients() throws Exception {
129-
testRateLimitWith1Clients(false, true);
131+
testRateLimitWith4Clients(false, true);
130132
}
131133

132134
private RateLimitFilter testRateLimitFilter(FilterDef filterDef, Context root) {

0 commit comments

Comments
 (0)