LEFT | RIGHT |
(no file at all) | |
1 package main_test | 1 package main_test |
2 | 2 |
3 import ( | 3 import ( |
4 . "launchpad.net/gocheck" | 4 . "launchpad.net/gocheck" |
| 5 "launchpad.net/gozk/zookeeper" |
5 main "launchpad.net/juju/go/cmd/jujud" | 6 main "launchpad.net/juju/go/cmd/jujud" |
6 "launchpad.net/juju/go/testing" | 7 » "launchpad.net/juju/go/testing" |
7 » "launchpad.net/gozk/zookeeper" | 8 » stdtesting "testing" |
8 stdtesting "testing" | |
9 ) | 9 ) |
10 | 10 |
11 var zkAddr string | 11 var zkAddr string |
12 | 12 |
13 func TestPackage(t *stdtesting.T) { | 13 func TestPackage(t *stdtesting.T) { |
14 srv := testing.StartZkServer() | 14 » srv := testing.StartZkServer() |
15 defer srv.Destroy() | 15 » defer srv.Destroy() |
16 var err error | 16 » var err error |
17 zkAddr, err = srv.Addr() | 17 » zkAddr, err = srv.Addr() |
18 if err != nil { | 18 » if err != nil { |
19 t.Fatalf("could not get ZooKeeper server address") | 19 » » t.Fatalf("could not get ZooKeeper server address") |
20 } | 20 } |
21 TestingT(t) | 21 » TestingT(t) |
22 } | 22 } |
23 | 23 |
24 type InitzkSuite struct{ | 24 type InitzkSuite struct { |
25 zkConn *zookeeper.Conn | 25 » zkConn *zookeeper.Conn |
26 path string | 26 » path string |
27 } | 27 } |
28 | 28 |
29 var _ = Suite(&InitzkSuite{}) | 29 var _ = Suite(&InitzkSuite{}) |
30 | 30 |
31 func (s *InitzkSuite) SetUpTest(c *C) { | 31 func (s *InitzkSuite) SetUpTest(c *C) { |
32 zk, session, err := zookeeper.Dial(zkAddr, 15e9) | 32 » zk, session, err := zookeeper.Dial(zkAddr, 15e9) |
33 c.Assert(err, IsNil) | 33 » c.Assert(err, IsNil) |
34 event := <-session | 34 » event := <-session |
35 c.Assert(event.Ok(), Equals, true) | 35 » c.Assert(event.Ok(), Equals, true) |
36 c.Assert(event.Type, Equals, zookeeper.EVENT_SESSION) | 36 » c.Assert(event.Type, Equals, zookeeper.EVENT_SESSION) |
37 c.Assert(event.State, Equals, zookeeper.STATE_CONNECTED) | 37 » c.Assert(event.State, Equals, zookeeper.STATE_CONNECTED) |
38 | 38 |
39 s.zkConn = zk | 39 » s.zkConn = zk |
40 s.path = "/watcher" | 40 » s.path = "/watcher" |
41 | 41 |
42 c.Assert(err, IsNil) | 42 » c.Assert(err, IsNil) |
43 } | 43 } |
44 | 44 |
45 func (s *InitzkSuite) TearDownTest(c *C) { | 45 func (s *InitzkSuite) TearDownTest(c *C) { |
46 testing.ZkRemoveTree(s.zkConn, s.path) | 46 » testing.ZkRemoveTree(s.zkConn, s.path) |
47 s.zkConn.Close() | 47 » s.zkConn.Close() |
48 } | 48 } |
49 | 49 |
50 func initInitzkCommand(args []string) (*main.InitzkCommand, error) { | 50 func initInitzkCommand(args []string) (*main.InitzkCommand, error) { |
51 c := &main.InitzkCommand{} | 51 c := &main.InitzkCommand{} |
52 return c, initCmd(c, args) | 52 return c, initCmd(c, args) |
53 } | 53 } |
54 | 54 |
55 func (s *InitzkSuite) TestParse(c *C) { | 55 func (s *InitzkSuite) TestParse(c *C) { |
56 args := []string{} | 56 args := []string{} |
57 _, err := initInitzkCommand(args) | 57 _, err := initInitzkCommand(args) |
(...skipping 12 matching lines...) Expand all Loading... |
70 | 70 |
71 args = append(args, "--zookeeper-servers", "zk1:2181,zk2:2181") | 71 args = append(args, "--zookeeper-servers", "zk1:2181,zk2:2181") |
72 izk, err = initInitzkCommand(args) | 72 izk, err = initInitzkCommand(args) |
73 c.Assert(err, IsNil) | 73 c.Assert(err, IsNil) |
74 c.Assert(izk.StateInfo.Addrs, DeepEquals, []string{"zk1:2181", "zk2:2181
"}) | 74 c.Assert(izk.StateInfo.Addrs, DeepEquals, []string{"zk1:2181", "zk2:2181
"}) |
75 | 75 |
76 args = append(args, "haha disregard that") | 76 args = append(args, "haha disregard that") |
77 _, err = initInitzkCommand(args) | 77 _, err = initInitzkCommand(args) |
78 c.Assert(err, ErrorMatches, `unrecognized args: \["haha disregard that"\
]`) | 78 c.Assert(err, ErrorMatches, `unrecognized args: \["haha disregard that"\
]`) |
79 } | 79 } |
LEFT | RIGHT |