[TH2-4989] Corrected toBuilder interface for ParsedMessage

th2-net · Jul 19, 2023 · 78d1c2f · 78d1c2f
1 parent 260387c
commit 78d1c2f
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/...in/kotlin/com/exactpro/th2/common/schema/message/impl/rabbitmq/transport/ParsedMessage.kt b/...in/kotlin/com/exactpro/th2/common/schema/message/impl/rabbitmq/transport/ParsedMessage.kt
@@ -93,12 +93,9 @@ class ParsedMessage private constructor(
     }
 
     //TODO: add override annotation
-    fun toBuilder(): Builder<Builder<*>> {
-        return when {
-            body === DEFAULT_BODY -> FromRawBuilderImpl(bodySupplier).setRawBody(rawBody)
-            bodySupplier === DEFAULT_BODY_SUPPLIER -> FromMapBuilderImpl().setBody(body)
-            else -> error("Message can't be convert to builder $this")
-        }.apply {
+    fun toBuilder(): FromMapBuilder {
+        return FromMapBuilderImpl().apply {
+            setBody(body)
             setId(id)
             setMetadata(metadata)
             setProtocol(protocol)