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')