Browse Source

Use Pool.Pick() rather than Node.Pick()

Nimi Wariboko Jr 11 years ago
parent
commit
91a482abd0
1 changed files with 2 additions and 2 deletions
  1. 2 2
      cassandra_test.go

+ 2 - 2
cassandra_test.go

@@ -585,7 +585,7 @@ func TestReprepareStatement(t *testing.T) {
 		t.Fatal("create:", err)
 		t.Fatal("create:", err)
 	}
 	}
 	stmt := "INSERT INTO test_reprepare_statement (foo, bar) VALUES (?, 7)"
 	stmt := "INSERT INTO test_reprepare_statement (foo, bar) VALUES (?, 7)"
-	conn := session.Node.Pick(nil)
+	conn := session.Pool.Pick(nil)
 	conn.prepMu.Lock()
 	conn.prepMu.Lock()
 	flight := new(inflightPrepare)
 	flight := new(inflightPrepare)
 	conn.prep[stmt] = flight
 	conn.prep[stmt] = flight
@@ -626,7 +626,7 @@ func TestReprepareBatch(t *testing.T) {
 		t.Fatal("create:", err)
 		t.Fatal("create:", err)
 	}
 	}
 	stmt := "INSERT INTO test_reprepare_statement_batch (foo, bar) VALUES (?, 7)"
 	stmt := "INSERT INTO test_reprepare_statement_batch (foo, bar) VALUES (?, 7)"
-	conn := session.Node.Pick(nil)
+	conn := session.Pool.Pick(nil)
 	conn.prepMu.Lock()
 	conn.prepMu.Lock()
 	flight := new(inflightPrepare)
 	flight := new(inflightPrepare)
 	conn.prep[stmt] = flight
 	conn.prep[stmt] = flight