MetaModelMockUp.java

package org.neo4j.rdf.sparql;

import java.util.HashMap;
import java.util.Map;

import org.neo4j.graphdb.Node;
import org.neo4j.meta.model.MetaModel;
import org.neo4j.rdf.store.representation.AbstractNode;

public class MetaModelMockUp implements MetaModelProxy
{
	private Map<String, Node> referenceNodes =
		new HashMap<String, Node>();
//	private RelationshipType instanceOf;
	private Map<String, Integer> counts;
	private MetaModel metaModel;
	
	public MetaModelMockUp(
		MetaModel metaModel, Map<String, Integer> counts )
	{
		this.counts = counts;
		this.metaModel = metaModel;
	}
	
	public MetaModel getMetaModel()
	{
		return this.metaModel;
	}

	public boolean isTypeProperty( String uri )
	{
//	    return RDF.TYPE.toString().equals( uri );
		return "http://www.w3.org/1999/02/22-rdf-syntax-ns#type".equals( uri );
	}
	
	public int getCount( AbstractNode abstractNode )
	{
		Integer count = null;
		if ( abstractNode.getUriOrNull() != null )
		{
			count = this.counts.get(
				abstractNode.getUriOrNull().getUriAsString() );
		}
		return count == null ? Integer.MAX_VALUE : count;
	}
	
	public void addClassNode( String uri, Node node )
	{
	    this.referenceNodes.put( uri, node );
	}
}