Restore watches on authentication success

zk/conn.go

@@ -294,6 +294,7 @@ func (c *Conn) loop() {
 				wg.Done()
 			}()
 
+			c.sendSetWatches()
 			wg.Wait()
 		}
 
@@ -426,8 +427,6 @@ func (c *Conn) authenticate() error {
 		return err
 	}
 
-	c.sendSetWatches()
-
 	// connect response
 
 	// package length

zk/zk_test.go

@@ -297,10 +297,12 @@ func TestSetWatchers(t *testing.T) {
 		t.Fatal("Children should return at least 1 child")
 	}
 
+	// Simulate network error by brutally closing the network connection.
 	zk.conn.Close()
 	if err := zk2.Delete(testPath, -1); err != nil && err != ErrNoNode {
 		t.Fatalf("Delete returned error: %+v", err)
 	}
+	// Allow some time for the `zk` session to reconnect and set watches.
 	time.Sleep(time.Millisecond * 100)
 
 	if path, err := zk2.Create("/gozk-test", []byte{1, 2, 3, 4}, 0, WorldACL(PermAll)); err != nil {