Move graph entities

README.md

@@ -22,6 +22,7 @@ RedisGraph Java client
 ```
 
 ### Snapshots
+To be used with RedisGraph 2.0 (not officially released)
 
 ```xml
   <repositories>
@@ -39,7 +40,7 @@ and
     <dependency>
       <groupId>com.redislabs</groupId>
       <artifactId>jredisgraph</artifactId>
-      <version>1.0.5-SNAPSHOT</version>
+      <version>2.0.0-SNAPSHOT</version>
     </dependency>
   </dependencies>
 ```

src/main/java/com/redislabs/redisgraph/RedisGraph.java

@@ -1,6 +1,6 @@
 package com.redislabs.redisgraph;
 
-import com.redislabs.redisgraph.impl.GraphCache;
+import com.redislabs.redisgraph.impl.graph_cache.GraphCache;
 import com.redislabs.redisgraph.impl.ResultSetImpl;
 import org.apache.commons.text.translate.AggregateTranslator;
 import org.apache.commons.text.translate.CharSequenceTranslator;

src/main/java/com/redislabs/redisgraph/impl/Edge.java → src/main/java/com/redislabs/redisgraph/graph_entities/Edge.java

@@ -1,4 +1,4 @@
-package com.redislabs.redisgraph.impl;
+package com.redislabs.redisgraph.graph_entities;
 
 import java.util.Objects;
 

src/main/java/com/redislabs/redisgraph/impl/GraphEntity.java → src/main/java/com/redislabs/redisgraph/graph_entities/GraphEntity.java

@@ -1,4 +1,4 @@
-package com.redislabs.redisgraph.impl;
+package com.redislabs.redisgraph.graph_entities;
 
 
 import com.redislabs.redisgraph.ResultSet.ResultSetScalarTypes;
@@ -51,6 +51,14 @@ public void addProperty(String name, ResultSetScalarTypes type, Object value){
 
     }
 
+    /**
+     *
+     * @return Entity's property names, as a Set
+     */
+    public Set<String> getEntityPropertyNames(){
+        return propertyMap.keySet();
+    }
+
     /**
      * Add a property to the entity
      * @param property

src/main/java/com/redislabs/redisgraph/impl/Node.java → src/main/java/com/redislabs/redisgraph/graph_entities/Node.java

@@ -1,4 +1,4 @@
-package com.redislabs.redisgraph.impl;
+package com.redislabs.redisgraph.graph_entities;
 
 import java.util.ArrayList;
 import java.util.List;

src/main/java/com/redislabs/redisgraph/impl/Property.java → src/main/java/com/redislabs/redisgraph/graph_entities/Property.java

@@ -1,4 +1,4 @@
-package com.redislabs.redisgraph.impl;
+package com.redislabs.redisgraph.graph_entities;
 
 import com.redislabs.redisgraph.ResultSet;
 

src/main/java/com/redislabs/redisgraph/impl/ResultSetImpl.java

@@ -1,9 +1,11 @@
 package com.redislabs.redisgraph.impl;
 
-import com.redislabs.redisgraph.Header;
-import com.redislabs.redisgraph.Record;
-import com.redislabs.redisgraph.ResultSet;
-import com.redislabs.redisgraph.Statistics;
+import com.redislabs.redisgraph.*;
+import com.redislabs.redisgraph.graph_entities.Edge;
+import com.redislabs.redisgraph.graph_entities.GraphEntity;
+import com.redislabs.redisgraph.graph_entities.Node;
+import com.redislabs.redisgraph.graph_entities.Property;
+import com.redislabs.redisgraph.impl.graph_cache.GraphCache;
 import redis.clients.jedis.util.SafeEncoder;
 
 import java.util.ArrayList;

src/main/java/com/redislabs/redisgraph/impl/GraphCache.java → src/main/java/com/redislabs/redisgraph/impl/graph_cache/GraphCache.java

@@ -1,4 +1,4 @@
-package com.redislabs.redisgraph.impl;
+package com.redislabs.redisgraph.impl.graph_cache;
 
 import com.redislabs.redisgraph.RedisGraph;
 

src/main/java/com/redislabs/redisgraph/impl/GraphCacheList.java → src/main/java/com/redislabs/redisgraph/impl/graph_cache/GraphCacheList.java

@@ -1,4 +1,4 @@
-package com.redislabs.redisgraph.impl;
+package com.redislabs.redisgraph.impl.graph_cache;
 
 import com.redislabs.redisgraph.Record;
 import com.redislabs.redisgraph.RedisGraph;

src/test/java/com/redislabs/redisgraph/RedisGraphAPITest.java

@@ -7,9 +7,9 @@
 import java.util.stream.Collectors;
 import java.util.stream.IntStream;
 
-import com.redislabs.redisgraph.impl.Edge;
-import com.redislabs.redisgraph.impl.Node;
-import com.redislabs.redisgraph.impl.Property;
+import com.redislabs.redisgraph.graph_entities.Edge;
+import com.redislabs.redisgraph.graph_entities.Node;
+import com.redislabs.redisgraph.graph_entities.Property;
 import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
@@ -280,6 +280,10 @@ public void testRecord(){
                 place, since, doubleValue, false, null),
                 record.values());
 
+        Node a = record.getValue("a");
+        for (String propertyName : expectedNode.getEntityPropertyNames()){
+            Assert.assertEquals(expectedNode.getProperty(propertyName) ,a.getProperty(propertyName));
+        }
 
         Assert.assertEquals( "roi", record.getString(2));
     	Assert.assertEquals( "32", record.getString(3));
@@ -407,11 +411,6 @@ public void testAdditionToProcedures(){
         Assert.assertNotNull(api.query("social", "CREATE (:person{name:'amit',age:30})"));
         Assert.assertNotNull(api.query("social", "MATCH (a:person), (b:person) WHERE (a.name = 'roi' AND b.name='amit')  CREATE (a)-[:knows]->(b)"));
 
-
-        List<ResultSet> resultSets = IntStream.range(0,16).parallel().
-                mapToObj(i-> api.query("social", "MATCH (a:person)-[r:knows]->(b:person) RETURN a,r")).
-                collect(Collectors.toList());
-
         //expected objects init
         Property nameProperty = new Property("name", ResultSet.ResultSetScalarTypes.PROPERTY_STRING, "roi");
         Property ageProperty = new Property("age", ResultSet.ResultSetScalarTypes.PROPERTY_INTEGER, 32);