Raise Jvm test coverage

Author: brudaswen

library/src/jvmTest/kotlin/kotlinx/serialization/csv/example/CsvJvmTest.kt

@@ -0,0 +1,98 @@
+package kotlinx.serialization.csv.example
+
+import kotlinx.coroutines.test.runTest
+import kotlinx.serialization.ExperimentalSerializationApi
+import kotlinx.serialization.builtins.ListSerializer
+import kotlinx.serialization.csv.Csv
+import kotlinx.serialization.csv.decodeFrom
+import kotlinx.serialization.csv.encodeTo
+import kotlinx.serialization.csv.example.Tire.Axis.FRONT
+import kotlinx.serialization.csv.example.Tire.Side.LEFT
+import kotlinx.serialization.csv.recordReader
+import kotlinx.serialization.csv.recordWriter
+import kotlinx.serialization.modules.SerializersModule
+import java.io.ByteArrayInputStream
+import java.io.ByteArrayOutputStream
+import kotlin.io.bufferedReader
+import kotlin.io.bufferedWriter
+import kotlin.test.Test
+import kotlin.test.assertEquals
+import kotlin.use
+import kotlin.uuid.Uuid
+
+@OptIn(ExperimentalSerializationApi::class)
+internal class CsvJvmTest {
+
+    private val tasla = Vehicle(
+        uuid = Uuid.parse("f9682dcb-30f7-4e88-915e-60e3b2758da7"),
+        type = VehicleType.CAR,
+        brand = "Tesla",
+    )
+
+    private val porsche = Vehicle(
+        uuid = Uuid.parse("5e1afd88-97a2-4373-a83c-44a49c552abd"),
+        type = VehicleType.CAR,
+        brand = "Porsche",
+    )
+
+    private val testData = listOf(
+        VehiclePartRecord(101, tasla, Tire(FRONT, LEFT, 245, 35, 21), 0.25),
+        VehiclePartRecord(201, porsche, Oil(20, 50), 0.2),
+    )
+
+    private val csv = Csv {
+        serializersModule = SerializersModule {
+            polymorphic(Part::class, Tire::class, Tire.serializer())
+            polymorphic(Part::class, Oil::class, Oil.serializer())
+        }
+    }
+
+    private val serializer = ListSerializer(VehiclePartRecord.serializer())
+
+    @Test
+    fun test_Csv_encodeTo_InputOutputStream() = runTest {
+        val output = ByteArrayOutputStream().use { output ->
+            csv.encodeTo(serializer, testData, output)
+            output.toByteArray()
+        }
+
+        val result = ByteArrayInputStream(output).use { input ->
+            csv.decodeFrom(serializer, input).toList()
+        }
+
+        assertEquals(testData, result)
+    }
+
+    @Test
+    fun test_Csv_decodeFrom_ReaderWriter() = runTest {
+        val output = ByteArrayOutputStream().use { output ->
+            output.bufferedWriter().use { writer ->
+                csv.encodeTo(serializer, testData, writer)
+            }
+            output.toByteArray()
+        }
+
+        val result = ByteArrayInputStream(output).use { input ->
+            csv.decodeFrom(serializer, input.bufferedReader()).toList()
+        }
+
+        assertEquals(testData, result)
+    }
+
+    @Test
+    fun test_Csv_recordReaderWriter_InputOutputStream() = runTest {
+        val serializer = VehiclePartRecord.serializer()
+
+        val output = ByteArrayOutputStream().use { output ->
+            val writer = csv.recordWriter(serializer, output)
+            testData.forEach(writer::write)
+            output.toByteArray()
+        }
+
+        val result = ByteArrayInputStream(output).use { input ->
+            csv.recordReader(serializer, input).asSequence().toList()
+        }
+
+        assertEquals(testData, result)
+    }
+}

library/src/jvmTest/kotlin/kotlinx/serialization/csv/example/SteamingTest.kt

@@ -12,8 +12,6 @@ import kotlinx.serialization.csv.example.Tire.Side.LEFT
 import kotlinx.serialization.csv.example.Tire.Side.RIGHT
 import kotlinx.serialization.csv.recordReader
 import kotlinx.serialization.csv.recordWriter
-import kotlinx.serialization.csv.sink.CsvSink
-import kotlinx.serialization.csv.source.CsvSource
 import kotlinx.serialization.modules.SerializersModule
 import java.io.PipedReader
 import java.io.PipedWriter
@@ -67,8 +65,8 @@ class SteamingTest {
         val input = PipedReader()
         val output = PipedWriter(input)
 
-        val writer = csv.recordWriter(VehiclePartRecord.serializer(), CsvSink(output))
-        val reader = csv.recordReader(VehiclePartRecord.serializer(), CsvSource(input))
+        val writer = csv.recordWriter(VehiclePartRecord.serializer(), output)
+        val reader = csv.recordReader(VehiclePartRecord.serializer(), input)
 
         val readerTask = async(Dispatchers.IO) {
             reader.asSequence().toList()
@@ -108,15 +106,15 @@ class SteamingTest {
         val output = PipedWriter(input)
 
         val readerTask = async(Dispatchers.IO) {
-            val reader = csv.recordReader(Tire.serializer(), CsvSource(input))
+            val reader = csv.recordReader(Tire.serializer(), input)
             reader.asSequence().onEach {
                 println("Read $it")
             }.toList()
         }
 
         val writerTask = async(Dispatchers.IO) {
             output.buffered().use { output ->
-                val writer = csv.recordWriter(Tire.serializer(), CsvSink(output))
+                val writer = csv.recordWriter(Tire.serializer(), output)
                 testData.forEach {
                     println("Writing $it")
                     writer.write(it)