@@ -20,24 +20,25 @@ import kotlin.test.Test
20
20
import kotlin.test.assertContentEquals
21
21
import kotlin.test.assertEquals
22
22
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 {
27
27
return Buffer ().apply { writeString(value.value) }
28
28
}
29
29
30
- override fun decodeFromSource (stream : Source ): Message {
31
- return Message (stream.readString())
30
+ override fun decodeFromSource (stream : Source ): CustomResolverMessage {
31
+ return CustomResolverMessage (stream.readString())
32
32
}
33
33
}
34
34
}
35
35
36
+
36
37
@Grpc
37
38
interface GrpcService {
38
39
suspend fun plainString (value : String ): String
39
40
40
- suspend fun message (value : Message ): Message
41
+ suspend fun message (value : CustomResolverMessage ): CustomResolverMessage
41
42
42
43
suspend fun krpc173 ()
43
44
@@ -53,8 +54,8 @@ class GrpcServiceImpl : GrpcService {
53
54
return " $value $value "
54
55
}
55
56
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} " )
58
59
}
59
60
60
61
override suspend fun krpc173 () {
@@ -88,7 +89,7 @@ class CustomResolverGrpcServiceTest : BaseGrpcServiceTest() {
88
89
resolver = simpleResolver,
89
90
impl = GrpcServiceImpl (),
90
91
) { service ->
91
- assertEquals(" test test" , service.message(Message (" test" )).value)
92
+ assertEquals(" test test" , service.message(CustomResolverMessage (" test" )).value)
92
93
}
93
94
94
95
@Test
0 commit comments