grpc-pb: Fix imports

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

Author: Jozott00

grpc/grpc-core/build.gradle.kts

@@ -7,6 +7,7 @@ import kotlinx.rpc.proto.kotlinMultiplatform
 import org.gradle.internal.extensions.stdlib.capitalized
 import org.jetbrains.kotlin.gradle.plugin.mpp.KotlinNativeTarget
 import org.jetbrains.kotlin.gradle.tasks.CInteropProcess
+import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
 
 plugins {
     alias(libs.plugins.conventions.kmp)
@@ -159,5 +160,12 @@ rpc {
                 dependsOn(gradle.includedBuild("protoc-gen").task(":jar"))
             }
         }
+
+        // generate protos before compiling tests
+        project.tasks.withType<KotlinCompile>().configureEach {
+            if (name.startsWith("compileTest")) {
+                dependsOn(project.tasks.withType<BufGenerateTask>())
+            }
+        }
     }
 }

protoc-gen/src/main/kotlin/kotlinx/rpc/protobuf/ModelToKotlinCommonGenerator.kt

@@ -78,6 +78,7 @@ class ModelToKotlinCommonGenerator(
 
             generateInternalDeclaredEntities(this@generateInternalKotlinFile)
 
+            import("kotlinx.rpc.grpc.pb.*")
             import("kotlinx.rpc.internal.utils.*")
             import("kotlinx.coroutines.flow.*")