def test_add_frame_item():
global has_called_check_method
has_called_check_method = False
# assert_called_once_with of Mock Class failes to compare Payload object.
# So here created custom check method for parameter checking
def check_add_item_args(token, payload, identifier, expiry, priority):
default_expiry = int(time.time()) + (60 * 60)
eq_(token, dummy_token)
eq_(payload.alert, 'Hey')
eq_(payload.badge, None)
eq_(payload.sound, None)
eq_(payload.custom, {})
eq_(payload.content_available, False)
eq_(identifier, 123)
eq_(priority, 10, 'priority should set default value when skipped')
eq_(expiry, default_expiry,
'expiry should set default value when skipped')
global has_called_check_method
has_called_check_method = True
worker = SendWorkerThread(**dummy_setting)
worker.count = 123
frame = Frame()
frame.add_item = check_add_item_args
worker.add_frame_item(frame, 'myapp', dummy_token, {'alert': 'Hey'})
ok_(has_called_check_method)
def test_add_frame_item():
global has_called_check_method
has_called_check_method = False
# assert_called_once_with of Mock Class failes to compare Payload object.
# So here created custom check method for parameter checking
def check_add_item_args(token, payload, identifier, expiry, priority):
default_expiry = int(time.time()) + (60 * 60)
eq_(token, dummy_token)
eq_(payload.alert, 'Hey')
eq_(payload.badge, None)
eq_(payload.sound, None)
eq_(payload.custom, {})
eq_(payload.content_available, False)
eq_(identifier, 123)
eq_(priority, 10, 'priority should set default value when skipped')
eq_(expiry, default_expiry, 'expiry should set default value when skipped')
global has_called_check_method
has_called_check_method = True
worker = SendWorkerThread(**dummy_setting)
worker.count = 123
frame = Frame()
frame.add_item = check_add_item_args
worker.add_frame_item(frame, 'myapp', dummy_token, {'alert': 'Hey'})
ok_(has_called_check_method)
def test_add_frame_item_with_full_args():
global has_called_check_method
has_called_check_method = False
# assert_called_once_with of Mock Class failes to compare Payload object.
# So here created custom check method for parameter checking
def check_add_item_args(token, payload, identifier, expiry, priority):
eq_(token, dummy_token)
eq_(payload.alert, 'Wow')
eq_(payload.badge, 50)
eq_(payload.sound, 'bell')
eq_(payload.custom, {'foo': 'bar'})
eq_(payload.content_available, True)
eq_(identifier, 1)
eq_(priority, 5)
eq_(expiry, 0)
global has_called_check_method
has_called_check_method = True
worker = SendWorkerThread(**dummy_setting)
worker.count = 1
frame = Frame()
frame.add_item = check_add_item_args
worker.add_frame_item(frame, 'myapp', dummy_token, {
'alert': 'Wow',
'badge': 50,
'sound': 'bell',
'custom': {'foo': 'bar'},
'content_available': True,
}, expiry=0, priority=5)
ok_(has_called_check_method)
def test_frame_content():
worker = SendWorkerThread(**dummy_setting)
frame = Frame()
ok_(len(frame.frame_data) == 0)
worker.add_frame_item(frame, 'myapp', dummy_token, {'alert': 'hello'}, expiry=None, priority=10)
hex_frame = b2a_hex(str(frame.frame_data))
# Notification command
eq_('02', hex_frame[0:2])
# Frame length
eq_('00000051', hex_frame[2:10])
# Item ID:1 Device Token
eq_('01', hex_frame[10:12])
# Token Length
eq_('0020', hex_frame[12:16])
# Token
eq_(dummy_token, hex_frame[16:80])
def test_frame_content():
worker = SendWorkerThread(**dummy_setting)
frame = Frame()
ok_(len(frame.frame_data) == 0)
worker.add_frame_item(frame,
'myapp',
dummy_token, {'alert': 'hello'},
expiry=None,
priority=10)
hex_frame = b2a_hex(str(frame.frame_data))
# Notification command
eq_('02', hex_frame[0:2])
# Frame length
eq_('00000051', hex_frame[2:10])
# Item ID:1 Device Token
eq_('01', hex_frame[10:12])
# Token Length
eq_('0020', hex_frame[12:16])
# Token
eq_(dummy_token, hex_frame[16:80])
def test_add_frame_item_with_full_args():
global has_called_check_method
has_called_check_method = False
# assert_called_once_with of Mock Class failes to compare Payload object.
# So here created custom check method for parameter checking
def check_add_item_args(token, payload, identifier, expiry, priority):
eq_(token, dummy_token)
eq_(payload.alert, 'Wow')
eq_(payload.badge, 50)
eq_(payload.sound, 'bell')
eq_(payload.custom, {'foo': 'bar'})
eq_(payload.content_available, True)
eq_(identifier, 1)
eq_(priority, 5)
eq_(expiry, 0)
global has_called_check_method
has_called_check_method = True
worker = SendWorkerThread(**dummy_setting)
worker.count = 1
frame = Frame()
frame.add_item = check_add_item_args
worker.add_frame_item(frame,
'myapp',
dummy_token, {
'alert': 'Wow',
'badge': 50,
'sound': 'bell',
'custom': {
'foo': 'bar'
},
'content_available': True,
},
expiry=0,
priority=5)
ok_(has_called_check_method)
