func (s *S) TestDelete(c *C) { conn, _ := s.init(c) _, err := conn.Create("/test", "", zk.EPHEMERAL, zk.WorldACL(zk.PERM_ALL)) c.Assert(err, IsNil) err = conn.Delete("/test", 5) c.Assert(err, NotNil) c.Check(zk.IsError(err, zk.ZBADVERSION), Equals, true, Commentf("%v", err)) err = conn.Delete("/test", -1) c.Assert(err, IsNil) err = conn.Delete("/test", -1) c.Assert(err, NotNil) c.Check(zk.IsError(err, zk.ZNONODE), Equals, true, Commentf("%v", err)) }
// By default, the ZooKeeper C client will hang indefinitely if a // handler is closed twice. We get in the way and prevent it. func (s *S) TestClosingTwiceDoesntHang(c *C) { conn, _ := s.init(c) err := conn.Close() c.Assert(err, IsNil) err = conn.Close() c.Assert(err, NotNil) c.Check(zk.IsError(err, zk.ZCLOSING), Equals, true, Commentf("%v", err)) }
func (s *S) TestGetAndError(c *C) { conn, _ := s.init(c) data, stat, err := conn.Get("/non-existent") c.Assert(data, Equals, "") c.Assert(stat, IsNil) c.Assert(err, ErrorMatches, `zookeeper: get "/non-existent": no node`) c.Check(zk.IsError(err, zk.ZNONODE), Equals, true, Commentf("%v", err)) }
func (s *S) TestChildren(c *C) { conn, _ := s.init(c) children, stat, err := conn.Children("/") c.Assert(err, IsNil) c.Assert(children, DeepEquals, []string{"zookeeper"}) c.Assert(stat.NumChildren(), Equals, 1) children, stat, err = conn.Children("/non-existent") c.Check(zk.IsError(err, zk.ZNONODE), Equals, true, Commentf("%v", err)) c.Assert(children, IsNil) c.Assert(stat, IsNil) }
func (s *S) TestGetAndWatchWithError(c *C) { c.Check(zk.CountPendingWatches(), Equals, 0) conn, _ := s.init(c) c.Check(zk.CountPendingWatches(), Equals, 1) _, _, watch, err := conn.GetW("/test") c.Assert(err, NotNil) c.Check(zk.IsError(err, zk.ZNONODE), Equals, true, Commentf("%v", err)) c.Assert(watch, IsNil) c.Check(zk.CountPendingWatches(), Equals, 1) }
func (s *S) TestExistsAndWatchWithError(c *C) { c.Check(zk.CountPendingWatches(), Equals, 0) conn, _ := s.init(c) c.Check(zk.CountPendingWatches(), Equals, 1) stat, watch, err := conn.ExistsW("///") c.Assert(err, NotNil) c.Check(zk.IsError(err, zk.ZBADARGUMENTS), Equals, true, Commentf("%v", err)) c.Assert(stat, IsNil) c.Assert(watch, IsNil) c.Check(zk.CountPendingWatches(), Equals, 1) }
func (s *S) TestCreateAndGet(c *C) { conn, _ := s.init(c) path, err := conn.Create("/test-", "bababum", zk.SEQUENCE|zk.EPHEMERAL, zk.WorldACL(zk.PERM_ALL)) c.Assert(err, IsNil) c.Assert(path, Matches, "/test-[0-9]+") // Check the error condition from Create(). _, err = conn.Create(path, "", zk.EPHEMERAL, zk.WorldACL(zk.PERM_ALL)) c.Check(zk.IsError(err, zk.ZNODEEXISTS), Equals, true, Commentf("%v", err)) data, _, err := conn.Get(path) c.Assert(err, IsNil) c.Assert(data, Equals, "bababum") }
func (s *S) TestSetACL(c *C) { conn, _ := s.init(c) _, err := conn.Create("/test", "", zk.EPHEMERAL, zk.WorldACL(zk.PERM_ALL)) c.Assert(err, IsNil) err = conn.SetACL("/test", zk.WorldACL(zk.PERM_ALL), 5) c.Assert(err, NotNil) c.Check(zk.IsError(err, zk.ZBADVERSION), Equals, true, Commentf("%v", err)) err = conn.SetACL("/test", zk.WorldACL(zk.PERM_READ), -1) c.Assert(err, IsNil) acl, _, err := conn.ACL("/test") c.Assert(err, IsNil) c.Assert(acl, DeepEquals, zk.WorldACL(zk.PERM_READ)) }
func (s *S) TestAddAuth(c *C) { conn, _ := s.init(c) acl := []zk.ACL{{zk.PERM_READ, "digest", "joe:enQcM3mIEHQx7IrPNStYBc0qfs8="}} _, err := conn.Create("/test", "", zk.EPHEMERAL, acl) c.Assert(err, IsNil) _, _, err = conn.Get("/test") c.Assert(err, NotNil) c.Check(zk.IsError(err, zk.ZNOAUTH), Equals, true, Commentf("%v", err)) err = conn.AddAuth("digest", "joe:passwd") c.Assert(err, IsNil) _, _, err = conn.Get("/test") c.Assert(err, IsNil) }
func (s *S) TestACL(c *C) { conn, _ := s.init(c) _, err := conn.Create("/test", "", zk.EPHEMERAL, zk.WorldACL(zk.PERM_ALL)) c.Assert(err, IsNil) acl, stat, err := conn.ACL("/test") c.Assert(err, IsNil) c.Assert(acl, DeepEquals, zk.WorldACL(zk.PERM_ALL)) c.Assert(stat, NotNil) c.Assert(stat.Version(), Equals, 0) acl, stat, err = conn.ACL("/non-existent") c.Assert(err, NotNil) c.Check(zk.IsError(err, zk.ZNONODE), Equals, true, Commentf("%v", err)) c.Assert(acl, IsNil) c.Assert(stat, IsNil) }
func (s *S) TestRecvTimeoutInitParameter(c *C) { conn, watch, err := zk.Dial(s.zkAddr, 0) c.Assert(err, IsNil) defer conn.Close() select { case <-watch: c.Fatal("Watch fired") default: } for i := 0; i != 1000; i++ { _, _, err := conn.Get("/zookeeper") if err != nil { c.Check(zk.IsError(err, zk.ZOPERATIONTIMEOUT), Equals, true, Commentf("%v", err)) c.SucceedNow() } } c.Fatal("Operation didn't timeout") }
func (s *S) TestRetryChangeFailsCreating(c *C) { conn, _ := s.init(c) // Read only! _, err := conn.Create("/test", "old", zk.EPHEMERAL, zk.WorldACL(zk.PERM_READ)) c.Assert(err, IsNil) var called bool err = conn.RetryChange("/test/sub", zk.EPHEMERAL, zk.WorldACL(zk.PERM_ALL), func(data string, stat *zk.Stat) (string, error) { called = true return "", nil }) c.Assert(err, NotNil) c.Check(zk.IsError(err, zk.ZNOAUTH), Equals, true, Commentf("%v", err)) stat, err := conn.Exists("/test/sub") c.Assert(err, IsNil) c.Assert(stat, IsNil) c.Assert(called, Equals, true) }
func (s *S) TestConcurrentClose(c *C) { // make sure the server is ready to receive connections. s.init(c) // Close should wait until all outstanding requests have // completed before returning. The idea of this test is that // any request that requests or changes a zookeeper node must // make at least one round trip to the server, so we interpose a // proxy between the client and the server which can stop all // incoming traffic on demand, thus blocking the request until // we want it to unblock. // // We assume that all requests take less than 0.1s to complete, // thus when we wait below, neither of the above goroutines // should complete within the allotted time (the request because // it's waiting for a reply from the server and the close // because it's waiting for the request to complete). If the // locking doesn't work, the Close will return early. If the // proxy blocking doesn't work, the request will return early. // // When we reenable incoming messages from the server, both // goroutines should complete. We can't tell which completes // first, but the fact that the close blocked is sufficient to // tell that the locking is working correctly. for i, f := range requestFuncs { c.Logf("iter %d", i) p := newProxy(c, s.zkAddr) conn, watch, err := zk.Dial(p.addr(), 5e9) c.Assert(err, IsNil) c.Assert((<-watch).Ok(), Equals, true) // sanity check that the connection is actually // up and running. _, err = conn.Exists("/nothing") c.Assert(err, IsNil) p.stopIncoming() reqDone := make(chan bool) closeDone := make(chan bool) go func() { f(conn, "/closetest") reqDone <- true }() go func() { // sleep for long enough for the request to be initiated and the read lock taken. time.Sleep(0.05e9) conn.Close() closeDone <- true }() select { case <-reqDone: c.Fatalf("request %d finished early", i) case <-closeDone: c.Fatalf("request %d close finished early", i) case <-time.After(0.1e9): } p.startIncoming() for reqDone != nil || closeDone != nil { select { case <-reqDone: reqDone = nil case <-closeDone: closeDone = nil case <-time.After(0.4e9): c.Fatalf("request %d timed out waiting for req (%p) and close(%p)", i, reqDone, closeDone) } } p.close() err = f(conn, "/closetest") c.Check(zk.IsError(err, zk.ZCLOSING), Equals, true, Commentf("%v", err)) } }