add tests for varint compatability with parseDelimitedFrom and writeDelimitedTo

Author: Scottmitch

servicetalk-data-protobuf/build.gradle

@@ -43,6 +43,7 @@ dependencies {
   testImplementation project(":servicetalk-test-resources")
   testImplementation project(":servicetalk-buffer-netty")
   testImplementation "org.junit.jupiter:junit-jupiter-api:$junit5Version"
+  testImplementation "org.junit.jupiter:junit-jupiter-params:$junit5Version"
   testImplementation "org.hamcrest:hamcrest-library:$hamcrestVersion"
   testImplementation "org.mockito:mockito-core:$mockitoCoreVersion"
 

servicetalk-data-protobuf/src/test/java/io/servicetalk/data/protobuf/VarIntLengthStreamingSerializerTest.java

@@ -0,0 +1,90 @@
+/*
+ * Copyright © 2021 Apple Inc. and the ServiceTalk project authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package io.servicetalk.data.protobuf;
+
+import io.servicetalk.buffer.api.Buffer;
+import io.servicetalk.serializer.utils.VarIntLengthStreamingSerializer;
+import io.servicetalk.serializer.utils.test.TestProtos.DummyMessage;
+
+import com.google.protobuf.Parser;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+
+import java.io.OutputStream;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.stream.Stream;
+
+import static io.servicetalk.buffer.api.Buffer.asInputStream;
+import static io.servicetalk.buffer.api.Buffer.asOutputStream;
+import static io.servicetalk.buffer.netty.BufferAllocators.DEFAULT_ALLOCATOR;
+import static io.servicetalk.concurrent.api.Publisher.from;
+import static io.servicetalk.concurrent.api.Publisher.fromIterable;
+import static io.servicetalk.serializer.utils.test.TestProtos.DummyMessage.newBuilder;
+import static io.servicetalk.serializer.utils.test.TestProtos.DummyMessage.parser;
+import static java.util.Arrays.asList;
+import static java.util.Collections.singletonList;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.contains;
+
+class VarIntLengthStreamingSerializerTest {
+    @ParameterizedTest(name = "pojos={0}")
+    @MethodSource("pojos")
+    void protobufWriteDelimitedToDeserialized(Collection<DummyMessage> msgs) throws Exception {
+        Parser<DummyMessage> parser = parser();
+        VarIntLengthStreamingSerializer<DummyMessage> serializer = new VarIntLengthStreamingSerializer<>(
+                ProtobufSerializerCache.INSTANCE.serializerDeserializer(parser), DummyMessage::getSerializedSize);
+
+        Buffer buffer = DEFAULT_ALLOCATOR.newBuffer();
+        OutputStream os = asOutputStream(buffer);
+        for (DummyMessage msg : msgs) {
+            msg.writeDelimitedTo(os);
+        }
+
+        assertThat(serializer.deserialize(from(buffer), DEFAULT_ALLOCATOR).toFuture().get(), contains(msgs.toArray()));
+    }
+
+    @ParameterizedTest(name = "pojos={0}")
+    @MethodSource("pojos")
+    void protobufParseDelimitedFromSerialized(Collection<DummyMessage> msgs) throws Exception {
+        Parser<DummyMessage> parser = parser();
+        VarIntLengthStreamingSerializer<DummyMessage> serializer = new VarIntLengthStreamingSerializer<>(
+                ProtobufSerializerCache.INSTANCE.serializerDeserializer(parser), DummyMessage::getSerializedSize);
+
+        Collection<Buffer> serialized = serializer.serialize(fromIterable(msgs), DEFAULT_ALLOCATOR)
+                .toFuture().get();
+
+        Collection<DummyMessage> deserialized = new ArrayList<>(serialized.size());
+        for (Buffer buf : serialized) {
+            deserialized.add(parser.parseDelimitedFrom(asInputStream(buf)));
+        }
+        assertThat(deserialized, contains(msgs.toArray()));
+    }
+
+    @SuppressWarnings("unused")
+    private static Stream<Arguments> pojos() {
+        return Stream.of(
+                Arguments.of(singletonList(newMsg("hello"))),
+                Arguments.of(asList(newMsg("hello"), newMsg("world"))),
+                Arguments.of(asList(newMsg("hello"), newMsg("world"), newMsg("!")))
+        );
+    }
+
+    private static DummyMessage newMsg(String msg) {
+        return newBuilder().setMessage(msg).build();
+    }
+}