1616package com .arpnetworking .tsdcore .model ;
1717
1818import com .arpnetworking .metrics .aggregation .protocol .Messages ;
19- import com .google .protobuf .GeneratedMessageV3 ;
19+ import com .google .protobuf .GeneratedMessage ;
2020import com .google .protobuf .UnknownFieldSet ;
2121import io .vertx .core .buffer .Buffer ;
2222import org .apache .pekko .util .ByteString ;
@@ -35,7 +35,7 @@ public class AggregationMessageTest {
3535
3636 @ Test
3737 public void testToBufferHostIdentification () {
38- final GeneratedMessageV3 protobufMessage = Messages .HostIdentification .getDefaultInstance ();
38+ final GeneratedMessage protobufMessage = Messages .HostIdentification .getDefaultInstance ();
3939 final AggregationMessage message = AggregationMessage .create (protobufMessage );
4040 Assert .assertNotNull (message );
4141 Assert .assertSame (protobufMessage , message .getMessage ());
@@ -60,14 +60,14 @@ public void testToBufferHostIdentification() {
6060
6161 @ Test (expected = IllegalArgumentException .class )
6262 public void testToBufferSerializedUnsupportedMessage () {
63- final GeneratedMessageV3 mockMessage = Mockito .mock (GeneratedMessageV3 .class );
63+ final GeneratedMessage mockMessage = Mockito .mock (GeneratedMessage .class );
6464 Mockito .doReturn (UnknownFieldSet .getDefaultInstance ()).when (mockMessage ).getUnknownFields ();
6565 AggregationMessage .create (mockMessage ).serializeToBuffer ();
6666 }
6767
6868 @ Test
6969 public void testToByteStringHostIdentification () {
70- final GeneratedMessageV3 protobufMessage = Messages .HostIdentification .getDefaultInstance ();
70+ final GeneratedMessage protobufMessage = Messages .HostIdentification .getDefaultInstance ();
7171 final AggregationMessage message = AggregationMessage .create (protobufMessage );
7272 Assert .assertNotNull (message );
7373 Assert .assertSame (protobufMessage , message .getMessage ());
@@ -92,14 +92,14 @@ public void testToByteStringHostIdentification() {
9292
9393 @ Test (expected = IllegalArgumentException .class )
9494 public void testToByteStringSerializedUnsupportedMessage () {
95- final GeneratedMessageV3 mockMessage = Mockito .mock (GeneratedMessageV3 .class );
95+ final GeneratedMessage mockMessage = Mockito .mock (GeneratedMessage .class );
9696 Mockito .doReturn (UnknownFieldSet .getDefaultInstance ()).when (mockMessage ).getUnknownFields ();
9797 AggregationMessage .create (mockMessage ).serializeToByteString ();
9898 }
9999
100100 @ Test
101101 public void testPekkoVsVertx () {
102- final GeneratedMessageV3 protobufMessage = Messages .HostIdentification .getDefaultInstance ();
102+ final GeneratedMessage protobufMessage = Messages .HostIdentification .getDefaultInstance ();
103103 final AggregationMessage message = AggregationMessage .create (protobufMessage );
104104 Assert .assertNotNull (message );
105105 Assert .assertSame (protobufMessage , message .getMessage ());
0 commit comments