16
16
package com .arpnetworking .tsdcore .model ;
17
17
18
18
import com .arpnetworking .metrics .aggregation .protocol .Messages ;
19
- import com .google .protobuf .GeneratedMessageV3 ;
19
+ import com .google .protobuf .GeneratedMessage ;
20
20
import com .google .protobuf .UnknownFieldSet ;
21
21
import io .vertx .core .buffer .Buffer ;
22
22
import org .apache .pekko .util .ByteString ;
@@ -35,7 +35,7 @@ public class AggregationMessageTest {
35
35
36
36
@ Test
37
37
public void testToBufferHostIdentification () {
38
- final GeneratedMessageV3 protobufMessage = Messages .HostIdentification .getDefaultInstance ();
38
+ final GeneratedMessage protobufMessage = Messages .HostIdentification .getDefaultInstance ();
39
39
final AggregationMessage message = AggregationMessage .create (protobufMessage );
40
40
Assert .assertNotNull (message );
41
41
Assert .assertSame (protobufMessage , message .getMessage ());
@@ -60,14 +60,14 @@ public void testToBufferHostIdentification() {
60
60
61
61
@ Test (expected = IllegalArgumentException .class )
62
62
public void testToBufferSerializedUnsupportedMessage () {
63
- final GeneratedMessageV3 mockMessage = Mockito .mock (GeneratedMessageV3 .class );
63
+ final GeneratedMessage mockMessage = Mockito .mock (GeneratedMessage .class );
64
64
Mockito .doReturn (UnknownFieldSet .getDefaultInstance ()).when (mockMessage ).getUnknownFields ();
65
65
AggregationMessage .create (mockMessage ).serializeToBuffer ();
66
66
}
67
67
68
68
@ Test
69
69
public void testToByteStringHostIdentification () {
70
- final GeneratedMessageV3 protobufMessage = Messages .HostIdentification .getDefaultInstance ();
70
+ final GeneratedMessage protobufMessage = Messages .HostIdentification .getDefaultInstance ();
71
71
final AggregationMessage message = AggregationMessage .create (protobufMessage );
72
72
Assert .assertNotNull (message );
73
73
Assert .assertSame (protobufMessage , message .getMessage ());
@@ -92,14 +92,14 @@ public void testToByteStringHostIdentification() {
92
92
93
93
@ Test (expected = IllegalArgumentException .class )
94
94
public void testToByteStringSerializedUnsupportedMessage () {
95
- final GeneratedMessageV3 mockMessage = Mockito .mock (GeneratedMessageV3 .class );
95
+ final GeneratedMessage mockMessage = Mockito .mock (GeneratedMessage .class );
96
96
Mockito .doReturn (UnknownFieldSet .getDefaultInstance ()).when (mockMessage ).getUnknownFields ();
97
97
AggregationMessage .create (mockMessage ).serializeToByteString ();
98
98
}
99
99
100
100
@ Test
101
101
public void testPekkoVsVertx () {
102
- final GeneratedMessageV3 protobufMessage = Messages .HostIdentification .getDefaultInstance ();
102
+ final GeneratedMessage protobufMessage = Messages .HostIdentification .getDefaultInstance ();
103
103
final AggregationMessage message = AggregationMessage .create (protobufMessage );
104
104
Assert .assertNotNull (message );
105
105
Assert .assertSame (protobufMessage , message .getMessage ());
0 commit comments