11import threading
22
3- from nose .tools import eq_
4-
53from kazoo .testing import KazooTestCase
64
75
86class KazooBarrierTests (KazooTestCase ):
97 def test_barrier_not_exist (self ):
108 b = self .client .Barrier ("/some/path" )
11- eq_ ( b .wait (), True )
9+ assert b .wait ()
1210
1311 def test_barrier_exists (self ):
1412 b = self .client .Barrier ("/some/path" )
1513 b .create ()
16- eq_ ( b .wait (0 ), False )
14+ assert not b .wait (0 )
1715 b .remove ()
18- eq_ ( b .wait (), True )
16+ assert b .wait ()
1917
2018 def test_remove_nonexistent_barrier (self ):
2119 b = self .client .Barrier ("/some/path" )
22- eq_ ( b .remove (), False )
20+ assert not b .remove ()
2321
2422
2523class KazooDoubleBarrierTests (KazooTestCase ):
26-
2724 def test_basic_barrier (self ):
2825 b = self .client .DoubleBarrier ("/some/path" , 1 )
29- eq_ ( b .participating , False )
26+ assert not b .participating
3027 b .enter ()
31- eq_ ( b .participating , True )
28+ assert b .participating
3229 b .leave ()
33- eq_ ( b .participating , False )
30+ assert not b .participating
3431
3532 def test_two_barrier (self ):
3633 av = threading .Event ()
@@ -61,23 +58,23 @@ def make_barrier_two():
6158 t2 = threading .Thread (target = make_barrier_two )
6259 t2 .start ()
6360
64- eq_ ( b1 .participating , False )
65- eq_ ( b2 .participating , False )
61+ assert not b1 .participating
62+ assert not b2 .participating
6663
6764 bv .set ()
6865 av .wait ()
6966 ev .wait ()
70- eq_ ( b1 .participating , True )
71- eq_ ( b2 .participating , True )
67+ assert b1 .participating
68+ assert b2 .participating
7269
7370 av .clear ()
7471 ev .clear ()
7572
7673 release_all .set ()
7774 av .wait ()
7875 ev .wait ()
79- eq_ ( b1 .participating , False )
80- eq_ ( b2 .participating , False )
76+ assert not b1 .participating
77+ assert not b2 .participating
8178 t1 .join ()
8279 t2 .join ()
8380
@@ -111,19 +108,19 @@ def make_barrier_two():
111108 t2 = threading .Thread (target = make_barrier_two )
112109 t2 .start ()
113110
114- eq_ ( b1 .participating , False )
115- eq_ ( b2 .participating , False )
111+ assert not b1 .participating
112+ assert not b2 .participating
116113
117114 bv .set ()
118- eq_ ( b1 .participating , False )
119- eq_ ( b2 .participating , False )
115+ assert not b1 .participating
116+ assert not b2 .participating
120117 b3 .enter ()
121118 ev .wait ()
122119 av .wait ()
123120
124- eq_ ( b1 .participating , True )
125- eq_ ( b2 .participating , True )
126- eq_ ( b3 .participating , True )
121+ assert b1 .participating
122+ assert b2 .participating
123+ assert b3 .participating
127124
128125 av .clear ()
129126 ev .clear ()
@@ -132,26 +129,26 @@ def make_barrier_two():
132129 b3 .leave ()
133130 av .wait ()
134131 ev .wait ()
135- eq_ ( b1 .participating , False )
136- eq_ ( b2 .participating , False )
137- eq_ ( b3 .participating , False )
132+ assert not b1 .participating
133+ assert not b2 .participating
134+ assert not b3 .participating
138135 t1 .join ()
139136 t2 .join ()
140137
141138 def test_barrier_existing_parent_node (self ):
142139 b = self .client .DoubleBarrier ('/some/path' , 1 )
143- self . assertFalse ( b .participating )
140+ assert b .participating is False
144141 self .client .create ('/some' , ephemeral = True )
145142 # the barrier cannot create children under an ephemeral node
146143 b .enter ()
147- self . assertFalse ( b .participating )
144+ assert b .participating is False
148145
149146 def test_barrier_existing_node (self ):
150147 b = self .client .DoubleBarrier ('/some' , 1 )
151- self . assertFalse ( b .participating )
148+ assert b .participating is False
152149 self .client .ensure_path (b .path )
153150 self .client .create (b .create_path , ephemeral = True )
154151 # the barrier will re-use an existing node
155152 b .enter ()
156- self . assertTrue ( b .participating )
153+ assert b .participating is True
157154 b .leave ()
0 commit comments