Add ITs for long lists, byte arrays and strings

Author: lutovich

driver/src/test/java/org/neo4j/driver/v1/integration/ParametersIT.java

@@ -24,6 +24,7 @@
 
 import java.io.IOException;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 import java.util.concurrent.ThreadLocalRandom;
 
@@ -35,11 +36,15 @@
 import org.neo4j.driver.v1.exceptions.ClientException;
 import org.neo4j.driver.v1.exceptions.ServiceUnavailableException;
 import org.neo4j.driver.v1.util.TestNeo4jSession;
+import org.neo4j.driver.v1.util.TestUtil;
 
+import static java.util.Collections.singletonMap;
+import static java.util.stream.Collectors.toList;
 import static org.hamcrest.CoreMatchers.containsString;
 import static org.hamcrest.CoreMatchers.equalTo;
 import static org.hamcrest.CoreMatchers.instanceOf;
 import static org.hamcrest.MatcherAssert.assertThat;
+import static org.junit.Assert.assertArrayEquals;
 import static org.junit.Assert.fail;
 import static org.junit.Assume.assumeFalse;
 import static org.junit.Assume.assumeTrue;
@@ -52,6 +57,8 @@
 
 public class ParametersIT
 {
+    private static final int LONG_VALUE_SIZE = 1_000_000;
+
     @Rule
     public TestNeo4jSession session = new TestNeo4jSession();
 
@@ -451,6 +458,35 @@ public void shouldNotBePossibleToUsePathAsParameterViaMapValue()
         expectIOExceptionWithMessage( mapValue, "Unknown type: PATH" );
     }
 
+    @Test
+    public void shouldSendAndReceiveLongString()
+    {
+        String string = TestUtil.randomString( LONG_VALUE_SIZE );
+        testSendAndReceiveValue( string );
+    }
+
+    @Test
+    public void shouldSendAndReceiveLongListOfLongs()
+    {
+        List<Long> longs = ThreadLocalRandom.current()
+                .longs( LONG_VALUE_SIZE )
+                .boxed()
+                .collect( toList() );
+
+        testSendAndReceiveValue( longs );
+    }
+
+    @Test
+    public void shouldSendAndReceiveLongArrayOfBytes()
+    {
+        assumeTrue( supportsBytes() );
+
+        byte[] bytes = new byte[LONG_VALUE_SIZE];
+        ThreadLocalRandom.current().nextBytes( bytes );
+
+        testSendAndReceiveValue( bytes );
+    }
+
     private void testBytesProperty( byte[] array )
     {
         assumeTrue( supportsBytes() );
@@ -509,4 +545,11 @@ private void expectIOExceptionWithMessage( Value value, String message )
             fail( "Expecting a ServiceUnavailableException but got " + e );
         }
     }
+
+    private void testSendAndReceiveValue( Object value )
+    {
+        StatementResult result = session.run( "RETURN $value", singletonMap( "value", value ) );
+        Object receivedValue = result.single().get( 0 ).asObject();
+        assertArrayEquals( new Object[]{value}, new Object[]{receivedValue} );
+    }
 }

driver/src/test/java/org/neo4j/driver/v1/tck/DriverComplianceSteps.java

@@ -29,7 +29,6 @@
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
-import java.util.Random;
 
 import org.neo4j.driver.v1.Record;
 import org.neo4j.driver.v1.Session;
@@ -39,11 +38,12 @@
 import org.neo4j.driver.v1.tck.tck.util.runners.CypherStatementRunner;
 import org.neo4j.driver.v1.tck.tck.util.runners.StatementRunner;
 import org.neo4j.driver.v1.tck.tck.util.runners.StringRunner;
+import org.neo4j.driver.v1.util.TestUtil;
 
 import static org.hamcrest.core.IsEqual.equalTo;
 import static org.junit.Assert.assertThat;
-import static org.neo4j.driver.v1.tck.Environment.driver;
 import static org.neo4j.driver.v1.Values.parameters;
+import static org.neo4j.driver.v1.tck.Environment.driver;
 import static org.neo4j.driver.v1.tck.Environment.expectedBoltValue;
 import static org.neo4j.driver.v1.tck.Environment.expectedJavaValue;
 import static org.neo4j.driver.v1.tck.Environment.listOfObjects;
@@ -77,9 +77,9 @@ public void a_value( String value ) throws Throwable
     }
 
     @Given( "^a String of size (\\d+)$" )
-    public void a_String_of_size( long size ) throws Throwable
+    public void a_String_of_size( int size ) throws Throwable
     {
-        expectedJavaValue = getRandomString( size );
+        expectedJavaValue = TestUtil.randomString( size );
         expectedBoltValue = Values.value( expectedJavaValue );
     }
 
@@ -191,18 +191,6 @@ public void result_should_be_equal_to_a_single_Type_of_Input() throws Throwable
         }
     }
 
-    public String getRandomString( long size )
-    {
-        StringBuilder stringBuilder = new StringBuilder();
-        String alphabet = "abcdefghijklmnopqrstuvwxyz";
-        Random random = new Random();
-        while ( size-- > 0 )
-        {
-            stringBuilder.append( alphabet.charAt( random.nextInt( alphabet.length() ) ) );
-        }
-        return stringBuilder.toString();
-    }
-
     public List<Object> getListOfRandomsOfTypes( Type type, long size )
     {
         List<Object> list = new ArrayList<>();

driver/src/test/java/org/neo4j/driver/v1/util/TestUtil.java

@@ -29,6 +29,7 @@
 import java.util.concurrent.CompletionStage;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.Future;
+import java.util.concurrent.ThreadLocalRandom;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
 import java.util.function.BooleanSupplier;
@@ -57,6 +58,7 @@
 public final class TestUtil
 {
     private static final long DEFAULT_WAIT_TIME_MS = MINUTES.toMillis( 1 );
+    private static final String ALPHANUMERICS = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz123456789";
 
     private TestUtil()
     {
@@ -257,6 +259,17 @@ public static void awaitCondition( BooleanSupplier condition, long value, TimeUn
         }
     }
 
+    public static String randomString( int size )
+    {
+        StringBuilder sb = new StringBuilder( size );
+        ThreadLocalRandom random = ThreadLocalRandom.current();
+        for ( int i = 0; i < size; i++ )
+        {
+            sb.append( ALPHANUMERICS.charAt( random.nextInt( ALPHANUMERICS.length() ) ) );
+        }
+        return sb.toString();
+    }
+
     private static void setupSuccessfulPullAll( Connection connection, String statement )
     {
         doAnswer( invocation ->