SimpleSpec.scala

import gremlin.scala._
import org.apache.commons.configuration.BaseConfiguration
import org.apache.tinkerpop.gremlin.driver.Cluster
import org.apache.tinkerpop.gremlin.driver.remote.DriverRemoteConnection
import org.apache.tinkerpop.gremlin.driver.ser.GryoMessageSerializerV3d0
import org.apache.tinkerpop.gremlin.structure.io.gryo.GryoMapper 
import org.apache.tinkerpop.gremlin.structure.util.empty.EmptyGraph
import org.janusgraph.graphdb.tinkerpop.JanusGraphIoRegistry
import org.janusgraph.core.JanusGraphFactory
import org.janusgraph.core.JanusGraph
import org.scalatest.{Matchers, WordSpec}


class SimpleSpec extends WordSpec with Matchers {

  "inmemory janusgraph instance" in {
    val conf = new BaseConfiguration()
    conf.setProperty("storage.backend","inmemory")
    val graph = JanusGraphFactory.open(conf)
    implicit val scalaGraph = graph.asScala

    val Name = Key[String]("name")
    val Planet = "planet"
    val Saturn = "saturn"

    (1 to 4) foreach { i ⇒
      graph + (Planet, Name -> s"vertex $i")
    }
    val saturnV = graph + (Saturn, Name -> Saturn)
    val sunV = graph + ("sun", Name -> "sun")
    saturnV --- "orbits" --> sunV

    graph.V.count.head shouldBe 6
    graph.E.count.head shouldBe 1

    val traversal = graph.V.value(Name)
    traversal.toList.size shouldBe 6

    graph.V.hasLabel(Saturn).count.head shouldBe 1

    val saturnQ = graph.V.hasLabel(Saturn).head
    saturnQ.value2(Name) shouldBe Saturn

    graph.close
  }

  /** 
    * prerequisite: 
    * download and extract janusgraph-0.3.1-hadoop2
    * start it with `bin/janusgraph.sh -v start`
    * note: you can stop it with `bin/janusgraph.sh -v stop`
    */
  "remote janusgraph" when {

    "not doing anything fancy" in {
      val serializer = new GryoMessageSerializerV3d0(GryoMapper.build.addRegistry(JanusGraphIoRegistry.getInstance))
      val cluster = Cluster.build.addContactPoint("localhost").port(8182).serializer(serializer).create
      implicit val graph = EmptyGraph.instance.asScala.configure(_.withRemote(DriverRemoteConnection.using(cluster)))

      graph.traversal.V.drop.iterate

      val Name = Key[String]("name")
      val Planet = "planet"

      val saturn: Vertex = graph + (Planet, Name -> "saturn")
      val sun   : Vertex = graph + (Planet, Name -> "sun")

      saturn --- "orbits" --> sun

      graph.V.count.head shouldBe 2
      graph.E.count.head shouldBe 1

      // println(graph.V.toList)
      // println(graph.E.toList)

      cluster.close()
    }

  }
}