fix test

apache · Feb 4, 2024 · 3d55930 · 3d55930
1 parent e83ac49
commit 3d55930
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/.../src/main/java/org/apache/seata/serializer/seata/protocol/AbstractResultMessageCodec.java b/.../src/main/java/org/apache/seata/serializer/seata/protocol/AbstractResultMessageCodec.java
@@ -40,6 +40,8 @@ public <T> void encode(T t, ByteBuf out) {
         String resultMsg = abstractResultMessage.getMsg();
         if (null != resultCode) {
             out.writeByte(resultCode.ordinal());
+        } else{
+            out.writeByte(ResultCode.values().length);
         }
         if (resultCode != ResultCode.Success) {
             if (StringUtils.isNotEmpty(resultMsg)) {
@@ -61,9 +63,11 @@ public <T> void encode(T t, ByteBuf out) {
     @Override
     public <T> void decode(T t, ByteBuffer in) {
         AbstractResultMessage abstractResultMessage = (AbstractResultMessage) t;
-
-        ResultCode resultCode = ResultCode.get(in.get());
-        abstractResultMessage.setResultCode(resultCode);
+        ResultCode resultCode = null;
+        if(in.get()<ResultCode.values().length){
+            resultCode = ResultCode.get(in.get());
+            abstractResultMessage.setResultCode(resultCode);
+        }
         if (resultCode != ResultCode.Success) {
             short len = in.getShort();
             if (len > 0) {

diff --git a/...est/java/org/apache/seata/serializer/seata/protocol/RegisterTMResponseSerializerTest.java b/...est/java/org/apache/seata/serializer/seata/protocol/RegisterTMResponseSerializerTest.java
@@ -57,4 +57,15 @@ public void test_codec(){
 //        Assert.assertEquals(registerTMResponse2.getMsg(), registerTMResponse.getMsg());
 //        Assert.assertEquals(registerTMResponse2.getByCode(), registerTMResponse.getByCode());
     }
+
+    @Test
+    public void test_codec1() {
+        RegisterTMResponse registerTMResponse = new RegisterTMResponse();
+        registerTMResponse.setIdentified(true);
+        registerTMResponse.setVersion("2.1.0-SNAPSHOT");
+        byte[] bytes = seataSerializer.serialize(registerTMResponse);
+        RegisterTMResponse registerTMResponse2 = seataSerializer.deserialize(bytes);
+        assertThat(registerTMResponse2.isIdentified()).isEqualTo(registerTMResponse.isIdentified());
+        assertThat(registerTMResponse2.getVersion()).isEqualTo(registerTMResponse.getVersion());
+    }
 }