test improvements

Author: epswartz

client_test.go

@@ -23,28 +23,48 @@ func TestExample(t *testing.T) {
 			"status": "single",
 		},
 	}
-	rg.AddNode(&john)
+	err := rg.AddNode(&john)
+	if err != nil {
+		t.Error(err)
+		t.Fail()
+	}
 
 	japan := Node{
 		Label: "country",
 		Properties: map[string]interface{}{
 			"name": "Japan",
 		},
 	}
-	rg.AddNode(&japan)
+	err = rg.AddNode(&japan)
+	if err != nil {
+		t.Error(err)
+		t.Fail()
+	}
 
 	edge := Edge{
 		Source:      &john,
 		Relation:    "visited",
 		Destination: &japan,
 	}
-	rg.AddEdge(&edge)
+	err = rg.AddEdge(&edge)
+	if err != nil {
+		t.Error(err)
+		t.Fail()
+	}
 
-	rg.Commit()
+	_, err = rg.Commit()
+	if err != nil {
+		t.Error(err)
+		t.Fail()
+	}
 
 	query := `MATCH (p:person)-[v:visited]->(c:country)
 		   RETURN p.name, p.age, v.purpose, c.name`
-	rs, _ := rg.Query(query)
+	rs, err := rg.Query(query)
+	if err != nil {
+		t.Error(err)
+		t.Fail()
+	}
 
 	rs.PrettyPrint()
 }
@@ -63,13 +83,26 @@ func TestFlush(t *testing.T) {
 				"name": fmt.Sprintf("%s Rubble", user),
 			},
 		}
-		rg.AddNode(&family)
-		rg.Flush()
+		err := rg.AddNode(&family)
+		if err != nil {
+			t.Error(err)
+			t.Fail()
+		}
+		_, err = rg.Flush()
+		if err != nil {
+			t.Error(err)
+			t.Fail()
+		}
 	}
 	query := `MATCH (p:person) RETURN p.name`
-	rs, _ := rg.Query(query)
+	rs, err := rg.Query(query)
+	if err != nil {
+		t.Error(err)
+		t.Fail()
+	}
 	if len(rs.Results) > 4 {
 		t.Errorf("There Should only be 4 entries but we get: %d", len(rs.Results))
+		t.Fail()
 	}
 
 }