42
42
import org .opensearch .core .common .breaker .CircuitBreakingException ;
43
43
import org .opensearch .core .common .bytes .BytesArray ;
44
44
import org .opensearch .test .OpenSearchTestCase ;
45
+ import org .opensearch .transport .nativeprotocol .NativeInboundMessage ;
45
46
import org .junit .Before ;
46
47
47
48
import java .io .IOException ;
@@ -107,7 +108,7 @@ public void testInboundAggregation() throws IOException {
107
108
}
108
109
109
110
// Signal EOS
110
- InboundMessage aggregated = aggregator .finishAggregation ();
111
+ NativeInboundMessage aggregated = aggregator .finishAggregation ();
111
112
112
113
assertThat (aggregated , notNullValue ());
113
114
assertFalse (aggregated .isPing ());
@@ -138,7 +139,7 @@ public void testInboundUnknownAction() throws IOException {
138
139
assertEquals (0 , content .refCount ());
139
140
140
141
// Signal EOS
141
- InboundMessage aggregated = aggregator .finishAggregation ();
142
+ NativeInboundMessage aggregated = aggregator .finishAggregation ();
142
143
143
144
assertThat (aggregated , notNullValue ());
144
145
assertTrue (aggregated .isShortCircuit ());
@@ -161,7 +162,7 @@ public void testCircuitBreak() throws IOException {
161
162
content1 .close ();
162
163
163
164
// Signal EOS
164
- InboundMessage aggregated1 = aggregator .finishAggregation ();
165
+ NativeInboundMessage aggregated1 = aggregator .finishAggregation ();
165
166
166
167
assertEquals (0 , content1 .refCount ());
167
168
assertThat (aggregated1 , notNullValue ());
@@ -180,7 +181,7 @@ public void testCircuitBreak() throws IOException {
180
181
content2 .close ();
181
182
182
183
// Signal EOS
183
- InboundMessage aggregated2 = aggregator .finishAggregation ();
184
+ NativeInboundMessage aggregated2 = aggregator .finishAggregation ();
184
185
185
186
assertEquals (1 , content2 .refCount ());
186
187
assertThat (aggregated2 , notNullValue ());
@@ -199,7 +200,7 @@ public void testCircuitBreak() throws IOException {
199
200
content3 .close ();
200
201
201
202
// Signal EOS
202
- InboundMessage aggregated3 = aggregator .finishAggregation ();
203
+ NativeInboundMessage aggregated3 = aggregator .finishAggregation ();
203
204
204
205
assertEquals (1 , content3 .refCount ());
205
206
assertThat (aggregated3 , notNullValue ());
@@ -263,7 +264,7 @@ public void testFinishAggregationWillFinishHeader() throws IOException {
263
264
content .close ();
264
265
265
266
// Signal EOS
266
- InboundMessage aggregated = aggregator .finishAggregation ();
267
+ NativeInboundMessage aggregated = aggregator .finishAggregation ();
267
268
268
269
assertThat (aggregated , notNullValue ());
269
270
assertFalse (header .needsToReadVariableHeader ());
0 commit comments