Skip to content

Commit

Permalink
Test index range seek for array in both General and Unique accessor
Browse files Browse the repository at this point in the history
  • Loading branch information
@burqen
burqen committed Sep 13, 2018
1 parent cc6d3b1 commit 45b82a0
Showing 1 changed file with 139 additions and 139 deletions.
278 changes: 139 additions & 139 deletions .../kernel-it/src/test/java/org/neo4j/kernel/api/index/SimpleIndexAccessorCompatibility.java
View file
Expand Up @@ -232,6 +232,145 @@ public void shouldScanAllValues() throws Exception
assertThat( result, containsInAnyOrder( allNodes ) ); assertThat( result, containsInAnyOrder( allNodes ) );
} }


@Test
public void testIndexRangeSeekByZonedDateTimeArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextDateTimeArray() );
}

@Test
public void testIndexRangeSeekByLocalDateTimeArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextLocalDateTimeArray() );
}

@Test
public void testIndexRangeSeekByDateArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextDateArray() );
}

@Test
public void testIndexRangeSeekByZonedTimeArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextTimeArray() );
}

@Test
public void testIndexRangeSeekByLocalTimeArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextLocalTimeArray() );
}

@Test
public void testIndexRangeSeekByDurationArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextDurationArray() );
}

@Test
public void testIndexRangeSeekByTextArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextBasicMultilingualPlaneStringArray() );
}

@Test
public void testIndexRangeSeekByBooleanArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextBooleanArray() );
}

@Test
public void testIndexRangeSeekByByteArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextByteArray() );
}

@Test
public void testIndexRangeSeekByShortArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextShortArray() );
}

@Test
public void testIndexRangeSeekByIntArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextIntArray() );
}

@Test
public void testIndexRangeSeekByLongArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextLongArray() );
}

@Test
public void testIndexRangeSeekByFloatArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextFloatArray() );
}

@Test
public void testIndexRangeSeekByDoubleArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextDoubleArray() );
}

private void testIndexRangeSeek( Supplier<Value> generator ) throws Exception
{
Assume.assumeTrue( testSuite.supportsGranularCompositeQueries() );

int count = random.nextInt( 5, 10 );
List<Value> values = new ArrayList<>();
List<IndexEntryUpdate<?>> updates = new ArrayList<>();
Set<Value> duplicateCheck = new HashSet<>();
for ( int i = 0; i < count; i++ )
{
Value value;
do
{
value = generator.get();
}
while ( !duplicateCheck.add( value ) );
values.add( value );
}
values.sort( Values.COMPARATOR );
for ( int i = 0; i < count; i++ )
{
updates.add( add( i + 1, descriptor.schema(), values.get( i ) ) );
}

updateAndCommit( updates );

for ( int f = 0; f < values.size(); f++ )
{
for ( int t = f; t < values.size(); t++ )
{
Value from = values.get( f );
Value to = values.get( t );
for ( boolean fromInclusive : new boolean[] {true, false} )
{
for ( boolean toInclusive : new boolean[] {true, false} )
{
assertThat( query( range( 1, from, fromInclusive, to, toInclusive ) ), equalTo( ids( f, fromInclusive, t, toInclusive ) ) );
}
}
}
}
}

private List<Long> ids( int fromIndex, boolean fromInclusive, int toIndex, boolean toInclusive )
{
List<Long> ids = new ArrayList<>();
int from = fromInclusive ? fromIndex : fromIndex + 1;
int to = toInclusive ? toIndex : toIndex - 1;
for ( int i = from; i <= to; i++ )
{
ids.add( (long) (i + 1) );
}
return ids;
}

// This behaviour is expected by General indexes // This behaviour is expected by General indexes


@Ignore( "Not a test. This is a compatibility suite" ) @Ignore( "Not a test. This is a compatibility suite" )
Expand Down Expand Up @@ -367,145 +506,6 @@ public void testIndexRangeSeekByDurationWithDuplicates() throws Exception
duration( 3, 1, 1, 1 ) ); duration( 3, 1, 1, 1 ) );
} }


@Test
public void testIndexRangeSeekByZonedDateTimeArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextDateTimeArray() );
}

@Test
public void testIndexRangeSeekByLocalDateTimeArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextLocalDateTimeArray() );
}

@Test
public void testIndexRangeSeekByDateArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextDateArray() );
}

@Test
public void testIndexRangeSeekByZonedTimeArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextTimeArray() );
}

@Test
public void testIndexRangeSeekByLocalTimeArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextLocalTimeArray() );
}

@Test
public void testIndexRangeSeekByDurationArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextDurationArray() );
}

@Test
public void testIndexRangeSeekByTextArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextBasicMultilingualPlaneStringArray() );
}

@Test
public void testIndexRangeSeekByBooleanArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextBooleanArray() );
}

@Test
public void testIndexRangeSeekByByteArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextByteArray() );
}

@Test
public void testIndexRangeSeekByShortArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextShortArray() );
}

@Test
public void testIndexRangeSeekByIntArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextIntArray() );
}

@Test
public void testIndexRangeSeekByLongArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextLongArray() );
}

@Test
public void testIndexRangeSeekByFloatArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextFloatArray() );
}

@Test
public void testIndexRangeSeekByDoubleArray() throws Exception
{
testIndexRangeSeek( () -> random.randomValues().nextDoubleArray() );
}

private void testIndexRangeSeek( Supplier<Value> generator ) throws Exception
{
Assume.assumeTrue( testSuite.supportsGranularCompositeQueries() );

int count = random.nextInt( 5, 10 );
List<Value> values = new ArrayList<>();
List<IndexEntryUpdate<?>> updates = new ArrayList<>();
Set<Value> duplicateCheck = new HashSet<>();
for ( int i = 0; i < count; i++ )
{
Value value;
do
{
value = generator.get();
}
while ( !duplicateCheck.add( value ) );
values.add( value );
}
values.sort( Values.COMPARATOR );
for ( int i = 0; i < count; i++ )
{
updates.add( add( i + 1, descriptor.schema(), values.get( i ) ) );
}

updateAndCommit( updates );

for ( int f = 0; f < values.size(); f++ )
{
for ( int t = f; t < values.size(); t++ )
{
Value from = values.get( f );
Value to = values.get( t );
for ( boolean fromInclusive : new boolean[] {true, false} )
{
for ( boolean toInclusive : new boolean[] {true, false} )
{
assertThat( query( range( 1, from, fromInclusive, to, toInclusive ) ), equalTo( ids( f, fromInclusive, t, toInclusive ) ) );
}
}
}
}
}

private List<Long> ids( int fromIndex, boolean fromInclusive, int toIndex, boolean toInclusive )
{
List<Long> ids = new ArrayList<>();
int from = fromInclusive ? fromIndex : fromIndex + 1;
int to = toInclusive ? toIndex : toIndex - 1;
for ( int i = from; i <= to; i++ )
{
ids.add( (long) (i + 1) );
}
return ids;
}

/** /**
* Helper for testing range seeks. Takes 4 ordered sample values. * Helper for testing range seeks. Takes 4 ordered sample values.
*/ */
Expand Down

0 comments on commit 45b82a0

Please sign in to comment.