(shardingsphere) branch master updated: Add more test cases on StreamMergedResult (#33183)

[zhangliang]

This is an automated email from the ASF dual-hosted git repository.

zhangliang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git


The following commit(s) were added to refs/heads/master by this push:
     new 13daacc1b1b Add more test cases on StreamMergedResult (#33183)
13daacc1b1b is described below

commit 13daacc1b1b26858eb0f247a68d8120938c9eb23
Author: Liang Zhang <zhangli...@apache.org>
AuthorDate: Wed Oct 9 13:03:32 2024 +0800

    Add more test cases on StreamMergedResult (#33183)
---
 .../merge/result/impl/stream/StreamMergedResultTest.java | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git 
a/infra/merge/src/test/java/org/apache/shardingsphere/infra/merge/result/impl/stream/StreamMergedResultTest.java
 
b/infra/merge/src/test/java/org/apache/shardingsphere/infra/merge/result/impl/stream/StreamMergedResultTest.java
index 38b5823d2e2..4e7a9d9ec16 100644
--- 
a/infra/merge/src/test/java/org/apache/shardingsphere/infra/merge/result/impl/stream/StreamMergedResultTest.java
+++ 
b/infra/merge/src/test/java/org/apache/shardingsphere/infra/merge/result/impl/stream/StreamMergedResultTest.java
@@ -22,6 +22,7 @@ import 
org.apache.shardingsphere.infra.merge.result.impl.stream.fixture.TestStre
 import org.junit.jupiter.api.Test;
 
 import java.io.InputStream;
+import java.io.Reader;
 import java.sql.Date;
 import java.sql.SQLException;
 import java.util.Calendar;
@@ -62,10 +63,19 @@ class StreamMergedResultTest {
     @Test
     void assertGetInputStream() throws SQLException {
         QueryResult queryResult = mock(QueryResult.class);
-        InputStream value = mock(InputStream.class);
-        when(queryResult.getInputStream(1, "Ascii")).thenReturn(value);
+        InputStream inputStream = mock(InputStream.class);
+        when(queryResult.getInputStream(1, "Ascii")).thenReturn(inputStream);
         streamMergedResult.setCurrentQueryResult(queryResult);
-        assertThat(streamMergedResult.getInputStream(1, "Ascii"), is(value));
+        assertThat(streamMergedResult.getInputStream(1, "Ascii"), 
is(inputStream));
+    }
+    
+    @Test
+    void assertGetCharacterStream() throws SQLException {
+        QueryResult queryResult = mock(QueryResult.class);
+        Reader reader = mock(Reader.class);
+        when(queryResult.getCharacterStream(1)).thenReturn(reader);
+        streamMergedResult.setCurrentQueryResult(queryResult);
+        assertThat(streamMergedResult.getCharacterStream(1), is(reader));
     }
     
     @Test