Fix test

AlbumenJ · Mar 21, 2024 · 584e139 · 584e139
1 parent 9a6352f
commit 584e139
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/...2/src/test/java/org/apache/dubbo/common/serialize/hessian2/Hessian2SerializationTest.java b/...2/src/test/java/org/apache/dubbo/common/serialize/hessian2/Hessian2SerializationTest.java
@@ -206,7 +206,7 @@ void testReadEvent() throws IOException, ClassNotFoundException {
     }
 
     @Test
-    void testReadByte() throws IOException {
+    void testReadByte() throws IOException, ClassNotFoundException {
         FrameworkModel frameworkModel = new FrameworkModel();
         Serialization serialization =
                 frameworkModel.getExtensionLoader(Serialization.class).getExtension("hessian2");
@@ -219,6 +219,19 @@ void testReadByte() throws IOException {
             objectOutput.writeObject((byte) 11);
             objectOutput.flushBuffer();
 
+            byte[] bytes = outputStream.toByteArray();
+            ByteArrayInputStream inputStream = new ByteArrayInputStream(bytes);
+            ObjectInput objectInput = serialization.deserialize(url, inputStream);
+            Assertions.assertEquals((byte) 11, objectInput.readObject());
+        }
+
+        // write byte, read byte
+        {
+            ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
+            ObjectOutput objectOutput = serialization.serialize(url, outputStream);
+            objectOutput.writeByte((byte) 11);
+            objectOutput.flushBuffer();
+
             byte[] bytes = outputStream.toByteArray();
             ByteArrayInputStream inputStream = new ByteArrayInputStream(bytes);
             ObjectInput objectInput = serialization.deserialize(url, inputStream);