(#28) Added getCharacterStream of substring

citrusframework · Mar 22, 2019 · 4c8a5f8 · 4c8a5f8
1 parent 5ff47a7
commit 4c8a5f8
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 0 deletions.
diff --git a/driver/src/main/java/com/consol/citrus/db/driver/data/CitrusClob.java b/driver/src/main/java/com/consol/citrus/db/driver/data/CitrusClob.java
@@ -132,6 +132,21 @@ public void free() {
 
     @Override
     public Reader getCharacterStream(final long pos, final long length) throws SQLException {
+        if(pos < 1){
+            throw new SQLException("position is < 1 but CLOB starts at 1.");
+        }
+
+        if(pos + length > stringBuilder.length()){
+            throw new SQLException("CLOB references exceeds actual CLOB size");
+        }
+
+        final long posWithOffset = applyOffset(pos);
+        if(fitsInInt(posWithOffset + length)){
+            final int intPos = (int) posWithOffset;
+            final String substring = stringBuilder.substring(intPos, intPos + (int)length);
+            return new StringReader(substring);
+        }
+
         return null;
     }
 

diff --git a/driver/src/test/java/com/consol/citrus/db/driver/data/CitrusClobTest.java b/driver/src/test/java/com/consol/citrus/db/driver/data/CitrusClobTest.java
@@ -8,6 +8,7 @@
 
 import java.io.InputStream;
 import java.io.OutputStream;
+import java.io.Reader;
 import java.io.Writer;
 import java.nio.charset.Charset;
 import java.sql.Clob;
@@ -299,6 +300,41 @@ public void testFree() throws SQLException {
         assertEquals(citrusClob.length(), 0);
     }
 
+    @Test
+    public void testGetCharacterStream() throws Exception {
+
+        //GIVEN
+        citrusClob.setString(1, sampleText);
+        final String expectedStreamContent = "calm and";
+
+        //WHEN
+        final Reader characterStream = citrusClob.getCharacterStream(6, 8);
+
+        //THEN
+        final String streamContent = IOUtils.toString(characterStream);
+        assertEquals(streamContent, expectedStreamContent);
+    }
+
+    @Test(expectedExceptions = SQLException.class)
+    public void testGetCharacterStreamThrowsExceptionOnInvalidPosition() throws SQLException {
+
+        //WHEN
+        citrusClob.getCharacterStream(0, 5);
+
+        //THEN
+        //ExceptionIsThrown
+    }
+
+    @Test(expectedExceptions = SQLException.class)
+    public void testGetCharacterStreamThrowsExceptionWhenReferenceExceedsClobSize() throws SQLException {
+
+        //WHEN
+        citrusClob.getCharacterStream(5, 42);
+
+        //THEN
+        //ExceptionIsThrown
+    }
+
     @Test
     public void testEqualsContract(){
         EqualsVerifier