fix(iox): 修复空读取测试用例及简化错误处理逻辑

ecodeclub · Jan 24, 2025 · 3422dbc · 3422dbc
1 parent 8d92ff4
commit 3422dbc
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/iox/multiple_bytes_test.go b/iox/multiple_bytes_test.go
@@ -34,7 +34,7 @@ func TestMultipleBytesReadWrite(t *testing.T) {
 			name:     "空读取",
 			write:    []byte{},
 			readSize: 1,
-			wantRead: []byte{},
+			wantRead: []byte{0},
 			wantN:    0,
 			wantErr:  io.EOF,
 		},
@@ -73,13 +73,9 @@ func TestMultipleBytesReadWrite(t *testing.T) {
 
 			read := make([]byte, tc.readSize)
 			n, err = mb.Read(read)
-			if tc.wantErr != nil {
-				assert.Equal(t, tc.wantErr, err)
-			} else {
-				assert.Nil(t, err)
-				assert.Equal(t, tc.wantN, n, "读取的字节数应该等于期望的字节数")
-				assert.Equal(t, tc.wantRead, read, "读取的数据应该等于期望读取的数据")
-			}
+			assert.Equal(t, tc.wantErr, err)
+			assert.Equal(t, tc.wantN, n, "读取的字节数应该等于期望的字节数")
+			assert.Equal(t, tc.wantRead, read, "读取的数据应该等于期望读取的数据")
 		})
 	}
 }