Fixing some titan-test/src/test errors

More compile error fixes for the TP3 M7->M8 upgrade
thinkaurelius · Apr 10, 2015 · 748419d · 748419d
1 parent feab178
commit 748419d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/titan-test/src/test/java/com/thinkaurelius/titan/graphdb/serializer/EdgeSerializerTest.java b/titan-test/src/test/java/com/thinkaurelius/titan/graphdb/serializer/EdgeSerializerTest.java
@@ -4,13 +4,12 @@
 import com.thinkaurelius.titan.core.EdgeLabel;
 import com.thinkaurelius.titan.core.Multiplicity;
 import com.thinkaurelius.titan.core.TitanEdge;
+import com.thinkaurelius.titan.core.TitanVertex;
 import com.thinkaurelius.titan.core.schema.TitanManagement;
 import com.thinkaurelius.titan.diskstorage.Entry;
 import com.thinkaurelius.titan.graphdb.database.EdgeSerializer;
 import com.thinkaurelius.titan.graphdb.database.StandardTitanGraph;
 import com.thinkaurelius.titan.graphdb.internal.InternalRelation;
-import org.apache.tinkerpop.gremlin.structure.Edge;
-import org.apache.tinkerpop.gremlin.structure.Vertex;
 import org.junit.Test;
 
 import static org.junit.Assert.assertEquals;
@@ -30,7 +29,7 @@ public void testValueOrdering() {
         mgmt.commit();
 
         TitanVertex v1 = graph.addVertex(), v2 = graph.addVertex();
-        Edge e1 = v1.addEdge("father",v2);
+        TitanEdge e1 = v1.addEdge("father",v2);
         for (int i=1;i<=5;i++) e1.property("key"+i,i);
 
         graph.tx().commit();

diff --git a/...rc/test/java/com/thinkaurelius/titan/graphdb/serializer/SerializerGraphConfiguration.java b/...rc/test/java/com/thinkaurelius/titan/graphdb/serializer/SerializerGraphConfiguration.java
@@ -7,6 +7,7 @@
 import com.thinkaurelius.titan.graphdb.configuration.GraphDatabaseConfiguration;
 import com.thinkaurelius.titan.graphdb.database.StandardTitanGraph;
 import org.apache.tinkerpop.gremlin.structure.Vertex;
+import org.apache.tinkerpop.gremlin.structure.VertexProperty;
 import org.junit.*;
 
 import static org.junit.Assert.*;
@@ -48,7 +49,7 @@ public void testOnlyRegisteredSerialization() {
         tx.commit();
 
         tx = graph.newTransaction();
-        v = (Vertex)tx.V().has("time",5).next();
+        v = tx.query().has("time",5).vertices().iterator().next();
         assertEquals(5.0, v.<Precision>value("any").doubleValue(),0.0);
         tx.rollback();