Fix flaky test

alibaba · Oct 17, 2024 · 4d71c70 · 4d71c70
1 parent 949c422
commit 4d71c70
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/core/src/test/java/com/alibaba/fastjson2/codec/JSONBTableTest3.java b/core/src/test/java/com/alibaba/fastjson2/codec/JSONBTableTest3.java
@@ -2,11 +2,17 @@
 
 import com.alibaba.fastjson2.JSON;
 import com.alibaba.fastjson2.JSONB;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import org.junit.jupiter.api.Test;
 
+import java.io.IOException;
+import java.util.Map;
+
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
 public class JSONBTableTest3 {
+    final ObjectMapper mapper = new ObjectMapper();
+
     @Test
     public void test_0() {
         A a = new A();
@@ -26,7 +32,7 @@ public void test_0() {
 
         byte[] bytes = JSONB.toBytes(a);
         A a1 = JSONB.parseObject(bytes, A.class);
-        assertEquals(JSON.toJSONString(a), JSON.toJSONString(a1));
+        assertMapsEqual(a, a1);
     }
 
     @Test
@@ -49,7 +55,17 @@ public void test_1() {
 
         byte[] bytes = JSONB.toBytes(a);
         A a1 = JSONB.parseObject(bytes, A.class);
-        assertEquals(JSON.toJSONString(a), JSON.toJSONString(a1));
+        assertMapsEqual(a, a1);
+    }
+
+    private void assertMapsEqual(A a, A a1) {
+        try {
+            Map<String, Object> mapa = mapper.readValue(JSON.toJSONString(a), Map.class);
+            Map<String, Object> mapa1 = mapper.readValue(JSON.toJSONString(a1), Map.class);
+            assertEquals(mapa, mapa1);
+        } catch (IOException e) {
+            throw new RuntimeException("Failed to read value as Map", e);
+        }
     }
 
     public static class A {