Skip to content

Commit 1d86b43

Browse files
author
Sebastian Molenda
committed
Enable presence heartbeat in tests
1 parent b227a7f commit 1d86b43

File tree

1 file changed

+21
-16
lines changed

1 file changed

+21
-16
lines changed

tests/integrational/native_threads/test_subscribe.py

Lines changed: 21 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ class TestPubNubSubscription(unittest.TestCase):
4040
filter_query_parameters=['seqn', 'pnsdk', 'tr', 'tt'], serializer='pn_json',
4141
allow_playback_repeats=True)
4242
def test_subscribe_unsubscribe(self):
43-
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True))
43+
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, enable_presence_heartbeat=True))
4444
ch = "test-subscribe-sub-unsub"
4545

4646
try:
@@ -70,7 +70,7 @@ def test_subscribe_unsubscribe(self):
7070

7171
def test_subscribe_pub_unsubscribe(self):
7272
ch = "test-subscribe-pub-unsubscribe"
73-
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True))
73+
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, enable_presence_heartbeat=True))
7474
subscribe_listener = SubscribeListener()
7575
publish_operation = NonSubscribeListener()
7676
message = "hey"
@@ -106,8 +106,8 @@ def test_subscribe_pub_unsubscribe(self):
106106

107107
def test_join_leave(self):
108108
ch = helper.gen_channel("test-subscribe-join-leave")
109-
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True))
110-
pubnub_listener = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True))
109+
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, enable_presence_heartbeat=True))
110+
pubnub_listener = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, enable_presence_heartbeat=True))
111111
callback_messages = SubscribeListener()
112112
callback_presence = SubscribeListener()
113113

@@ -157,7 +157,7 @@ def test_cg_subscribe_unsubscribe(self):
157157
ch = "test-subscribe-unsubscribe-channel"
158158
gr = "test-subscribe-unsubscribe-group"
159159

160-
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True))
160+
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, enable_presence_heartbeat=True))
161161
callback_messages = SubscribeListener()
162162
cg_operation = NonSubscribeListener()
163163

@@ -193,7 +193,7 @@ def test_subscribe_cg_publish_unsubscribe(self):
193193
gr = "test-subscribe-unsubscribe-group"
194194
message = "hey"
195195

196-
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True))
196+
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, enable_presence_heartbeat=True))
197197
callback_messages = SubscribeListener()
198198
non_subscribe_listener = NonSubscribeListener()
199199

@@ -228,8 +228,8 @@ def test_subscribe_cg_publish_unsubscribe(self):
228228
def test_subscribe_cg_join_leave(self):
229229
ch = helper.gen_channel("test-subscribe-unsubscribe-channel")
230230
gr = helper.gen_channel("test-subscribe-unsubscribe-group")
231-
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True))
232-
pubnub_listener = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True))
231+
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, enable_presence_heartbeat=True))
232+
pubnub_listener = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, enable_presence_heartbeat=True))
233233
callback_messages = SubscribeListener()
234234
callback_presence = SubscribeListener()
235235

@@ -285,8 +285,8 @@ def test_subscribe_cg_join_leave(self):
285285
def test_subscribe_pub_unencrypted_unsubscribe(self):
286286
ch = helper.gen_channel("test-subscribe-pub-unencrypted-unsubscribe")
287287

288-
pubnub_plain = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True))
289-
pubnub = PubNub(pnconf_enc_env_copy(enable_subscribe=True, daemon=True))
288+
pubnub_plain = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, enable_presence_heartbeat=True))
289+
pubnub = PubNub(pnconf_enc_env_copy(enable_subscribe=True, daemon=True, enable_presence_heartbeat=True))
290290

291291
subscribe_listener = SubscribeListener()
292292
publish_operation = NonSubscribeListener()
@@ -335,7 +335,7 @@ def test_subscribe_pub_unencrypted_unsubscribe(self):
335335
def test_subscribe_retry_policy_none(self):
336336
ch = "test-subscribe-retry-policy-none"
337337
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, origin='127.0.0.1',
338-
reconnect_policy=PNReconnectionPolicy.NONE))
338+
reconnect_policy=PNReconnectionPolicy.NONE, enable_presence_heartbeat=True))
339339
listener = DisconnectListener()
340340

341341
try:
@@ -356,7 +356,8 @@ def mock_calculate(*args, **kwargs):
356356
with patch('pubnub.managers.LinearDelay.calculate', wraps=mock_calculate) as calculate_mock:
357357
ch = "test-subscribe-retry-policy-linear"
358358
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, origin='127.0.0.1',
359-
reconnect_policy=PNReconnectionPolicy.LINEAR))
359+
reconnect_policy=PNReconnectionPolicy.LINEAR,
360+
enable_presence_heartbeat=True))
360361
listener = DisconnectListener()
361362

362363
try:
@@ -379,7 +380,8 @@ def mock_calculate(*args, **kwargs):
379380
with patch('pubnub.managers.ExponentialDelay.calculate', wraps=mock_calculate) as calculate_mock:
380381
ch = "test-subscribe-retry-policy-exponential"
381382
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, origin='127.0.0.1',
382-
reconnect_policy=PNReconnectionPolicy.EXPONENTIAL))
383+
reconnect_policy=PNReconnectionPolicy.EXPONENTIAL,
384+
enable_presence_heartbeat=True))
383385
listener = DisconnectListener()
384386

385387
try:
@@ -403,7 +405,8 @@ def mock_calculate(*args, **kwargs):
403405
ch = "test-subscribe-retry-policy-linear"
404406
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, origin='127.0.0.1',
405407
maximum_reconnection_retries=3,
406-
reconnect_policy=PNReconnectionPolicy.LINEAR))
408+
reconnect_policy=PNReconnectionPolicy.LINEAR,
409+
enable_presence_heartbeat=True))
407410
listener = DisconnectListener()
408411

409412
try:
@@ -427,7 +430,8 @@ def mock_calculate(*args, **kwargs):
427430
ch = "test-subscribe-retry-policy-exponential"
428431
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, origin='127.0.0.1',
429432
maximum_reconnection_retries=3,
430-
reconnect_policy=PNReconnectionPolicy.EXPONENTIAL))
433+
reconnect_policy=PNReconnectionPolicy.EXPONENTIAL,
434+
enable_presence_heartbeat=True))
431435
listener = DisconnectListener()
432436

433437
try:
@@ -451,7 +455,8 @@ def mock_calculate(*args, **kwargs):
451455
ch = "test-subscribe-retry-policy-linear"
452456
pubnub = PubNub(pnconf_env_copy(enable_subscribe=True, daemon=True, origin='127.0.0.1',
453457
maximum_reconnection_retries=3, reconnection_interval=1,
454-
reconnect_policy=PNReconnectionPolicy.LINEAR))
458+
reconnect_policy=PNReconnectionPolicy.LINEAR,
459+
enable_presence_heartbeat=True))
455460
listener = DisconnectListener()
456461

457462
try:

0 commit comments

Comments
 (0)