@@ -20,24 +20,25 @@ import kotlin.test.Test
2020import kotlin.test.assertContentEquals
2121import kotlin.test.assertEquals
2222
23- @WithCodec(Message .Companion ::class )
24- class Message (val value : String ) {
25- companion object : SourcedMessageCodec <Message > {
26- override fun encodeToSource (value : Message ): Source {
23+ @WithCodec(CustomResolverMessage .Companion ::class )
24+ class CustomResolverMessage (val value : String ) {
25+ companion object Companion : SourcedMessageCodec<CustomResolverMessage > {
26+ override fun encodeToSource (value : CustomResolverMessage ): Source {
2727 return Buffer ().apply { writeString(value.value) }
2828 }
2929
30- override fun decodeFromSource (stream : Source ): Message {
31- return Message (stream.readString())
30+ override fun decodeFromSource (stream : Source ): CustomResolverMessage {
31+ return CustomResolverMessage (stream.readString())
3232 }
3333 }
3434}
3535
36+
3637@Grpc
3738interface GrpcService {
3839 suspend fun plainString (value : String ): String
3940
40- suspend fun message (value : Message ): Message
41+ suspend fun message (value : CustomResolverMessage ): CustomResolverMessage
4142
4243 suspend fun krpc173 ()
4344
@@ -53,8 +54,8 @@ class GrpcServiceImpl : GrpcService {
5354 return " $value $value "
5455 }
5556
56- override suspend fun message (value : Message ): Message {
57- return Message (" ${value.value} ${value.value} " )
57+ override suspend fun message (value : CustomResolverMessage ): CustomResolverMessage {
58+ return CustomResolverMessage (" ${value.value} ${value.value} " )
5859 }
5960
6061 override suspend fun krpc173 () {
@@ -88,7 +89,7 @@ class CustomResolverGrpcServiceTest : BaseGrpcServiceTest() {
8889 resolver = simpleResolver,
8990 impl = GrpcServiceImpl (),
9091 ) { service ->
91- assertEquals(" test test" , service.message(Message (" test" )).value)
92+ assertEquals(" test test" , service.message(CustomResolverMessage (" test" )).value)
9293 }
9394
9495 @Test
0 commit comments