/**
* Copyright (c) 2002-2011 "Neo Technology,"
* Network Engine for Objects in Lund AB [http://neotechnology.com]
*
* This file is part of Neo4j.
*
* Neo4j is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.neo4j.kernel.impl.nioneo.store;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import javax.transaction.xa.XAException;
import javax.transaction.xa.XAResource;
import javax.transaction.xa.Xid;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.neo4j.graphdb.Node;
import org.neo4j.graphdb.Relationship;
import org.neo4j.graphdb.RelationshipType;
import org.neo4j.helpers.collection.MapUtil;
import org.neo4j.kernel.CommonFactories;
import org.neo4j.kernel.IdGeneratorFactory;
import org.neo4j.kernel.impl.AbstractNeo4jTestCase;
import org.neo4j.kernel.impl.core.LockReleaser;
import org.neo4j.kernel.impl.core.PropertyIndex;
import org.neo4j.kernel.impl.nioneo.xa.NeoStoreXaConnection;
import org.neo4j.kernel.impl.nioneo.xa.NeoStoreXaDataSource;
import org.neo4j.kernel.impl.nioneo.xa.NodeEventConsumer;
import org.neo4j.kernel.impl.nioneo.xa.RelationshipEventConsumer;
import org.neo4j.kernel.impl.nioneo.xa.RelationshipTypeEventConsumer;
import org.neo4j.kernel.impl.transaction.LockManager;
import org.neo4j.kernel.impl.transaction.XidImpl;
import org.neo4j.kernel.impl.transaction.xaframework.LogBufferFactory;
import org.neo4j.kernel.impl.transaction.xaframework.TxIdGenerator;
import org.neo4j.kernel.impl.util.ArrayMap;
public class TestNeoStore extends AbstractNeo4jTestCase
{
private static final IdGeneratorFactory ID_GENERATOR_FACTORY =
CommonFactories.defaultIdGeneratorFactory();
private NodeEventConsumer nStore;
private PropertyStore pStore;
private RelationshipTypeEventConsumer relTypeStore;
private RelationshipEventConsumer rStore;
private NeoStoreXaDataSource ds;
private NeoStoreXaConnection xaCon;
@Override
protected boolean restartGraphDbBetweenTests()
{
return true;
}
private String path()
{
String path = AbstractNeo4jTestCase.getStorePath( "test-neostore" );
new File( path ).mkdirs();
return path;
}
private String file( String name )
{
return path() + File.separator + name;
}
@Before
public void setUpNeoStore() throws Exception
{
NeoStore.createStore( file( "neo" ), MapUtil.map(
IdGeneratorFactory.class, ID_GENERATOR_FACTORY,
FileSystemAbstraction.class, CommonFactories.defaultFileSystemAbstraction(),
LogBufferFactory.class, CommonFactories.defaultLogBufferFactory( MapUtil.stringMap() ) ) );
}
private static class MyPropertyIndex extends
org.neo4j.kernel.impl.core.PropertyIndex
{
private static Map<String,PropertyIndex> stringToIndex = new HashMap<String,PropertyIndex>();
private static Map<Integer,PropertyIndex> intToIndex = new HashMap<Integer,PropertyIndex>();
protected MyPropertyIndex( String key, int keyId )
{
super( key, keyId );
}
public static Iterable<PropertyIndex> index( String key )
{
if ( stringToIndex.containsKey( key ) )
{
return Arrays.asList( new PropertyIndex[] { stringToIndex
.get( key ) } );
}
return Collections.emptyList();
}
public static PropertyIndex getIndexFor( int index )
{
return intToIndex.get( index );
}
public static void add( MyPropertyIndex index )
{
// TODO Auto-generated method stub
stringToIndex.put( index.getKey(), index );
intToIndex.put( index.getKeyId(), index );
}
}
private PropertyIndex createDummyIndex( int id, String key )
{
MyPropertyIndex index = new MyPropertyIndex( key, id );
MyPropertyIndex.add( index );
return index;
}
private void initializeStores() throws IOException
{
try
{
LockManager lockManager = getEmbeddedGraphDb().getConfig()
.getLockManager();
LockReleaser lockReleaser = getEmbeddedGraphDb().getConfig()
.getLockReleaser();
ds = new NeoStoreXaDataSource( MapUtil.genericMap(
LockManager.class, lockManager,
LockReleaser.class, lockReleaser,
IdGeneratorFactory.class, ID_GENERATOR_FACTORY,
FileSystemAbstraction.class, CommonFactories.defaultFileSystemAbstraction(),
LogBufferFactory.class, CommonFactories.defaultLogBufferFactory( MapUtil.stringMap() ),
TxIdGenerator.class, TxIdGenerator.DEFAULT,
"store_dir", path(),
"neo_store", file( "neo" ),
"logical_log", file( "nioneo_logical.log" ) ) );
// ds = new NeoStoreXaDataSource( file( "neo" ), file( "nioneo_logical.log" ),
// lockManager, lockReleaser );
}
catch ( InstantiationException e )
{
throw new IOException( "" + e );
}
xaCon = (NeoStoreXaConnection) ds.getXaConnection();
nStore = xaCon.getNodeConsumer();
pStore = xaCon.getPropertyStore();
relTypeStore = xaCon.getRelationshipTypeConsumer();
rStore = xaCon.getRelationshipConsumer();
}
private Xid dummyXid;
private byte txCount = (byte) 0;
XAResource xaResource;
private void startTx() throws XAException
{
dummyXid = new XidImpl( new byte[txCount], new byte[txCount] );
txCount++;
xaResource = xaCon.getXaResource();
xaResource.start( dummyXid, XAResource.TMNOFLAGS );
}
private void commitTx() throws XAException
{
xaResource.end( dummyXid, XAResource.TMSUCCESS );
xaResource.commit( dummyXid, true );
// xaCon.clearAllTransactions();
}
@After
public void tearDownNeoStore()
{
File file = new File( file( "neo" ) );
file.delete();
file = new File( file( "neo.id" ) );
file.delete();
file = new File( file( "neo.nodestore.db" ) );
file.delete();
file = new File( file( "neo.nodestore.db.id" ) );
file.delete();
file = new File( file( "neo.propertystore.db" ) );
file.delete();
file = new File( file( "neo.propertystore.db.id" ) );
file.delete();
file = new File( file( "neo.propertystore.db.index" ) );
file.delete();
file = new File( file( "neo.propertystore.db.index.id" ) );
file.delete();
file = new File( file( "neo.propertystore.db.index.keys" ) );
file.delete();
file = new File( file( "neo.propertystore.db.index.keys.id" ) );
file.delete();
file = new File( file( "neo.propertystore.db.strings" ) );
file.delete();
file = new File( file( "neo.propertystore.db.strings.id" ) );
file.delete();
file = new File( file( "neo.propertystore.db.arrays" ) );
file.delete();
file = new File( file( "neo.propertystore.db.arrays.id" ) );
file.delete();
file = new File( file( "neo.relationshipstore.db" ) );
file.delete();
file = new File( file( "neo.relationshipstore.db.id" ) );
file.delete();
file = new File( file( "neo.relationshiptypestore.db" ) );
file.delete();
file = new File( file( "neo.relationshiptypestore.db.id" ) );
file.delete();
file = new File( file( "neo.relationshiptypestore.db.names" ) );
file.delete();
file = new File( file( "neo.relationshiptypestore.db.names.id" ) );
file.delete();
file = new File( "." );
for ( File nioFile : file.listFiles() )
{
if ( nioFile.getName().startsWith( "nioneo_logical.log" ) )
{
nioFile.delete();
}
}
}
private PropertyIndex index( String key )
{
Iterator<PropertyIndex> itr = MyPropertyIndex.index( key ).iterator();
if ( !itr.hasNext() )
{
int id = (int) ds.nextId( PropertyIndex.class );
PropertyIndex index = createDummyIndex( id, key );
xaCon.getPropertyIndexConsumer().createPropertyIndex( id, key );
return index;
}
return itr.next();
}
@Test
public void testCreateNeoStore() throws Exception
{
initializeStores();
startTx();
// setup test population
long node1 = ds.nextId( Node.class );
nStore.createNode( node1 );
long node2 = ds.nextId( Node.class );
nStore.createNode( node2 );
long n1prop1 = pStore.nextId();
long n1prop2 = pStore.nextId();
long n1prop3 = pStore.nextId();
nStore.addProperty( node1, n1prop1, index( "prop1" ), "string1" );
nStore.addProperty( node1, n1prop2, index( "prop2" ), new Integer(
1 ) );
nStore.addProperty( node1, n1prop3, index( "prop3" ), new Boolean(
true ) );
long n2prop1 = pStore.nextId();
long n2prop2 = pStore.nextId();
long n2prop3 = pStore.nextId();
nStore.addProperty( node2, n2prop1, index( "prop1" ), "string2" );
nStore.addProperty( node2, n2prop2, index( "prop2" ), new Integer(
2 ) );
nStore.addProperty( node2, n2prop3, index( "prop3" ), new Boolean(
false ) );
int relType1 = (int) ds.nextId( RelationshipType.class );
relTypeStore.addRelationshipType( relType1, "relationshiptype1" );
int relType2 = (int) ds.nextId( RelationshipType.class );
relTypeStore.addRelationshipType( relType2, "relationshiptype2" );
long rel1 = ds.nextId( Relationship.class );
rStore.createRelationship( rel1, node1, node2, relType1 );
long rel2 = ds.nextId( Relationship.class );
rStore.createRelationship( rel2, node2, node1, relType2 );
long r1prop1 = pStore.nextId();
long r1prop2 = pStore.nextId();
long r1prop3 = pStore.nextId();
rStore.addProperty( rel1, r1prop1, index( "prop1" ), "string1" );
rStore.addProperty( rel1, r1prop2, index( "prop2" ),
new Integer( 1 ) );
rStore.addProperty( rel1, r1prop3, index( "prop3" ), new Boolean(
true ) );
long r2prop1 = pStore.nextId();
long r2prop2 = pStore.nextId();
long r2prop3 = pStore.nextId();
rStore.addProperty( rel2, r2prop1, index( "prop1" ), "string2" );
rStore.addProperty( rel2, r2prop2, index( "prop2" ),
new Integer( 2 ) );
rStore.addProperty( rel2, r2prop3, index( "prop3" ), new Boolean(
false ) );
commitTx();
ds.close();
initializeStores();
startTx();
// validate node
validateNodeRel1( node1, n1prop1, n1prop2, n1prop3, rel1, rel2,
relType1, relType2 );
validateNodeRel2( node2, n2prop1, n2prop2, n2prop3, rel1, rel2,
relType1, relType2 );
// validate rels
validateRel1( rel1, r1prop1, r1prop2, r1prop3, node1, node2,
relType1 );
validateRel2( rel2, r2prop1, r2prop2, r2prop3, node2, node1,
relType2 );
validateRelTypes( relType1, relType2 );
// validate reltypes
validateRelTypes( relType1, relType2 );
commitTx();
ds.close();
initializeStores();
startTx();
// validate and delete rels
deleteRel1( rel1, r1prop1, r1prop2, r1prop3, node1, node2, relType1 );
deleteRel2( rel2, r2prop1, r2prop2, r2prop3, node2, node1, relType2 );
// validate and delete nodes
deleteNode1( node1, n1prop1, n1prop2, n1prop3 );
deleteNode2( node2, n2prop1, n2prop2, n2prop3 );
commitTx();
ds.close();
initializeStores();
startTx();
assertEquals( false, nStore.loadLightNode( node1 ) );
assertEquals( false, nStore.loadLightNode( node2 ) );
testGetRels( new long[] { rel1, rel2 } );
// testGetProps( neoStore, new int[] {
// n1prop1, n1prop2, n1prop3, n2prop1, n2prop2, n2prop3,
// r1prop1, r1prop2, r1prop3, r2prop1, r2prop2, r2prop3
// } );
long nodeIds[] = new long[10];
for ( int i = 0; i < 3; i++ )
{
nodeIds[i] = ds.nextId( Node.class );
nStore.createNode( nodeIds[i] );
nStore.addProperty( nodeIds[i], pStore.nextId(),
index( "nisse" ), new Integer( 10 - i ) );
}
for ( int i = 0; i < 2; i++ )
{
long id = ds.nextId( Relationship.class );
rStore.createRelationship( id,
nodeIds[i], nodeIds[i + 1], relType1 );
rStore.deleteRelationship( id );
}
for ( int i = 0; i < 3; i++ )
{
RelationshipChainPosition pos =
rStore.getRelationshipChainPosition( nodeIds[i] );
for ( RelationshipData rel :
rStore.getMoreRelationships( nodeIds[i], pos ) )
{
rStore.deleteRelationship( rel.getId() );
}
nStore.deleteNode( nodeIds[i] );
}
commitTx();
ds.close();
}
private Object getValue( PropertyRecord propertyRecord ) throws IOException
{
try
{
return propertyRecord.getType().getValue( propertyRecord, pStore );
}
catch ( InvalidRecordException ex )
{
throw new IOException( ex );
}
}
private void validateNodeRel1( long node, long prop1, long prop2, long prop3,
long rel1, long rel2, int relType1, int relType2 ) throws IOException
{
assertTrue( nStore.loadLightNode( node ) );
ArrayMap<Integer,PropertyData> props = nStore.getProperties( node,
false );
int count = 0;
for ( int keyId : props.keySet() )
{
long id = props.get( keyId ).getId();
PropertyRecord record = pStore.getRecord( id );
PropertyData data = new PropertyData( id, getValue( record ) );
if ( data.getId() == prop1 )
{
assertEquals( "prop1", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( "string1", data.getValue() );
nStore.changeProperty( node, prop1, "-string1" );
}
else if ( data.getId() == prop2 )
{
assertEquals( "prop2", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Integer( 1 ), data.getValue() );
nStore.changeProperty( node, prop2, new Integer( -1 ) );
}
else if ( data.getId() == prop3 )
{
assertEquals( "prop3", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Boolean( true ), data.getValue() );
nStore.changeProperty( node, prop3, new Boolean( false ) );
}
else
{
throw new IOException();
}
count++;
}
assertEquals( 3, count );
count = 0;
RelationshipChainPosition pos = rStore.getRelationshipChainPosition( node );
while ( true )
{
Iterable<RelationshipData> relData = rStore.getMoreRelationships( node, pos );
if ( !relData.iterator().hasNext() )
{
break;
}
for ( RelationshipData rel : relData )
{
if ( rel.getId() == rel1 )
{
assertEquals( node, rel.firstNode() );
assertEquals( relType1, rel.relationshipType() );
}
else if ( rel.getId() == rel2 )
{
assertEquals( node, rel.secondNode() );
assertEquals( relType2, rel.relationshipType() );
}
else
{
throw new IOException();
}
count++;
}
}
assertEquals( 2, count );
}
private void validateNodeRel2( long node, long prop1, long prop2, long prop3,
long rel1, long rel2, int relType1, int relType2 ) throws IOException
{
assertTrue( nStore.loadLightNode( node ) );
ArrayMap<Integer,PropertyData> props = nStore.getProperties( node,
false );
int count = 0;
for ( int keyId : props.keySet() )
{
long id = props.get( keyId ).getId();
PropertyRecord record = pStore.getRecord( id );
PropertyData data = new PropertyData( id, getValue( record ) );
if ( data.getId() == prop1 )
{
assertEquals( "prop1", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( "string2", data.getValue() );
nStore.changeProperty( node, prop1, "-string2" );
}
else if ( data.getId() == prop2 )
{
assertEquals( "prop2", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Integer( 2 ), data.getValue() );
nStore.changeProperty( node, prop2, new Integer( -2 ) );
}
else if ( data.getId() == prop3 )
{
assertEquals( "prop3", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Boolean( false ), data.getValue() );
nStore.changeProperty( node, prop3, new Boolean( true ) );
}
else
{
throw new IOException();
}
count++;
}
assertEquals( 3, count );
count = 0;
RelationshipChainPosition pos = rStore.getRelationshipChainPosition( node );
while ( true )
{
Iterable<RelationshipData> relData = rStore.getMoreRelationships( node, pos );
if ( !relData.iterator().hasNext() )
{
break;
}
for ( RelationshipData rel : relData )
{
if ( rel.getId() == rel1 )
{
assertEquals( node, rel.secondNode() );
assertEquals( relType1, rel.relationshipType() );
}
else if ( rel.getId() == rel2 )
{
assertEquals( node, rel.firstNode() );
assertEquals( relType2, rel.relationshipType() );
}
else
{
throw new IOException();
}
count++;
}
}
assertEquals( 2, count );
}
private void validateRel1( long rel, long prop1, long prop2, long prop3,
long firstNode, long secondNode, int relType ) throws IOException
{
ArrayMap<Integer,PropertyData> props = rStore.getProperties( rel,
false );
int count = 0;
for ( int keyId : props.keySet() )
{
long id = props.get( keyId ).getId();
PropertyRecord record = pStore.getRecord( id );
PropertyData data = new PropertyData( id, getValue( record ) );
if ( data.getId() == prop1 )
{
assertEquals( "prop1", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( "string1", data.getValue() );
rStore.changeProperty( rel, prop1, "-string1" );
}
else if ( data.getId() == prop2 )
{
assertEquals( "prop2", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Integer( 1 ), data.getValue() );
rStore.changeProperty( rel, prop2, new Integer( -1 ) );
}
else if ( data.getId() == prop3 )
{
assertEquals( "prop3", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Boolean( true ), data.getValue() );
rStore.changeProperty( rel, prop3, new Boolean( false ) );
}
else
{
throw new IOException();
}
count++;
}
assertEquals( 3, count );
RelationshipData relData = rStore.getRelationship( rel );
assertEquals( firstNode, relData.firstNode() );
assertEquals( secondNode, relData.secondNode() );
assertEquals( relType, relData.relationshipType() );
}
private void validateRel2( long rel, long prop1, long prop2, long prop3,
long firstNode, long secondNode, int relType ) throws IOException
{
ArrayMap<Integer,PropertyData> props = rStore.getProperties( rel,
false );
int count = 0;
for ( int keyId : props.keySet() )
{
long id = props.get( keyId ).getId();
PropertyRecord record = pStore.getRecord( id );
PropertyData data = new PropertyData( id, getValue( record ) );
if ( data.getId() == prop1 )
{
assertEquals( "prop1", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( "string2", data.getValue() );
rStore.changeProperty( rel, prop1, "-string2" );
}
else if ( data.getId() == prop2 )
{
assertEquals( "prop2", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Integer( 2 ), data.getValue() );
rStore.changeProperty( rel, prop2, new Integer( -2 ) );
}
else if ( data.getId() == prop3 )
{
assertEquals( "prop3", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Boolean( false ), data.getValue() );
rStore.changeProperty( rel, prop3, new Boolean( true ) );
}
else
{
throw new IOException();
}
count++;
}
assertEquals( 3, count );
RelationshipData relData = rStore.getRelationship( rel );
assertEquals( firstNode, relData.firstNode() );
assertEquals( secondNode, relData.secondNode() );
assertEquals( relType, relData.relationshipType() );
}
private void validateRelTypes( int relType1, int relType2 )
throws IOException
{
RelationshipTypeData data = relTypeStore.getRelationshipType( relType1 );
assertEquals( relType1, data.getId() );
assertEquals( "relationshiptype1", data.getName() );
data = relTypeStore.getRelationshipType( relType2 );
assertEquals( relType2, data.getId() );
assertEquals( "relationshiptype2", data.getName() );
RelationshipTypeData allData[] = relTypeStore.getRelationshipTypes();
assertEquals( 2, allData.length );
for ( int i = 0; i < 2; i++ )
{
if ( allData[i].getId() == relType1 )
{
assertEquals( relType1, allData[i].getId() );
assertEquals( "relationshiptype1", allData[i].getName() );
}
else if ( allData[i].getId() == relType2 )
{
assertEquals( relType2, allData[i].getId() );
assertEquals( "relationshiptype2", allData[i].getName() );
}
else
{
throw new IOException();
}
}
}
private void deleteRel1( long rel, long prop1, long prop2, long prop3,
long firstNode, long secondNode, int relType ) throws IOException
{
ArrayMap<Integer,PropertyData> props = rStore.getProperties( rel,
false );
int count = 0;
for ( int keyId : props.keySet() )
{
long id = props.get( keyId ).getId();
PropertyRecord record = pStore.getRecord( id );
PropertyData data = new PropertyData( id, getValue( record ) );
if ( data.getId() == prop1 )
{
assertEquals( "prop1", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( "-string1", data.getValue() );
}
else if ( data.getId() == prop2 )
{
assertEquals( "prop2", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Integer( -1 ), data.getValue() );
}
else if ( data.getId() == prop3 )
{
assertEquals( "prop3", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Boolean( false ), data.getValue() );
rStore.removeProperty( rel, prop3 );
}
else
{
throw new IOException();
}
count++;
}
assertEquals( 3, count );
assertEquals( 3, rStore.getProperties( rel, false ).size() );
RelationshipData relData = rStore.getRelationship( rel );
assertEquals( firstNode, relData.firstNode() );
assertEquals( secondNode, relData.secondNode() );
assertEquals( relType, relData.relationshipType() );
rStore.deleteRelationship( rel );
RelationshipChainPosition firstPos =
rStore.getRelationshipChainPosition( firstNode );
Iterator<RelationshipData> first =
rStore.getMoreRelationships( firstNode, firstPos ).iterator();
first.next();
RelationshipChainPosition secondPos =
rStore.getRelationshipChainPosition( secondNode );
Iterator<RelationshipData> second =
rStore.getMoreRelationships( secondNode, secondPos ).iterator();
second.next();
assertTrue( first.hasNext() );
assertTrue( second.hasNext() );
}
private void deleteRel2( long rel, long prop1, long prop2, long prop3,
long firstNode, long secondNode, int relType ) throws IOException
{
ArrayMap<Integer,PropertyData> props = rStore.getProperties( rel,
false );
int count = 0;
for ( int keyId : props.keySet() )
{
long id = props.get( keyId ).getId();
PropertyRecord record = pStore.getRecord( id );
PropertyData data = new PropertyData( id, getValue( record ) );
if ( data.getId() == prop1 )
{
assertEquals( "prop1", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( "-string2", data.getValue() );
}
else if ( data.getId() == prop2 )
{
assertEquals( "prop2", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Integer( -2 ), data.getValue() );
}
else if ( data.getId() == prop3 )
{
assertEquals( "prop3", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Boolean( true ), data.getValue() );
rStore.removeProperty( rel, prop3 );
}
else
{
throw new IOException();
}
count++;
}
assertEquals( 3, count );
assertEquals( 3, rStore.getProperties( rel, false ).size() );
RelationshipData relData = rStore.getRelationship( rel );
assertEquals( firstNode, relData.firstNode() );
assertEquals( secondNode, relData.secondNode() );
assertEquals( relType, relData.relationshipType() );
rStore.deleteRelationship( rel );
RelationshipChainPosition firstPos =
rStore.getRelationshipChainPosition( firstNode );
Iterator<RelationshipData> first =
rStore.getMoreRelationships( firstNode, firstPos ).iterator();
RelationshipChainPosition secondPos =
rStore.getRelationshipChainPosition( secondNode );
Iterator<RelationshipData> second =
rStore.getMoreRelationships( secondNode, secondPos ).iterator();
assertTrue( first.hasNext() );
assertTrue( second.hasNext() );
}
private void deleteNode1( long node, long prop1, long prop2, long prop3 )
throws IOException
{
ArrayMap<Integer,PropertyData> props = nStore.getProperties( node,
false );
int count = 0;
for ( int keyId : props.keySet() )
{
long id = props.get( keyId ).getId();
PropertyRecord record = pStore.getRecord( id );
PropertyData data = new PropertyData( id, getValue( record ) );
if ( data.getId() == prop1 )
{
assertEquals( "prop1", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( "-string1", data.getValue() );
}
else if ( data.getId() == prop2 )
{
assertEquals( "prop2", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Integer( -1 ), data.getValue() );
}
else if ( data.getId() == prop3 )
{
assertEquals( "prop3", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Boolean( false ), data.getValue() );
nStore.removeProperty( node, prop3 );
}
else
{
throw new IOException();
}
count++;
}
assertEquals( 3, count );
assertEquals( 3, nStore.getProperties( node, false ).size() );
RelationshipChainPosition pos =
rStore.getRelationshipChainPosition( node );
Iterator<RelationshipData> rels =
rStore.getMoreRelationships( node, pos ).iterator();
assertTrue( rels.hasNext() );
nStore.deleteNode( node );
}
private void deleteNode2( long node, long prop1, long prop2, long prop3 )
throws IOException
{
ArrayMap<Integer,PropertyData> props = nStore.getProperties( node,
false );
int count = 0;
for ( int keyId : props.keySet() )
{
long id = props.get( keyId ).getId();
PropertyRecord record = pStore.getRecord( id );
PropertyData data = new PropertyData( id, getValue( record ) );
if ( data.getId() == prop1 )
{
assertEquals( "prop1", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( "-string2", data.getValue() );
}
else if ( data.getId() == prop2 )
{
assertEquals( "prop2", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Integer( -2 ), data.getValue() );
}
else if ( data.getId() == prop3 )
{
assertEquals( "prop3", MyPropertyIndex.getIndexFor(
keyId ).getKey() );
assertEquals( new Boolean( true ), data.getValue() );
nStore.removeProperty( node, prop3 );
}
else
{
throw new IOException();
}
count++;
}
assertEquals( 3, count );
assertEquals( 3, nStore.getProperties( node, false ).size() );
RelationshipChainPosition pos =
rStore.getRelationshipChainPosition( node );
Iterator<RelationshipData> rels =
rStore.getMoreRelationships( node, pos ).iterator();
assertTrue( rels.hasNext() );
nStore.deleteNode( node );
}
private void testGetRels( long relIds[] )
{
for ( long relId : relIds )
{
assertEquals( null, rStore.getRelationship( relId ) );
}
}
@Test
public void testRels1() throws Exception
{
initializeStores();
startTx();
int relType1 = (int) ds.nextId( RelationshipType.class );
relTypeStore.addRelationshipType( relType1, "relationshiptype1" );
long nodeIds[] = new long[3];
for ( int i = 0; i < 3; i++ )
{
nodeIds[i] = ds.nextId( Node.class );
nStore.createNode( nodeIds[i] );
nStore.addProperty( nodeIds[i], pStore.nextId(),
index( "nisse" ), new Integer( 10 - i ) );
}
for ( int i = 0; i < 2; i++ )
{
rStore.createRelationship( ds.nextId( Relationship.class ),
nodeIds[i], nodeIds[i + 1], relType1 );
}
commitTx();
startTx();
for ( int i = 0; i < 3; i+=2 )
{
RelationshipChainPosition pos =
rStore.getRelationshipChainPosition( nodeIds[i] );
for ( RelationshipData rel :
rStore.getMoreRelationships( nodeIds[i], pos ) )
{
rStore.deleteRelationship( rel.getId() );
}
nStore.deleteNode( nodeIds[i] );
}
commitTx();
ds.close();
}
public void testRels2() throws Exception
{
initializeStores();
startTx();
int relType1 = (int) ds.nextId( RelationshipType.class );
relTypeStore.addRelationshipType( relType1, "relationshiptype1" );
long nodeIds[] = new long[3];
for ( int i = 0; i < 3; i++ )
{
nodeIds[i] = ds.nextId( Node.class );
nStore.createNode( nodeIds[i] );
nStore.addProperty( nodeIds[i], pStore.nextId(),
index( "nisse" ), new Integer( 10 - i ) );
}
for ( int i = 0; i < 2; i++ )
{
rStore.createRelationship( ds.nextId( Relationship.class ),
nodeIds[i], nodeIds[i + 1], relType1 );
}
rStore.createRelationship( ds.nextId( Relationship.class ),
nodeIds[0], nodeIds[2], relType1 );
commitTx();
startTx();
for ( int i = 0; i < 3; i++ )
{
RelationshipChainPosition pos =
rStore.getRelationshipChainPosition( nodeIds[i] );
for ( RelationshipData rel :
rStore.getMoreRelationships( nodeIds[i], pos ) )
{
rStore.deleteRelationship( rel.getId() );
}
nStore.deleteNode( nodeIds[i] );
}
commitTx();
ds.close();
}
public void testRels3() throws Exception
{
// test linked list stuff during relationship delete
initializeStores();
startTx();
int relType1 = (int) ds.nextId( RelationshipType.class );
relTypeStore.addRelationshipType( relType1, "relationshiptype1" );
long nodeIds[] = new long[8];
for ( int i = 0; i < nodeIds.length; i++ )
{
nodeIds[i] = ds.nextId( Node.class );
nStore.createNode( nodeIds[i] );
}
for ( int i = 0; i < nodeIds.length / 2; i++ )
{
rStore.createRelationship( ds.nextId( Relationship.class ),
nodeIds[i], nodeIds[i * 2], relType1 );
}
long rel5 = ds.nextId( Relationship.class );
rStore.createRelationship( rel5, nodeIds[0], nodeIds[5], relType1 );
long rel2 = ds.nextId( Relationship.class );
rStore.createRelationship( rel2, nodeIds[1], nodeIds[2], relType1 );
long rel3 = ds.nextId( Relationship.class );
rStore.createRelationship( rel3, nodeIds[1], nodeIds[3], relType1 );
long rel6 = ds.nextId( Relationship.class );
rStore.createRelationship( rel6, nodeIds[1], nodeIds[6], relType1 );
long rel1 = ds.nextId( Relationship.class );
rStore.createRelationship( rel1, nodeIds[0], nodeIds[1], relType1 );
long rel4 = ds.nextId( Relationship.class );
rStore.createRelationship( rel4, nodeIds[0], nodeIds[4], relType1 );
long rel7 = ds.nextId( Relationship.class );
rStore.createRelationship( rel7, nodeIds[0], nodeIds[7], relType1 );
commitTx();
startTx();
rStore.deleteRelationship( rel7 );
rStore.deleteRelationship( rel4 );
rStore.deleteRelationship( rel1 );
rStore.deleteRelationship( rel6 );
rStore.deleteRelationship( rel3 );
rStore.deleteRelationship( rel2 );
rStore.deleteRelationship( rel5 );
// nStore.deleteNode( nodeIds[2] );
// nStore.deleteNode( nodeIds[3] );
// nStore.deleteNode( nodeIds[1] );
// nStore.deleteNode( nodeIds[4] );
// nStore.deleteNode( nodeIds[0] );
commitTx();
ds.close();
}
public void testProps1() throws Exception
{
initializeStores();
startTx();
long nodeId = ds.nextId( Node.class );
nStore.createNode( nodeId );
long propertyId = pStore.nextId();
nStore.addProperty( nodeId, propertyId, index( "nisse" ),
new Integer( 10 ) );
commitTx();
ds.close();
initializeStores();
startTx();
nStore.changeProperty( nodeId, propertyId, new Integer( 5 ) );
nStore.removeProperty( nodeId, propertyId );
nStore.deleteNode( nodeId );
commitTx();
ds.close();
}
@Test
public void testSetBlockSize() throws Exception
{
tearDownNeoStore();
Map<Object,Object> config = new HashMap<Object, Object>();
config.put( "string_block_size", "62" );
config.put( "array_block_size", "302" );
config.put( IdGeneratorFactory.class, CommonFactories.defaultIdGeneratorFactory() );
config.put( FileSystemAbstraction.class, CommonFactories.defaultFileSystemAbstraction() );
NeoStore.createStore( file( "neo" ), config );
initializeStores();
assertEquals( 62 + 13, pStore.getStringBlockSize() );
assertEquals( 302 + 13, pStore.getArrayBlockSize() );
ds.close();
}
}