grpc-pb: Revert removal of grpc service tests for JVM

Signed-off-by: Johannes Zottele <[email protected]>

Author: Jozott00

grpc/grpc-core/src/jvmTest/kotlin/kotlinx/rpc/grpc/core/test/GrpcServerTest.kt

@@ -0,0 +1,37 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.rpc.grpc.core.test
+
+import kotlinx.coroutines.sync.Mutex
+import kotlinx.coroutines.sync.withLock
+import kotlinx.coroutines.test.runTest
+import kotlinx.rpc.RpcServer
+import kotlinx.rpc.grpc.GrpcClient
+import kotlinx.rpc.grpc.GrpcServer
+
+abstract class GrpcServerTest {
+    private val serverMutex = Mutex()
+
+    abstract fun RpcServer.registerServices()
+
+    protected fun runGrpcTest(test: suspend (GrpcClient) -> Unit, ) = runTest {
+        serverMutex.withLock {
+            val grpcClient = GrpcClient("localhost", 8080) {
+                usePlaintext()
+            }
+
+            val grpcServer = GrpcServer(8080, builder = {
+                registerServices()
+            })
+
+            grpcServer.start()
+            test(grpcClient)
+            grpcServer.shutdown()
+            grpcServer.awaitTermination()
+            grpcClient.shutdown()
+            grpcClient.awaitTermination()
+        }
+    }
+}

grpc/grpc-core/src/jvmTest/kotlin/kotlinx/rpc/grpc/core/test/StreamingTest.kt

@@ -0,0 +1,80 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.rpc.grpc.core.test
+
+import StreamingTestService
+import kotlinx.coroutines.flow.*
+import kotlinx.rpc.RpcServer
+import kotlinx.rpc.withService
+import kotlin.test.Test
+import kotlin.test.assertEquals
+
+class StreamingTestServiceImpl : StreamingTestService {
+    override fun Server(message: kotlinx.rpc.grpc.test.References): Flow<kotlinx.rpc.grpc.test.References> {
+        return flow { emit(message); emit(message); emit(message) }
+    }
+
+    override suspend fun Client(message: Flow<kotlinx.rpc.grpc.test.References>): kotlinx.rpc.grpc.test.References {
+        return message.last()
+    }
+
+    override fun Bidi(message: Flow<kotlinx.rpc.grpc.test.References>): Flow<kotlinx.rpc.grpc.test.References> {
+        return message
+    }
+}
+
+class StreamingTest : GrpcServerTest() {
+    override fun RpcServer.registerServices() {
+        registerService<StreamingTestService> { StreamingTestServiceImpl() }
+    }
+
+    @Test
+    fun testServerStreaming() = runGrpcTest { grpcClient ->
+        val service = grpcClient.withService<StreamingTestService>()
+        service.Server(kotlinx.rpc.grpc.test.References {
+            other = kotlinx.rpc.grpc.test.Other {
+                field = 42
+            }
+        }).toList().run {
+            assertEquals(3, size)
+
+            forEach {
+                assertEquals(42, it.other.field)
+            }
+        }
+    }
+
+    @Test
+    fun testClientStreaming() = runGrpcTest { grpcClient ->
+        val service = grpcClient.withService<StreamingTestService>()
+        val result = service.Client(flow {
+            repeat(3) {
+                emit(kotlinx.rpc.grpc.test.References {
+                    other = kotlinx.rpc.grpc.test.Other {
+                        field = 42 + it
+                    }
+                })
+            }
+        })
+
+        assertEquals(44, result.other.field)
+    }
+
+    @Test
+    fun testBidiStreaming() = runGrpcTest { grpcClient ->
+        val service = grpcClient.withService<StreamingTestService>()
+        service.Bidi(flow {
+            repeat(3) {
+                emit(kotlinx.rpc.grpc.test.References {
+                    other = kotlinx.rpc.grpc.test.Other {
+                        field = 42 + it
+                    }
+                })
+            }
+        }).collectIndexed { i, it ->
+            assertEquals(42 + i, it.other.field)
+        }
+    }
+}

grpc/grpc-core/src/jvmTest/kotlin/kotlinx/rpc/grpc/core/test/TestPrimitiveService.kt

@@ -0,0 +1,34 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.rpc.grpc.core.test
+
+import kotlinx.rpc.RpcServer
+import kotlinx.rpc.grpc.test.AllPrimitives
+import kotlinx.rpc.grpc.test.PrimitiveService
+import kotlinx.rpc.withService
+import kotlin.test.Test
+import kotlin.test.assertEquals
+
+class PrimitiveServiceImpl : PrimitiveService {
+    override suspend fun Echo(message: AllPrimitives): AllPrimitives {
+        return message
+    }
+}
+
+class TestPrimitiveService : GrpcServerTest() {
+    override fun RpcServer.registerServices() {
+        registerService<PrimitiveService> { PrimitiveServiceImpl() }
+    }
+
+    @Test
+    fun testPrimitive(): Unit = runGrpcTest { grpcClient ->
+        val service = grpcClient.withService<PrimitiveService>()
+        val result = service.Echo(AllPrimitives {
+            int32 = 42
+        })
+
+        assertEquals(42, result.int32)
+    }
+}