Added Tests for Binary Mappings

Author: Philipp Wagner

JSqlServerBulkInsert/src/main/java/de/bytefish/jsqlserverbulkinsert/SqlServerBulkInsert.java

@@ -117,6 +117,10 @@ protected void mapBigInt(String columnName, Func2<TEntity, BigInteger> propertyG
         addColumn(columnName, Types.BIGINT, wrapper);
     }
 
+    protected void mapBigIntLong(String columnName, Func2<TEntity, Long> propertyGetter) {
+        addColumn(columnName, Types.BIGINT, propertyGetter);
+    }
+
     protected void mapDate(String columnName, Func2<TEntity, LocalDate> propertyGetter) {
         addColumn(columnName, Types.DATE, propertyGetter);
     }
@@ -153,6 +157,10 @@ protected void mapString(String columnName, Func2<TEntity, String> propertyGette
         addColumn(columnName, Types.NVARCHAR, propertyGetter);
     }
 
+    protected void mapVarBinary(String columnName, int maxLength, Func2<TEntity, byte[]> propertyGetter) {
+        addColumn(columnName, Types.VARBINARY, maxLength, 0, false, propertyGetter);
+    }
+
     private void addColumn(String name, int type, boolean isAutoIncrement, Func2<TEntity, Object> propertyGetter)
     {
         // Create the current Column Meta Data:

JSqlServerBulkInsert/src/main/java/de/bytefish/jsqlserverbulkinsert/util/BinaryUtils.java

@@ -0,0 +1,8 @@
+// Copyright (c) Philipp Wagner. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+
+package de.bytefish.jsqlserverbulkinsert.util;
+
+public class BinaryUtils {
+
+}

JSqlServerBulkInsert/src/test/java/de/bytefish/jsqlserverbulkinsert/test/mapping/BigIntegerLongMappingTest.java

@@ -0,0 +1,83 @@
+// Copyright (c) Philipp Wagner. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+
+package de.bytefish.jsqlserverbulkinsert.test.mapping;
+
+import de.bytefish.jsqlserverbulkinsert.SqlServerBulkInsert;
+import de.bytefish.jsqlserverbulkinsert.test.base.TransactionalTestBase;
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.math.BigInteger;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.Arrays;
+import java.util.List;
+
+public class BigIntegerLongMappingTest extends TransactionalTestBase {
+
+    private class BigIntegerEntity extends SampleEntity<Long> {
+
+        public BigIntegerEntity(Long value) {
+            super(value);
+        }
+    }
+
+    private class BigIntegerInsert extends SqlServerBulkInsert<BigIntegerEntity> {
+
+        public BigIntegerInsert() {
+            super("dbo", "UnitTest");
+
+            mapBigIntLong("BigIntegerValue", BigIntegerEntity::getValue);
+        }
+
+    }
+
+    @Override
+    protected void onSetUpInTransaction() throws Exception {
+        createTestTable();
+    }
+
+    @Test
+    public void bulkInsertPersonDataTest() throws SQLException {
+        long longValue = 47878778228484l;
+        // Create the Value:
+        List<BigIntegerEntity> entities = Arrays.asList(new BigIntegerEntity(longValue));
+        // Create the BulkInserter:
+        BigIntegerInsert localDateInsert = new BigIntegerInsert();
+        // Now save all entities of a given stream:
+        localDateInsert.saveAll(connection, entities.stream());
+        // And assert all have been written to the database:
+        ResultSet rs = getAll();
+        // We have a Value:
+        Assert.assertEquals(true, rs.next());
+        // Get the Date we have written:
+        long resultBigIntegerValue = rs.getLong("BigIntegerValue");
+        // Assert both are equal:
+        Assert.assertEquals(longValue, resultBigIntegerValue);
+        // Assert only one record was read:
+        Assert.assertEquals(false, rs.next());
+    }
+
+    private ResultSet getAll() throws SQLException {
+
+        String sqlStatement = "SELECT * FROM dbo.UnitTest";
+
+        Statement statement = connection.createStatement();
+
+        return statement.executeQuery(sqlStatement);
+    }
+
+    private void createTestTable() throws SQLException {
+        String sqlStatement = "CREATE TABLE [dbo].[UnitTest]\n" +
+                "            (\n" +
+                "                BigIntegerValue bigint\n" +
+                "            );";
+
+        Statement statement = connection.createStatement();
+
+        statement.execute(sqlStatement);
+    }
+
+}

JSqlServerBulkInsert/src/test/java/de/bytefish/jsqlserverbulkinsert/test/mapping/BinaryMappingTest.java

@@ -0,0 +1,86 @@
+// Copyright (c) Philipp Wagner. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+
+package de.bytefish.jsqlserverbulkinsert.test.mapping;
+
+import de.bytefish.jsqlserverbulkinsert.SqlServerBulkInsert;
+import de.bytefish.jsqlserverbulkinsert.test.base.TransactionalTestBase;
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.Arrays;
+import java.util.List;
+
+public class BinaryMappingTest extends TransactionalTestBase {
+
+    private class BinaryDataEntity extends SampleEntity<byte[]> {
+
+        public BinaryDataEntity(byte[] value) {
+            super(value);
+        }
+    }
+
+    private class ByteInsert extends SqlServerBulkInsert<BinaryDataEntity> {
+
+        public ByteInsert() {
+            super("dbo", "UnitTest");
+
+            mapVarBinary("ByteValue", 255, BinaryDataEntity::getValue);
+        }
+
+    }
+
+    @Override
+    protected void onSetUpInTransaction() throws Exception {
+        createTestTable();
+    }
+
+    @Test
+    public void bulkInsertPersonDataTest() throws SQLException {
+        byte[] binaryDataValue = new byte[] { 1, 2, 3};
+        // Create the Value:
+        List<BinaryDataEntity> entities = Arrays.asList(new BinaryDataEntity(binaryDataValue));
+        // Create the BulkInserter:
+        ByteInsert localDateInsert = new ByteInsert();
+        // Now save all entities of a given stream:
+        localDateInsert.saveAll(connection, entities.stream());
+        // And assert all have been written to the database:
+        ResultSet rs = getAll();
+        // We have a Value:
+        Assert.assertEquals(true, rs.next());
+        // Get the Date we have written:
+        byte[] resultByteValue = rs.getBytes("ByteValue");
+        // Assert both are equal:
+        Assert.assertEquals(binaryDataValue.length, resultByteValue.length);
+        // Check Content:
+        for(int i = 0; i < resultByteValue.length; i++) {
+            Assert.assertEquals((byte) binaryDataValue[i], (byte) resultByteValue[i]);
+        }
+        // Assert only one record was read:
+        Assert.assertEquals(false, rs.next());
+    }
+
+    private ResultSet getAll() throws SQLException {
+
+        String sqlStatement = "SELECT * FROM dbo.UnitTest";
+
+        Statement statement = connection.createStatement();
+
+        return statement.executeQuery(sqlStatement);
+    }
+
+    private void createTestTable() throws SQLException {
+        String sqlStatement = "CREATE TABLE [dbo].[UnitTest]\n" +
+                "            (\n" +
+                "                ByteValue VARBINARY(255)\n" +
+                "            );";
+
+        Statement statement = connection.createStatement();
+
+        statement.execute(sqlStatement);
+    }
+
+}

JSqlServerBulkInsert/src/test/java/de/bytefish/jsqlserverbulkinsert/test/mapping/IntMappingTest.java

@@ -40,7 +40,7 @@ protected void onSetUpInTransaction() throws Exception {
 
     @Test
     public void bulkInsertPersonDataTest() throws SQLException {
-        Integer IntegerValue = 3412358;
+        Integer IntegerValue = 214483647;
         // Create the Value:
         List<IntegerEntity> entities = Arrays.asList(new IntegerEntity(IntegerValue));
         // Create the BulkInserter: