Skip to content

Commit

Permalink
Add tests for sparse nodes
Browse files Browse the repository at this point in the history
  • Loading branch information
@pontusmelke
pontusmelke committed Apr 10, 2018
1 parent dc8883d commit 9521a03
Showing 1 changed file with 143 additions and 16 deletions.
159 changes: 143 additions & 16 deletions community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/helpers/NodesTest.java
View file
Expand Up @@ -29,10 +29,9 @@


public class NodesTest public class NodesTest
{ {
private static final StubNodeCursor NODE = new StubNodeCursor();


@Test @Test
public void shouldCountOutgoing() public void shouldCountOutgoingDense()
{ {
// Given // Given
StubGroupCursor groupCursor = new StubGroupCursor( StubGroupCursor groupCursor = new StubGroupCursor(
Expand All @@ -45,14 +44,35 @@ public void shouldCountOutgoing()
StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor ); StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor );


// When // When
int count = countOutgoing( NODE, cursors ); int count = countOutgoing( new StubNodeCursor( true ), cursors );


// Then // Then
assertThat( count, equalTo( 24 ) ); assertThat( count, equalTo( 24 ) );
} }


@Test @Test
public void shouldCountIncoming() public void shouldCountOutgoingSparse()
{
// Given
StubRelationshipCursor relationshipCursor = new StubRelationshipCursor(
new TestRelationshipChain( 11 )
.outgoing( 55, 0, 1 )
.incoming( 56, 0, 1 )
.outgoing( 57, 0, 1 )
.loop( 58, 0 ) );
StubCursorFactory cursors = new StubCursorFactory().withRelationshipTraversalCursors( relationshipCursor );

// When
StubNodeCursor nodeCursor = new StubNodeCursor( false ).withNode( 11 );
nodeCursor.next();
int count = countOutgoing( nodeCursor, cursors );

// Then
assertThat( count, equalTo( 3 ) );
}

@Test
public void shouldCountIncomingDense()
{ {
// Given // Given
StubGroupCursor groupCursor = new StubGroupCursor( StubGroupCursor groupCursor = new StubGroupCursor(
Expand All @@ -65,14 +85,36 @@ public void shouldCountIncoming()
StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor ); StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor );


// When // When
int count = countIncoming( new StubNodeCursor(), cursors ); int count = countIncoming( new StubNodeCursor( true ), cursors );


// Then // Then
assertThat( count, equalTo( 17 ) ); assertThat( count, equalTo( 17 ) );
} }


@Test @Test
public void shouldCountAll() public void shouldCountIncomingSparse()
{
// Given
StubRelationshipCursor relationshipCursor = new StubRelationshipCursor(
new TestRelationshipChain( 11 )
.outgoing( 55, 0, 1 )
.incoming( 56, 0, 1 )
.outgoing( 57, 0, 1 )
.loop( 58, 0 ) );
StubCursorFactory cursors = new StubCursorFactory().withRelationshipTraversalCursors( relationshipCursor );

StubNodeCursor nodeCursor = new StubNodeCursor( false ).withNode( 11 );
nodeCursor.next();

// When
int count = countIncoming( nodeCursor, cursors );

// Then
assertThat( count, equalTo( 2 ) );
}

@Test
public void shouldCountAllDense()
{ {
// Given // Given
StubGroupCursor groupCursor = new StubGroupCursor( StubGroupCursor groupCursor = new StubGroupCursor(
Expand All @@ -85,14 +127,36 @@ public void shouldCountAll()
StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor ); StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor );


// When // When
int count = countAll( new StubNodeCursor(), cursors ); int count = countAll( new StubNodeCursor( true ), cursors );


// Then // Then
assertThat( count, equalTo( 29 ) ); assertThat( count, equalTo( 29 ) );
} }


@Test @Test
public void shouldCountOutgoingWithType() public void shouldCountAllSparse()
{
// Given
StubRelationshipCursor relationshipCursor = new StubRelationshipCursor(
new TestRelationshipChain( 11 )
.outgoing( 55, 0, 1 )
.incoming( 56, 0, 1 )
.outgoing( 57, 0, 1 )
.loop( 58, 0 ) );
StubCursorFactory cursors = new StubCursorFactory().withRelationshipTraversalCursors( relationshipCursor );

StubNodeCursor nodeCursor = new StubNodeCursor( false ).withNode( 11 );
nodeCursor.next();

// When
int count = countAll( nodeCursor, cursors );

// Then
assertThat( count, equalTo( 4 ) );
}

@Test
public void shouldCountOutgoingDenseWithType()
{ {
// Given // Given
StubGroupCursor groupCursor = new StubGroupCursor( StubGroupCursor groupCursor = new StubGroupCursor(
Expand All @@ -102,12 +166,34 @@ public void shouldCountOutgoingWithType()
StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor, groupCursor ); StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor, groupCursor );


// Then // Then
assertThat( countOutgoing( new StubNodeCursor(), cursors, 1 ), equalTo( 6 ) ); assertThat( countOutgoing( new StubNodeCursor( true ), cursors, 1 ), equalTo( 6 ) );
assertThat( countOutgoing( new StubNodeCursor(), cursors, 2 ), equalTo( 4 ) ); assertThat( countOutgoing( new StubNodeCursor( true ), cursors, 2 ), equalTo( 4 ) );
}

@Test
public void shouldCountOutgoingSparseWithType()
{
// Given
StubRelationshipCursor relationshipCursor = new StubRelationshipCursor(
new TestRelationshipChain( 11 )
.outgoing( 55, 0, 1 )
.incoming( 56, 0, 1 )
.outgoing( 57, 0, 1 )
.loop( 58, 2 ) );
StubCursorFactory cursors = new StubCursorFactory( true )
.withRelationshipTraversalCursors( relationshipCursor );

// Then
StubNodeCursor nodeCursor = new StubNodeCursor( false ).withNode( 11 );
nodeCursor.next();
assertThat( countOutgoing( nodeCursor, cursors, 1 ), equalTo( 2 ) );
nodeCursor = new StubNodeCursor( false ).withNode( 11 );
nodeCursor.next();
assertThat( countOutgoing( nodeCursor, cursors, 2 ), equalTo( 1 ) );
} }


@Test @Test
public void shouldCountIncomingWithType() public void shouldCountIncomingWithTypeDense()
{ {
// Given // Given
StubGroupCursor groupCursor = new StubGroupCursor( StubGroupCursor groupCursor = new StubGroupCursor(
Expand All @@ -117,12 +203,33 @@ public void shouldCountIncomingWithType()
StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor, groupCursor ); StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor, groupCursor );


// Then // Then
assertThat( countIncoming( new StubNodeCursor(), cursors, 1 ), equalTo( 6 ) ); assertThat( countIncoming( new StubNodeCursor( true ), cursors, 1 ), equalTo( 6 ) );
assertThat( countIncoming( new StubNodeCursor(), cursors, 2 ), equalTo( 4 ) ); assertThat( countIncoming( new StubNodeCursor( true ), cursors, 2 ), equalTo( 4 ) );
}
@Test
public void shouldCountIncomingWithTypeSparse()
{
// Given
StubRelationshipCursor relationshipCursor = new StubRelationshipCursor(
new TestRelationshipChain( 11 )
.outgoing( 55, 0, 1 )
.incoming( 56, 0, 1 )
.outgoing( 57, 0, 1 )
.loop( 58, 2 ) );
StubCursorFactory cursors = new StubCursorFactory( true )
.withRelationshipTraversalCursors( relationshipCursor );

// Then
StubNodeCursor nodeCursor = new StubNodeCursor( false ).withNode( 11 );
nodeCursor.next();
assertThat( countIncoming( nodeCursor, cursors, 1 ), equalTo( 1 ) );
nodeCursor = new StubNodeCursor( false ).withNode( 11 );
nodeCursor.next();
assertThat( countIncoming( nodeCursor, cursors, 2 ), equalTo( 1 ) );
} }


@Test @Test
public void shouldCountAllWithType() public void shouldCountAllWithTypeDense()
{ {
// Given // Given
StubGroupCursor groupCursor = new StubGroupCursor( StubGroupCursor groupCursor = new StubGroupCursor(
Expand All @@ -132,8 +239,28 @@ public void shouldCountAllWithType()
StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor, groupCursor ); StubCursorFactory cursors = new StubCursorFactory().withGroupCursors( groupCursor, groupCursor );


// Then // Then
assertThat( countAll( new StubNodeCursor(), cursors, 1 ), equalTo( 7 ) ); assertThat( countAll( new StubNodeCursor( true ), cursors, 1 ), equalTo( 7 ) );
assertThat( countAll( new StubNodeCursor(), cursors, 2 ), equalTo( 5 ) ); assertThat( countAll( new StubNodeCursor( true ), cursors, 2 ), equalTo( 5 ) );
}

@Test
public void shouldCountAllWithTypeSparse()
{
// Given
StubRelationshipCursor relationshipCursor = new StubRelationshipCursor(
new TestRelationshipChain( 11 )
.outgoing( 55, 0, 1 )
.incoming( 56, 0, 1 )
.outgoing( 57, 0, 1 )
.loop( 58, 2 ) );
StubCursorFactory cursors = new StubCursorFactory( true )
.withRelationshipTraversalCursors( relationshipCursor );

// Then
StubNodeCursor nodeCursor = new StubNodeCursor( false ).withNode( 11 );
nodeCursor.next();
assertThat( countAll( nodeCursor, cursors, 1 ), equalTo( 3 ) );
assertThat( countAll( nodeCursor, cursors, 2 ), equalTo( 1) );
} }


private StubGroupCursor.GroupData group() private StubGroupCursor.GroupData group()
Expand Down

0 comments on commit 9521a03

Please sign in to comment.