def test_send_with_test():
"""
When all messages are test=True, send method never called
"""
q = Queue()
worker = SendWorkerThread(
**{
'task_queue': q,
'name': 'test_thread',
'use_sandbox': True,
'cert_file': 'dummy.cert',
'key_file': 'dummy.key'
})
worker.send = mock.Mock()
worker.check_error = mock.Mock()
q.put({
'appid': 'unittest',
'token': dummy_token,
'aps': {
'alert': 'Hello',
'sound': 'default'
},
'test': True
})
worker.main()
eq_(False, worker.send.called)
eq_(False, worker.check_error.called)
def test_send():
"""
send method should called
"""
q = Queue()
worker = SendWorkerThread(
**{
'task_queue': q,
'name': 'test_thread',
'use_sandbox': True,
'cert_file': 'dummy.cert',
'key_file': 'dummy.key'
})
worker.send = mock.Mock()
worker.check_error = mock.Mock()
q.put({
'appid': 'unittest',
'token': dummy_token,
'aps': {
'alert': 'Hello',
'sound': 'default'
}
})
worker.main()
eq_(True, worker.send.called)
eq_(True, worker.check_error.called)
def test_increment_count():
q = Queue()
worker = SendWorkerThread(
**{
'task_queue': q,
'name': 'test_thread',
'use_sandbox': True,
'cert_file': 'dummy.cert',
'key_file': 'dummy.key'
})
worker.count = 99
worker.send = mock.Mock()
worker.check_error = mock.Mock()
q.put({
'appid': 'unittest',
'token': dummy_token,
'aps': {
'alert': 'Hello'
}
})
q.put({
'appid': 'unittest',
'token': dummy_token,
'aps': {
'alert': 'Woooooooooo'
}
})
worker.main()
eq_(101, worker.count, 'worker.count should incremented by messages')
def test_send_with_test():
"""
When all messages are test=True, send method never called
"""
q = Queue()
worker = SendWorkerThread(**{
'task_queue': q,
'name': 'test_thread',
'use_sandbox': True,
'cert_file': 'dummy.cert',
'key_file': 'dummy.key'
})
worker.send = mock.Mock()
worker.check_error = mock.Mock()
q.put({
'appid': 'unittest',
'token': dummy_token,
'aps': {
'alert': 'Hello',
'sound': 'default'
},
'test': True
})
worker.main()
eq_(False, worker.send.called)
eq_(False, worker.check_error.called)
def test_send():
"""
send method should called
"""
q = Queue()
worker = SendWorkerThread(**{
'task_queue': q,
'name': 'test_thread',
'use_sandbox': True,
'cert_file': 'dummy.cert',
'key_file': 'dummy.key'
})
worker.send = mock.Mock()
worker.check_error = mock.Mock()
q.put({
'appid': 'unittest',
'token': dummy_token,
'aps': {
'alert': 'Hello',
'sound': 'default'
}
})
worker.main()
eq_(True, worker.send.called)
eq_(True, worker.check_error.called)
def test_increment_count():
q = Queue()
worker = SendWorkerThread(**{
'task_queue': q,
'name': 'test_thread',
'use_sandbox': True,
'cert_file': 'dummy.cert',
'key_file': 'dummy.key'
})
worker.count = 99
worker.send = mock.Mock()
worker.check_error = mock.Mock()
q.put({
'appid': 'unittest',
'token': dummy_token,
'aps': {'alert': 'Hello'}
})
q.put({
'appid': 'unittest',
'token': dummy_token,
'aps': {'alert': 'Woooooooooo'}
})
worker.main()
eq_(101, worker.count, 'worker.count should incremented by messages')
