TestNodeQueue.java

package org.neo4j.util;

import java.util.ArrayList;
import java.util.Collection;

import org.neo4j.graphdb.DynamicRelationshipType;
import org.neo4j.graphdb.Node;
import org.neo4j.graphdb.RelationshipType;
import org.neo4j.graphdb.Transaction;

public class TestNodeQueue extends Neo4jTest
{
	private static enum RelTypes implements RelationshipType
	{
		TEST_QUEUE,
	}
	
	public void testSome() throws Exception
	{
		// TODO Test diabled for now since it always fails.
		if ( true )
		{
			return;
		}
		
		Collection<Node> rootNodes = new ArrayList<Node>();
		for ( int i = 0; i < 10; i++ )
		{
			Transaction tx = graphDb().beginTx();
	        try
	        {
	        	Node rootNode = graphDb().createNode();
	        	rootNodes.add( rootNode );
		        NodeQueue q = new NodeQueue( graphDb(),
		        	rootNode, RelTypes.TEST_QUEUE );
		        for ( int ii = 0; ii < 10000; ii++ )
		        {
		        	q.add();
		        }
		        tx.success();
	        }
	        finally
	        {
		        tx.finish();
	        }
		}
		
		Transaction tx = graphDb().beginTx();
        try
        {
	        for ( Node node : rootNodes )
	        {
	        	NodeQueue q = new NodeQueue( graphDb(), node,
	        		RelTypes.TEST_QUEUE );
	        	while ( q.peek() != null )
	        	{
	        		q.remove();
	        	}
	        	node.delete();
	        }
	        tx.success();
        }
        finally
        {
	        tx.finish();
        }
	}
	
	public void testMany() throws Exception
	{
	    Transaction tx = graphDb().beginTx();
	    
	    Node rootNode = graphDb().createNode();
	    NodeQueue q = new NodeQueue( graphDb(), rootNode, RelTypes.TEST_QUEUE );
	    for ( int i = 0; i < 10; i++ )
	    {
	        Node node = q.add();
	        node.setProperty( "p", i );
	    }
	    
	    Node node = q.peek();
	    assertEquals( 0, node.getProperty( "p" ) );
	    Node[] nodes = q.peek( 4 );
	    assertEquals( 4, nodes.length );
	    for ( int i = 0; i < 4; i++ )
	    {
	        assertEquals( i, nodes[ i ].getProperty( "p" ) );
	    }
	    assertEquals( 10, q.peek( 20 ).length );
	    
	    q.remove( 3 );
	    assertEquals( 3, q.peek().getProperty( "p" ) );
	    assertEquals( 7, q.peek( 20 ).length );
	    
	    q.remove( 7 );
	    rootNode.delete();
	    
	    tx.success();
	    tx.finish();
	}
	
	public void testFixedLengthList() throws Exception
	{
	    Transaction tx = graphDb().beginTx();
	    
	    Node rootNode = graphDb().createNode();
	    FixedLengthNodeList list = new FixedLengthNodeList( graphDb(), rootNode,
	        DynamicRelationshipType.withName( "LIST_TEST" ), 5 );
	    assertNull( list.peek() );
	    Node a = list.add();
	    assertEquals( a, list.peek() );
	    assertEquals( a, list.peek( 10 )[ 0 ] );
	    Node b = list.add();
	    assertEquals( b, list.peek() );
	    assertEquals( b, list.peek( 10 )[ 0 ] );
        assertEquals( a, list.peek( 10 )[ 1 ] );
        Node c = list.add();
        Node d = list.add();
        Node e = list.add();
        assertEquals( 5, list.peek( 10 ).length );
        assertEquals( e, list.peek() );
        assertEquals( a, list.peek( 10 )[ 4 ] );
        Node f = list.add();
        assertEquals( 5, list.peek( 10 ).length );
        assertEquals( f, list.peek() );
        assertEquals( b, list.peek( 10 )[ 4 ] );
        Node g = list.add();
        assertEquals( 5, list.peek( 10 ).length );
        assertEquals( g, list.peek() );
        assertEquals( c, list.peek( 10 )[ 4 ] );
        
        list.remove( 10 );
        rootNode.delete();
	    
	    tx.success();
	    tx.finish();
	}
}