28
28
import org .junit .runner .RunWith ;
29
29
import org .mockito .Mock ;
30
30
import org .mockito .junit .MockitoJUnitRunner ;
31
+ import rx .Observable ;
31
32
32
33
/**
33
34
* Tests for {@link BaseFilter}. Currently named BaseFilter2Test as there is an existing class named BaseFilterTest.
@@ -79,12 +80,12 @@ public ZuulMessage apply(ZuulMessage req) {
79
80
@ Test
80
81
void validateDefaultConcurrencyLimit () {
81
82
final int [] limit = {0 };
82
- class ConcInboundFilter extends BaseSyncFilter {
83
+ class ConcInboundFilter extends BaseFilter {
83
84
84
85
@ Override
85
- public ZuulMessage apply (ZuulMessage input ) {
86
+ public Observable applyAsync (ZuulMessage input ) {
86
87
limit [0 ] = Math .max (filterConcurrencyCustom .get (), filterConcurrencyDefault .get ());
87
- return null ;
88
+ return Observable . just ( "Done" ) ;
88
89
}
89
90
90
91
@ Override
@@ -97,7 +98,7 @@ public boolean shouldFilter(ZuulMessage msg) {
97
98
return true ;
98
99
}
99
100
}
100
- new ConcInboundFilter ().apply (new ZuulMessageImpl (new SessionContext (), new Headers ()));
101
+ new ConcInboundFilter ().applyAsync (new ZuulMessageImpl (new SessionContext (), new Headers ()));
101
102
Truth .assertThat (limit [0 ]).isEqualTo (4000 );
102
103
}
103
104
@@ -108,12 +109,12 @@ void validateFilterConcurrencyLimitOverride() {
108
109
configuration .setProperty ("zuul.ConcInboundFilter.in.concurrency.limit" , 4000 );
109
110
final int [] limit = {0 };
110
111
111
- class ConcInboundFilter extends BaseSyncFilter {
112
+ class ConcInboundFilter extends BaseFilter {
112
113
113
114
@ Override
114
- public ZuulMessage apply (ZuulMessage input ) {
115
+ public Observable applyAsync (ZuulMessage input ) {
115
116
limit [0 ] = Math .max (filterConcurrencyCustom .get (), filterConcurrencyDefault .get ());
116
- return null ;
117
+ return Observable . just ( "Done" ) ;
117
118
}
118
119
119
120
@ Override
@@ -126,7 +127,7 @@ public boolean shouldFilter(ZuulMessage msg) {
126
127
return true ;
127
128
}
128
129
}
129
- new ConcInboundFilter ().apply (new ZuulMessageImpl (new SessionContext (), new Headers ()));
130
+ new ConcInboundFilter ().applyAsync (new ZuulMessageImpl (new SessionContext (), new Headers ()));
130
131
Truth .assertThat (limit [0 ]).isEqualTo (7000 );
131
132
}
132
133
}
0 commit comments