@@ -85,7 +85,7 @@ def test_subscription():
8585
8686def test_ack ():
8787 policy = create_policy ()
88- policy ._consumer .stopped .set ()
88+ policy ._consumer .stopped .clear ()
8989 with mock .patch .object (policy ._consumer , 'send_request' ) as send_request :
9090 policy .ack ('ack_id_string' , 20 )
9191 send_request .assert_called_once_with (types .StreamingPullRequest (
@@ -97,7 +97,7 @@ def test_ack():
9797
9898def test_ack_no_time ():
9999 policy = create_policy ()
100- policy ._consumer .stopped .set ()
100+ policy ._consumer .stopped .clear ()
101101 with mock .patch .object (policy ._consumer , 'send_request' ) as send_request :
102102 policy .ack ('ack_id_string' )
103103 send_request .assert_called_once_with (types .StreamingPullRequest (
@@ -110,6 +110,7 @@ def test_ack_paused():
110110 policy = create_policy ()
111111 policy ._paused = True
112112 consumer = policy ._consumer
113+ consumer .stopped .set ()
113114
114115 with mock .patch .object (consumer , 'resume' ) as resume :
115116 policy .ack ('ack_id_string' )
@@ -206,20 +207,21 @@ def test_modify_ack_deadline():
206207
207208def test_maintain_leases_inactive_consumer ():
208209 policy = create_policy ()
209- policy ._consumer .stopped .clear ()
210+ policy ._consumer .stopped .set ()
210211 assert policy .maintain_leases () is None
211212
212213
213214def test_maintain_leases_ack_ids ():
214215 policy = create_policy ()
215- policy ._consumer .stopped .set ()
216+ policy ._consumer .stopped .clear ()
216217 policy .lease ('my ack id' , 50 )
217218
218219 # Mock the sleep object.
219220 with mock .patch .object (time , 'sleep' , autospec = True ) as sleep :
220221 def trigger_inactive (seconds ):
221222 assert 0 < seconds < 10
222- policy ._consumer .stopped .clear ()
223+ policy ._consumer .stopped .set ()
224+
223225 sleep .side_effect = trigger_inactive
224226
225227 # Also mock the consumer, which sends the request.
@@ -234,11 +236,12 @@ def trigger_inactive(seconds):
234236
235237def test_maintain_leases_no_ack_ids ():
236238 policy = create_policy ()
237- policy ._consumer .stopped .set ()
239+ policy ._consumer .stopped .clear ()
238240 with mock .patch .object (time , 'sleep' , autospec = True ) as sleep :
239241 def trigger_inactive (seconds ):
240242 assert 0 < seconds < 10
241- policy ._consumer .stopped .clear ()
243+ policy ._consumer .stopped .set ()
244+
242245 sleep .side_effect = trigger_inactive
243246 policy .maintain_leases ()
244247 sleep .assert_called ()
0 commit comments