StakeSquid/dshackle
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Write response upstreamId in streaming (#419)

Browse Source
This commit is contained in:
KirillPamPam
2024-02-16 12:57:38 +04:00
committed by GitHub
parent 9ca803b39d
commit e612a62c71
2 changed files with 156 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

132
src/test/kotlin/io/emeraldpay/dshackle/rpc/NativeCallTest.kt Normal file
View File

@@ -0,0 +1,132 @@
package io.emeraldpay.dshackle.rpc
import com.google.protobuf.ByteString
import io.emeraldpay.api.proto.BlockchainOuterClass.NativeCallReplyItem
import io.emeraldpay.api.proto.BlockchainOuterClass.NativeCallRequest
import io.emeraldpay.dshackle.config.MainConfig
import io.emeraldpay.dshackle.upstream.MultistreamHolder
import io.emeraldpay.dshackle.upstream.rpcclient.stream.Chunk
import io.emeraldpay.dshackle.upstream.signature.ResponseSigner
import org.junit.jupiter.api.Test
import org.mockito.kotlin.doReturn
import org.mockito.kotlin.mock
import org.mockito.kotlin.spy
import org.springframework.cloud.sleuth.Tracer
import reactor.core.publisher.Flux
import reactor.core.publisher.Mono
import reactor.test.StepVerifier
import java.time.Duration
class NativeCallTest {
@Test
fun `nativeCall always returns item with response upstreamId`() {
val request = Mono.just(NativeCallRequest.newBuilder().build())
val nativeCall = spy(
NativeCall(
mock<MultistreamHolder>(),
mock<ResponseSigner>(),
MainConfig(),
mock<Tracer>(),
),
) {
on { nativeCallResult(request) } doReturn Flux.just(
NativeCall.CallResult.ok(1, null, "0x1".toByteArray(), null, "id", null),
)
}
StepVerifier.create(nativeCall.nativeCall(request))
.expectNext(
NativeCallReplyItem.newBuilder()
.setUpstreamId("id")
.setId(1)
.setSucceed(true)
.setPayload(ByteString.copyFrom("0x1".toByteArray()))
.build(),
)
.expectComplete()
.verify(Duration.ofSeconds(1))
}
@Test
fun `nativeCall always returns error item with response upstreamId`() {
val request = Mono.just(NativeCallRequest.newBuilder().build())
val nativeCall = spy(
NativeCall(
mock<MultistreamHolder>(),
mock<ResponseSigner>(),
MainConfig(),
mock<Tracer>(),
),
) {
on { nativeCallResult(request) } doReturn Flux.just(
NativeCall.CallResult(1, null, null, NativeCall.CallError(50001, "message", null, null, "upId"), null, null),
)
}
StepVerifier.create(nativeCall.nativeCall(request))
.expectNext(
NativeCallReplyItem.newBuilder()
.setUpstreamId("upId")
.setId(1)
.setSucceed(false)
.setErrorMessage("message")
.setItemErrorCode(50001)
.build(),
)
.expectComplete()
.verify(Duration.ofSeconds(1))
}
@Test
fun `nativeCall always returns item with response upstreamId in the first chunk only`() {
val request = Mono.just(NativeCallRequest.newBuilder().build())
val chunks = Flux.just(
Chunk("0x1".toByteArray(), false),
Chunk("111".toByteArray(), false),
Chunk("222".toByteArray(), true),
)
val nativeCall = spy(
NativeCall(
mock<MultistreamHolder>(),
mock<ResponseSigner>(),
MainConfig(),
mock<Tracer>(),
),
) {
on { nativeCallResult(request) } doReturn Flux.just(
NativeCall.CallResult.ok(1, null, "".toByteArray(), null, "upId", null, chunks),
)
}
StepVerifier.create(nativeCall.nativeCall(request))
.expectNext(
NativeCallReplyItem.newBuilder()
.setUpstreamId("upId")
.setId(1)
.setChunked(true)
.setSucceed(true)
.setPayload(ByteString.copyFrom("0x1".toByteArray()))
.build(),
)
.expectNext(
NativeCallReplyItem.newBuilder()
.setId(1)
.setChunked(true)
.setSucceed(true)
.setPayload(ByteString.copyFrom("111".toByteArray()))
.build(),
)
.expectNext(
NativeCallReplyItem.newBuilder()
.setId(1)
.setChunked(true)
.setFinalChunk(true)
.setSucceed(true)
.setPayload(ByteString.copyFrom("222".toByteArray()))
.build(),
)
.expectComplete()
.verify(Duration.ofSeconds(1))
}
}