fixed missing WithNeo calls

decoder_test.go

@@ -38,7 +38,7 @@ func TestDecode(t *testing.T){
 	if err != nil {
 		require.Nil(t, err)
 	}
-	driverPool.Reclaim(conn)
+	defer driverPool.Reclaim(conn)
 
 
 	rows, err := dsl.QB().WithNeo(conn).Cypher(query).Query(nil)

delete_test.go

@@ -15,7 +15,7 @@ func TestDelete(t *testing.T){
 	if err != nil {
 		require.Nil(t, err)
 	}
-	driverPool.Reclaim(conn)
+	defer driverPool.Reclaim(conn)
 
 	del := a{
 		Id: 0,

save.go

@@ -81,7 +81,7 @@ func saveDepth(sess *driver.BoltConn, obj interface{}, depth int) error {
 	return relateNodes(sess, relations, ids)
 }
 
-func createNodes(sess *driver.BoltConn, crNodes map[string]map[string]nodeCreateConf) (map[string]int64, error){
+func createNodes(conn *driver.BoltConn, crNodes map[string]map[string]nodeCreateConf) (map[string]int64, error){
 	idMap := map[string]int64{}
 
 	for label, nodes := range crNodes{
@@ -124,6 +124,7 @@ func createNodes(sess *driver.BoltConn, crNodes map[string]map[string]nodeCreate
 				},
 				Alias: "id",
 			}).
+			WithNeo(conn).
 			Query(map[string]interface{}{
 				"rows": rows,
 			})
@@ -157,7 +158,7 @@ func createNodes(sess *driver.BoltConn, crNodes map[string]map[string]nodeCreate
 	return idMap, nil
 }
 
-func relateNodes(sess *driver.BoltConn, relations map[string][]relCreateConf, ids map[string]int64) error{
+func relateNodes(conn *driver.BoltConn, relations map[string][]relCreateConf, ids map[string]int64) error{
 	if relations == nil || len(relations) == 0{
 		return errors.New("relations can not be nil or empty")
 	}
@@ -245,6 +246,7 @@ func relateNodes(sess *driver.BoltConn, relations map[string][]relCreateConf, id
 				Path: mergePath,
 			}).
 			Cypher("SET rel += row.props").
+			WithNeo(conn).
 			Exec(map[string]interface{}{
 				"rows": params,
 			})

save_test.go

@@ -150,7 +150,7 @@ func TestSave(t *testing.T){
 	if err != nil {
 		require.Nil(t, err)
 	}
-	driverPool.Reclaim(conn)
+	defer driverPool.Reclaim(conn)
 
 	req.Nil(saveDepth(conn, comp2, defaultSaveDepth))
 }