GraphAdminTest.java

package com.collabinate.server.engine;

import org.junit.After;
import org.junit.Test;

import com.tinkerpop.blueprints.GraphFactory;
import com.tinkerpop.blueprints.KeyIndexableGraph;
import com.tinkerpop.blueprints.impls.tg.TinkerGraphFactory;

/**
 * Test class for the graph implementation of CollabinateAdmin.
 * 
 * @author mafuba
 *
 */
public class GraphAdminTest extends CollabinateAdminTest
{
	private CollabinateGraph graph;
	
	@Override
	CollabinateAdmin getAdmin()
	{
		graph = new CollabinateGraph(
				(KeyIndexableGraph)GraphFactory.open(
				"src/test/resources/graph.properties"));
		graph.setAllowCommits(false);
		return new GraphAdmin(graph);
	}
	
	@Override
	CollabinateAdmin getNewAdmin()
	{
		return new GraphAdmin(
				new CollabinateGraph(TinkerGraphFactory.createTinkerGraph()));
	}

	@After
	public void teardown()
	{
		if (null != graph)
			graph.shutdown();
	}
	
	@Test
	public void should_not_allow_null_graph()
	{
		exception.expect(IllegalArgumentException.class);
		new GraphAdmin(null);
	}
}