forked from pubnub/python
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathtest_history.py
More file actions
58 lines (47 loc) · 1.93 KB
/
test_history.py
File metadata and controls
58 lines (47 loc) · 1.93 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
from pubnub import Pubnub
import time
import random
from nose.tools import with_setup
pubnub = Pubnub("ds","ds")
pubnub_enc = Pubnub(publish_key="ds",subscribe_key="ds",cipher_key="enigma")
pubnub_pam = Pubnub(publish_key="pam", subscribe_key="pam", secret_key="pam")
def rand(msg):
return "rand-" + str(random.random()) + "-" + msg
channel = rand("channel")
channel_enc = rand("channel_enc")
channel_pam = rand("channel_pam")
messages = []
def setup_func():
for i in range(0,20):
msg = rand("message-" + str(i))
messages.append(msg)
pubnub_pam.grant(channel=channel_pam, read=True, write=True, ttl=144000)
pubnub.publish(channel=channel, message=msg)
pubnub_enc.publish(channel=channel_enc, message=msg)
pubnub_pam.publish(channel=channel_pam, message=msg)
@with_setup(setup_func)
def test_1():
time.sleep(3)
hresp = pubnub.history(channel=channel, count=20)
hresp2 = pubnub_enc.history(channel=channel_enc, count=20)
hresp3 = pubnub_pam.history(channel=channel_pam, count=20)
hresp4 = pubnub_pam.history(channel=channel_pam + "no_rw", count=20)
assert hresp[0] == messages
assert hresp2[0] == messages
assert hresp3[0] == messages
assert hresp4['message'] == 'Forbidden'
assert channel_pam + "no_rw" in hresp4['payload']['channels']
def test_2():
time.sleep(3)
hresp = pubnub.history(channel=channel, count=20, include_token=True)
hresp2 = pubnub_enc.history(channel=channel_enc, count=20, include_token=True)
hresp3 = pubnub_pam.history(channel=channel_pam, count=20, include_token=True)
hresp4 = pubnub_pam.history(channel=channel_pam + "no_rw", count=20, include_token=True)
assert len(hresp[0]) == len(messages)
assert hresp[0][0]['timetoken']
assert len(hresp2[0]) == len(messages)
assert hresp2[0][0]['timetoken']
assert len(hresp3[0]) == len(messages)
assert hresp3[0][0]['timetoken']
assert hresp4['message'] == 'Forbidden'
assert channel_pam + "no_rw" in hresp4['payload']['channels']