src/test/java/com/thinkaurelius/titan/graphdb/thrift/ThriftGraphIterativeTest.java

   1 package com.thinkaurelius.titan.graphdb.thrift;
   2
   3 import com.thinkaurelius.titan.CassandraStorageSetup;
   4 import com.thinkaurelius.titan.diskstorage.BackendException;
   5 import com.thinkaurelius.titan.diskstorage.cassandra.thrift.CassandraThriftStoreManager;
   6 import com.thinkaurelius.titan.diskstorage.configuration.BasicConfiguration;
   7 import com.thinkaurelius.titan.diskstorage.configuration.WriteConfiguration;
   8 import com.thinkaurelius.titan.diskstorage.keycolumnvalue.KeyColumnValueStoreManager;
   9 import com.thinkaurelius.titan.graphdb.TitanGraphIterativeBenchmark;
  10 import com.thinkaurelius.titan.graphdb.configuration.GraphDatabaseConfiguration;
  11 import org.junit.BeforeClass;
  12
  13 public class ThriftGraphIterativeTest extends TitanGraphIterativeBenchmark {
  14
  15     @Override
  16     public WriteConfiguration getConfiguration() {
  17         return CassandraStorageSetup.getCassandraThriftGraphConfiguration(getClass().getSimpleName());
  18     }
  19
  20     @Override
  21     public KeyColumnValueStoreManager openStorageManager() throws BackendException {
  22         return new CassandraThriftStoreManager(new BasicConfiguration(GraphDatabaseConfiguration.ROOT_NS,getConfiguration(), BasicConfiguration.Restriction.NONE));
  23     }
  24
  25
  26     @BeforeClass
  27     public static void beforeClass() {
  28         CassandraStorageSetup.startCleanEmbedded();
  29     }
  30 }