31 lines
632 B
Python
31 lines
632 B
Python
|
import unittest
|
||
|
|
from unittest import skip
|
||
|
|
|
||
|
|
import jwt
|
||
|
|
|
||
|
|
from music.model.user import User
|
||
|
|
from music.notif.apns import get_token, send_notification, get_secret
|
||
|
|
|
||
|
|
|
||
|
|
class TestAPNS(unittest.TestCase):
|
||
|
|
@skip
|
||
|
|
def test_get_token(self):
|
||
|
|
|
||
|
|
token = get_token()
|
||
|
|
secret = get_secret()
|
||
|
|
|
||
|
|
decoded = jwt.decode(token, secret, algorithms=['ES256'])
|
||
|
|
token
|
||
|
|
|
||
|
|
@skip
|
||
|
|
def test_notification(self):
|
||
|
|
|
||
|
|
for id in User.collection.filter('username', '==', 'andy').get().apns_tokens:
|
||
|
|
notif = send_notification(id, "test")
|
||
|
|
|
||
|
|
print(notif)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|