/*
 * Hibernate OGM, Domain model persistence for NoSQL datastores
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
 */
package org.hibernate.ogm.datastore.neo4j.test;

import static org.fest.assertions.Assertions.assertThat;
import static org.junit.Assert.assertTrue;

import java.util.Map;
import java.util.Properties;

import org.hibernate.boot.registry.classloading.internal.ClassLoaderServiceImpl;
import org.hibernate.ogm.datastore.neo4j.Neo4jProperties;
import org.hibernate.ogm.datastore.neo4j.embedded.impl.EmbeddedNeo4jGraphDatabaseServiceFactoryProvider;
import org.hibernate.ogm.datastore.neo4j.embedded.impl.EmbeddedNeo4jInternalProperties;
import org.hibernate.ogm.datastore.neo4j.spi.GraphDatabaseServiceFactory;
import org.hibernate.ogm.datastore.neo4j.utils.EmbeddedNeo4jTestHelperDelegate;
import org.junit.Test;
import org.neo4j.graphdb.GraphDatabaseService;
import org.neo4j.graphdb.Label;
import org.neo4j.graphdb.Node;
import org.neo4j.graphdb.QueryExecutionException;
import org.neo4j.graphdb.Relationship;
import org.neo4j.graphdb.RelationshipType;
import org.neo4j.graphdb.ResourceIterable;
import org.neo4j.graphdb.ResourceIterator;
import org.neo4j.graphdb.Result;
import org.neo4j.graphdb.Transaction;
import org.neo4j.graphdb.event.KernelEventHandler;
import org.neo4j.graphdb.event.TransactionEventHandler;
import org.neo4j.graphdb.index.IndexManager;
import org.neo4j.graphdb.schema.Schema;
import org.neo4j.graphdb.traversal.BidirectionalTraversalDescription;
import org.neo4j.graphdb.traversal.TraversalDescription;
import org.neo4j.kernel.impl.factory.GraphDatabaseFacade;

/**
 * @author Davide D'Alto <davide@hibernate.org>
 */
public class Neo4jGraphDatabaseServiceLoaderTest {

	@Test
	public void testEmbeddedIsTheDefaultGraphDatabaseService() throws Exception {
		Properties properties = new Properties();
		properties.put( Neo4jProperties.DATABASE_PATH, EmbeddedNeo4jTestHelperDelegate.dbLocation() );
		EmbeddedNeo4jGraphDatabaseServiceFactoryProvider graphService = new EmbeddedNeo4jGraphDatabaseServiceFactoryProvider();
		GraphDatabaseService db = graphService.load( properties, new ClassLoaderServiceImpl() ).create();
		db.shutdown();
		assertThat( db.getClass() ). isEqualTo( GraphDatabaseFacade.class );
	}

	@Test
	public void testSelectedGraphDatabaseServiceIsLoaded() throws Exception {
		Properties properties = new Properties();
		properties.put( Neo4jProperties.DATABASE_PATH, EmbeddedNeo4jTestHelperDelegate.dbLocation() );
		properties.put( EmbeddedNeo4jInternalProperties.NEO4J_GRAPHDB_FACTORYCLASS, MockGraphServiceFactory.class.getName() );

		EmbeddedNeo4jGraphDatabaseServiceFactoryProvider graphService = new EmbeddedNeo4jGraphDatabaseServiceFactoryProvider();
		GraphDatabaseService db = graphService.load( properties, new ClassLoaderServiceImpl() ).create();
		db.shutdown();
		assertThat( db.getClass() ). isEqualTo( MockGraphDatabaseService.class );
	}

	@Test
	public void testPropertiesArePassed() throws Exception {
		Properties properties = new Properties();
		properties.put( Neo4jProperties.DATABASE_PATH, EmbeddedNeo4jTestHelperDelegate.dbLocation() );
		properties.put( EmbeddedNeo4jInternalProperties.NEO4J_GRAPHDB_FACTORYCLASS, MockGraphServiceFactory.class.getName() );
		EmbeddedNeo4jGraphDatabaseServiceFactoryProvider graphService = new EmbeddedNeo4jGraphDatabaseServiceFactoryProvider();
		MockGraphDatabaseService db = (MockGraphDatabaseService) graphService.load( properties, new ClassLoaderServiceImpl() ).create();
		db.shutdown();
		assertTrue( "GraphDatabaseService factory cannot read the configuration properties", db.isConfigurationReadable() );
	}

	public static class MockGraphServiceFactory implements GraphDatabaseServiceFactory {

		boolean configurationReadable = false;

		@Override
		public void initialize(Map<?, ?> properties) {
			configurationReadable = MockGraphServiceFactory.class.getName().equals(
					properties.get( EmbeddedNeo4jInternalProperties.NEO4J_GRAPHDB_FACTORYCLASS ) );
		}

		@Override
		public GraphDatabaseService create() {
			return new MockGraphDatabaseService( configurationReadable );
		}

	}

	public static class MockGraphDatabaseService implements GraphDatabaseService {

		private final boolean configurationReadable;

		public MockGraphDatabaseService(boolean readableConfiguration) {
			this.configurationReadable = readableConfiguration;
		}

		public boolean isConfigurationReadable() {
			return configurationReadable;
		}

		@Override
		public Node createNode() {
			return null;
		}

		@Override
		public Node getNodeById(long id) {
			return null;
		}

		@Override
		public Relationship getRelationshipById(long id) {
			return null;
		}

		@Override
		public void shutdown() {
		}

		@Override
		public Transaction beginTx() {
			return null;
		}

		@Override
		public <T> TransactionEventHandler<T> registerTransactionEventHandler(TransactionEventHandler<T> handler) {
			return null;
		}

		@Override
		public <T> TransactionEventHandler<T> unregisterTransactionEventHandler(TransactionEventHandler<T> handler) {
			return null;
		}

		@Override
		public KernelEventHandler registerKernelEventHandler(KernelEventHandler handler) {
			return null;
		}

		@Override
		public KernelEventHandler unregisterKernelEventHandler(KernelEventHandler handler) {
			return null;
		}

		@Override
		public IndexManager index() {
			return null;
		}

		@Override
		public Node createNode(Label... labels) {
			return null;
		}

		@Override
		public Schema schema() {
			return null;
		}

		@Override
		public boolean isAvailable(long timeout) {
			return false;
		}

		@Override
		public TraversalDescription traversalDescription() {
			return null;
		}

		@Override
		public BidirectionalTraversalDescription bidirectionalTraversalDescription() {
			return null;
		}

		@Override
		public ResourceIterator<Node> findNodes(Label label, String key, Object value) {
			return null;
		}

		@Override
		public Node findNode(Label label, String key, Object value) {
			return null;
		}

		@Override
		public ResourceIterator<Node> findNodes(Label label) {
			return null;
		}

		@Override
		public Result execute(String query) throws QueryExecutionException {
			return null;
		}

		@Override
		public Result execute(String query, Map<String, Object> parameters) throws QueryExecutionException {
			return null;
		}

		@Override
		public ResourceIterable<Node> getAllNodes() {
			return null;
		}

		@Override
		public ResourceIterable<Relationship> getAllRelationships() {
			return null;
		}

		@Override
		public ResourceIterable<Label> getAllLabelsInUse() {
			return null;
		}

		@Override
		public ResourceIterable<RelationshipType> getAllRelationshipTypesInUse() {
			return null;
		}

		@Override
		public ResourceIterable<Label> getAllLabels() {
			return null;
		}

		@Override
		public ResourceIterable<RelationshipType> getAllRelationshipTypes() {
			return null;
		}

		@Override
		public ResourceIterable<String> getAllPropertyKeys() {
			return null;
		}
	}
}