Enable RawHandler test (#385)

restatedev · Nov 2, 2024 · 9ad0950 · 9ad0950
1 parent 2175ba2
commit 9ad0950
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/.github/workflows/integration.yaml b/.github/workflows/integration.yaml
@@ -48,7 +48,7 @@ jobs:
     name: "Features integration test (sdk-test-suite version ${{ matrix.sdk-test-suite }})"
     strategy:
       matrix:
-        sdk-test-suite: [ "2.0" ]
+        sdk-test-suite: [ "2.1" ]
     permissions:
       contents: read
       issues: read

diff --git a/test-services/src/main/kotlin/dev/restate/sdk/testservices/TestUtilsServiceImpl.kt b/test-services/src/main/kotlin/dev/restate/sdk/testservices/TestUtilsServiceImpl.kt
@@ -27,6 +27,10 @@ class TestUtilsServiceImpl : TestUtilsService {
     return context.request().headers()
   }
 
+  override suspend fun rawEcho(context: Context, input: ByteArray): ByteArray {
+    return input
+  }
+
   override suspend fun createAwakeableAndAwaitIt(
       ctx: Context,
       req: CreateAwakeableAndAwaitItRequest

diff --git a/test-services/src/main/kotlin/dev/restate/sdk/testservices/contracts/TestUtils.kt b/test-services/src/main/kotlin/dev/restate/sdk/testservices/contracts/TestUtils.kt
@@ -9,6 +9,7 @@
 package dev.restate.sdk.testservices.contracts
 
 import dev.restate.sdk.annotation.Handler
+import dev.restate.sdk.annotation.Raw
 import dev.restate.sdk.annotation.Service
 import dev.restate.sdk.kotlin.Context
 import kotlinx.serialization.SerialName
@@ -57,6 +58,9 @@ interface TestUtilsService {
   /** Echo ingress headers */
   @Handler suspend fun echoHeaders(context: Context): Map<String, String>
 
+  /** Just echo */
+  @Handler @Raw suspend fun rawEcho(context: Context, @Raw input: ByteArray): ByteArray
+
   /** Create an awakeable, register it to AwakeableHolder#hold, then await it. */
   @Handler
   suspend fun createAwakeableAndAwaitIt(