Fix race conditions in t test

samuel · Jan 30, 2018 · 07f1d62 · 07f1d62
1 parent ff6eb7f
commit 07f1d62
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/zk/conn_test.go b/zk/conn_test.go
@@ -30,7 +30,7 @@ func TestRecurringReAuthHang(t *testing.T) {
 	if err != nil {
 		panic(err)
 	}
-	for conn.state != StateHasSession {
+	for conn.State() != StateHasSession {
 		time.Sleep(50 * time.Millisecond)
 	}
 
@@ -40,7 +40,7 @@ func TestRecurringReAuthHang(t *testing.T) {
 	}()
 
 	// Add auth.
-	conn.creds = append(conn.creds, authCreds{"digest", []byte("test:test")})
+	conn.AddAuth("digest", []byte("test:test"))
 
 	currentServer := conn.server
 	conn.debugCloseRecvLoop = true