(#28) Added getBinaryStream with limitation

citrusframework · Apr 17, 2019 · be92434 · be92434
1 parent 97d3a81
commit be92434
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/driver/src/main/java/com/consol/citrus/db/driver/data/CitrusBlob.java b/driver/src/main/java/com/consol/citrus/db/driver/data/CitrusBlob.java
@@ -99,7 +99,13 @@ public void free() {
 
     @Override
     public InputStream getBinaryStream(final long pos, final long length) {
-        return null;
+        if(lobUtils.fitsInInt(pos) && lobUtils.fitsInInt(length)){
+            final int intPosWithOffset = (int) lobUtils.applyOffset(pos);
+            final int endIndex = intPosWithOffset + (int) length;
+            final byte[] subarray = ArrayUtils.subarray(content, intPosWithOffset, endIndex);
+            return new ByteArrayInputStream(subarray);
+        }
+        return new ByteArrayInputStream(ArrayUtils.EMPTY_BYTE_ARRAY);
     }
 
     @Override

diff --git a/driver/src/test/java/com/consol/citrus/db/driver/data/CitrusBlobTest.java b/driver/src/test/java/com/consol/citrus/db/driver/data/CitrusBlobTest.java
@@ -161,4 +161,18 @@ public void testPositionWithBlobPattern() throws SQLException {
         //THEN
         assertEquals(position, 6);
     }
+
+    @Test
+    public void testGetLimitedBinaryStream() throws IOException {
+
+        //GIVEN
+        final byte[] expectedStreamContent = "Me Up".getBytes();
+        citrusBlob.setBytes(1, sampleBytes);
+
+        //WHEN
+        final InputStream limitedBinarySteam = citrusBlob.getBinaryStream(6, 5);
+
+        //THEN
+        assertEquals(IOUtils.toByteArray(limitedBinarySteam), expectedStreamContent);
+    }
 }