multi-label test

Author: swilly22

client_test.go

@@ -430,6 +430,29 @@ func TestUtils(t *testing.T) {
 	assert.Equal(t, res, "{object: {foo: 1}}")
 }
 
+func TestMultiLabelNode(t *testing.T) {
+	// clear database
+	graph.Flush()
+	err := graph.Delete()
+	assert.Nil(t, err)
+
+	// create a multi label node
+	multiLabelNode := NodeNew([]string{"A","B"}, nil)
+	graph.AddNode(multiLabelNode)
+	res, err := graph.Commit()
+
+	// fetch node
+	res, err = graph.Query("MATCH (n) RETURN n")
+	res.Next()
+	r := res.Record()
+	n := r.GetByIndex(0).(*Node)
+
+	// expecting 2 labels
+	assert.Equal(t, len(n.Labels), 2, "expecting 2 labels")
+	assert.Equal(t, n.Labels[0], "A")
+	assert.Equal(t, n.Labels[1], "B")
+}
+
 func TestNodeMapDatatype(t *testing.T) {
 	graph.Flush()
 	err := graph.Delete()

edge.go

@@ -17,6 +17,7 @@ type Edge struct {
 	graph       *Graph
 }
 
+// Create a new Edge
 func EdgeNew(relation string, srcNode *Node, destNode *Node, properties map[string]interface{}) *Edge {
 	p := properties
 	if p == nil {
@@ -32,15 +33,18 @@ func EdgeNew(relation string, srcNode *Node, destNode *Node, properties map[stri
 	}
 }
 
+// Assign a new property to edge
 func (e *Edge) SetProperty(key string, value interface{}) {
 	e.Properties[key] = value
 }
 
+// Retrieves property from edge
 func (e *Edge) GetProperty(key string) interface{} {
 	v, _ := e.Properties[key]
 	return v
 }
 
+// Returns edge source node ID
 func (e Edge) SourceNodeID() uint64 {
 	if e.Source != nil {
 		return e.Source.ID
@@ -49,6 +53,7 @@ func (e Edge) SourceNodeID() uint64 {
 	}
 }
 
+// Returns edge destination node ID
 func (e Edge) DestNodeID() uint64 {
 	if e.Source != nil {
 		return e.Destination.ID
@@ -57,6 +62,7 @@ func (e Edge) DestNodeID() uint64 {
 	}
 }
 
+// Returns a string representation of edge
 func (e Edge) String() string {
 	if len(e.Properties) == 0 {
 		return "{}"
@@ -71,6 +77,7 @@ func (e Edge) String() string {
 	return s
 }
 
+// String makes Edge satisfy the Stringer interface
 func (e Edge) Encode() string {
 	s := []string{"(", e.Source.Alias, ")"}
 

node.go

@@ -5,7 +5,7 @@ import (
 	"strings"
 )
 
-// Node represents a node within a graph.
+// Node represents a node within a graph
 type Node struct {
 	ID         uint64
 	Labels     []string
@@ -14,6 +14,7 @@ type Node struct {
 	graph      *Graph
 }
 
+// Create a new Node
 func NodeNew(labels []string, alias string, properties map[string]interface{}) *Node {
 
 	p := properties
@@ -29,15 +30,18 @@ func NodeNew(labels []string, alias string, properties map[string]interface{}) *
 	}
 }
 
+// Asssign a new property to node
 func (n *Node) SetProperty(key string, value interface{}) {
 	n.Properties[key] = value
 }
 
+// Retrieves property from node
 func (n Node) GetProperty(key string) interface{} {
 	v, _ := n.Properties[key]
 	return v
 }
 
+// Returns a string representation of a node
 func (n Node) String() string {
 	if len(n.Properties) == 0 {
 		return "{}"
@@ -52,7 +56,7 @@ func (n Node) String() string {
 	return s
 }
 
-// String makes Node satisfy the Stringer interface.
+// String makes Node satisfy the Stringer interface
 func (n Node) Encode() string {
 	s := []string{"("}
 

query_result.go

@@ -59,7 +59,7 @@ type QueryResult struct {
 	header             QueryResultHeader
 	results            []*Record
 	statistics         map[string]float64
-	current_record_idx int
+	currentRecordIdx   int
 }
 
 func QueryResultNew(g *Graph, response interface{}) (*QueryResult, error) {
@@ -175,10 +175,10 @@ func (qr *QueryResult) parseNode(cell interface{}) *Node {
 
 	c, _ := redis.Values(cell, nil)
 	id, _ := redis.Uint64(c[0], nil)
-	label_ids, _ := redis.Ints(c[1], nil)
-	labels := make([]string, len(label_ids))
-	for i := 0; i < len(label_ids); i++ {
-		labels[i] = qr.graph.getLabel(label_ids[i])
+	labelIds, _ := redis.Ints(c[1], nil)
+	labels := make([]string, len(labelIds))
+	for i := 0; i < len(labelIds); i++ {
+		labels[i] = qr.graph.getLabel(labelIds[i])
 	}
 
 	rawProps, _ := redis.Values(c[2], nil)