def manager(): # mock event handlers manager = EventHandlerManager() manager._event_handler_map.clear() manager._add_event_handler(EventHandlerA1()) manager._add_event_handler(EventHandlerB1()) manager._add_event_handler(EventHandlerA2()) return manager
def test_event_manager_event_mapping_handlers(setup): """Test of event mapping handlers are properly created according to data in db.""" manager = EventHandlerManager() assert manager.get_handlers(0) == [] handlers_1 = manager.get_handlers(1) assert len(handlers_1) == 1 assert handlers_1[0].event_id == 1 assert set(handlers_1[0].message_ids) == set([1, 2]) handlers_2 = manager.get_handlers(2) assert len(handlers_2) == 1 assert handlers_2[0].event_id == 2 assert set(handlers_2[0].message_ids) == set([3])
def test_event_manager_event_mapping_handlers(setup): """Test of event mapping handlers are properly created according to data in db.""" manager = EventHandlerManager() assert manager.get_handlers(0) == [] handlers_1 = manager.get_handlers(1) assert len(handlers_1) == 1 assert handlers_1[0].event_id == 1 assert set(handlers_1[0].message_ids) == set([1, 2]) handlers_2 = manager.get_handlers(2) assert len(handlers_2) == 1 assert handlers_2[0].event_id == 2 assert set(handlers_2[0].message_ids) == set([3])
def init(): context.setup_configuration(CONFIGURATION_FILENAME) database.init_db() context.log_queue = multiprocessing.Queue() context.request_processor = RequestProcessor() context.event_handler_manager = EventHandlerManager()
def test_build_messages_multiple_event_msg(setup): context.event_handler_manager = EventHandlerManager() single_request = EventRequestSingle(1, 1, {}, 1442502890000) request = EventRequestBatch([single_request]) messages = request.build_messages() print messages assert len(messages) == 2 assert set([message['message_id'] for message in messages]) == {1, 2}
def test_build_messages_single_event_msg(setup): context.event_handler_manager = EventHandlerManager() single_request = EventRequestSingle(1, 2, {}, 1442502890000) request = EventRequestBatch([single_request]) messages = request.build_messages() print messages assert len(messages) == 1 assert messages[0]['message_id'] == 3
def test_build_messages_missing_event(setup): context.event_handler_manager = EventHandlerManager() event_proto = EventMessage_pb2.Event() event_proto.user_id = 1 event_proto.event_id = 5 event_proto.event_type = 2 event_proto.timestamp = 1442502890000 request = EventRequestBatch([event_proto]) assert len(request.build_messages()) == 0
def test_login_event_persists_user_data(setup): '''Test that user data is persisted after login event is received''' context.event_handler_manager = EventHandlerManager() event_batch = create_batch_with_login_event( user_id=1338, platform_id=2, device_token='str_device_token') event_request = EventRequestBatch([event_batch]) event_request.process() device_tokens = list(database.get_device_tokens(1338)) assert device_tokens == [(2, 'str_device_token')]
def test_login_event_duplicate(setup): '''Tests that user data is persisted correctly for duplicated login events''' context.event_handler_manager = EventHandlerManager() event_batch = create_batch_with_login_event( user_id=1338, platform_id=1, device_token='str_device_token_1') event_request_platform1 = EventRequestBatch([event_batch]) event_request_platform1.process() event_request_platform1.process() device_tokens = list(database.get_device_tokens(1338)) assert device_tokens == [(1, 'str_device_token_1')]
def test_build_messages_multiple_event_msg(setup): context.event_handler_manager = EventHandlerManager() event_proto = EventMessage_pb2.Event() event_proto.user_id = 1 event_proto.event_id = 1 event_proto.event_type = 2 event_proto.timestamp = 1442502890000 request = EventRequestBatch([event_proto]) messages = request.build_messages() print messages assert len(messages) == 2 assert set([message['message_id'] for message in messages]) == {1, 2}
def test_build_messages_single_event_msg(setup): context.event_handler_manager = EventHandlerManager() event_proto = EventMessage_pb2.Event() event_proto.user_id = 1 event_proto.event_id = 2 event_proto.event_type = 2 event_proto.timestamp = 1442502890000 request = EventRequestBatch([event_proto]) messages = request.build_messages() print messages assert len(messages) == 1 assert messages[0]['message_id'] == 3
def manager(): # mock event handlers manager = EventHandlerManager() manager._event_handler_map.clear() manager._add_event_handler(EventHandlerA1()) manager._add_event_handler(EventHandlerB1()) manager._add_event_handler(EventHandlerA2()) return manager
def test_login_event_same_platform_different_device(setup): '''Tests that both devices are persisted if they have different tokens''' context.event_handler_manager = EventHandlerManager() event_batch_platform1 = create_batch_with_login_event( user_id=1338, platform_id=1, device_token='str_device_token_1') event_request_platform1 = EventRequestBatch([event_batch_platform1]) event_request_platform1.process() event_batch_platform2 = create_batch_with_login_event( user_id=1338, platform_id=1, device_token='str_device_token_2') event_request_platform2 = EventRequestBatch([event_batch_platform2]) event_request_platform2.process() device_tokens = list(database.get_device_tokens(1338)) assert sorted(device_tokens) == [(1, 'str_device_token_1'), (1, 'str_device_token_2')]
def test_login_event_more_platforms(setup): '''Tests that user data is persisted for more platforms''' context.event_handler_manager = EventHandlerManager() event_batch_platform1 = create_batch_with_login_event( user_id=1338, platform_id=1, device_token='str_device_token_1') event_request_platform1 = EventRequestBatch([event_batch_platform1]) event_request_platform1.process() event_batch_platform2 = create_batch_with_login_event( user_id=1338, platform_id=2, device_token='str_device_token_2') event_request_platform2 = EventRequestBatch([event_batch_platform2]) event_request_platform2.process() device_tokens = list(database.get_device_tokens(1338)) assert sorted(device_tokens) == [(1, 'str_device_token_1'), (2, 'str_device_token_2')]
def test_login_and_then_other_event(setup, mocker): '''Tests that both devices are persisted if they have different tokens''' context.event_handler_manager = EventHandlerManager() event_batch_platform1 = create_batch_with_login_event( user_id=1338, platform_id=1, device_id='str_device_id_1', device_token='str_device_token_1') event_request_platform1 = EventRequestBatch([event_batch_platform1]) event_request_platform1.process() # test no message event3 = EventRequestSingle(1, -1, None, 1442502890000) event_request_other3 = EventRequestBatch([event3]) messages3 = event_request_other3.build_messages() assert len(messages3) == 0
def init(): database.init_db() current_revision = database.get_current_revision() head_revision = database.get_head_revision() if head_revision != current_revision: print( "Database is not up to date! " "To upgrade database run pushkin --configuration {cfg} --upgrade-db" .format(cfg=CONFIGURATION_FILENAME)) sys.exit(1) context.log_queue = multiprocessing.Queue() context.request_processor = RequestProcessor() context.event_handler_manager = EventHandlerManager() context.message_blacklist = { row.login_id: set(row.blacklist) for row in database.get_all_message_blacklist() }
def test_login_and_then_other_event(setup, mocker): '''Tests that both devices are persisted if they have different tokens''' context.event_handler_manager = EventHandlerManager() event_batch_platform1 = create_batch_with_login_event( user_id=1338, platform_id=1, device_token='str_device_token_1') event_request_platform1 = EventRequestBatch([event_batch_platform1]) event_request_platform1.process() # test no message event1 = EventRequestSingle(1, -1, None, 1442502890000) event_request_other1 = EventRequestBatch([event1]) messages1 = event_request_other1.build_messages() assert len(messages1) == 0 # test parameter from event params = {'title_param': 'param title', 'text_param': 'param content'} event2 = EventRequestSingle(user_id=1, event_id=3, pairs=params, timestamp=2442502890000) event_request_other2 = EventRequestBatch([event2]) messages2 = event_request_other2.build_messages() assert len(messages2) == 1 assert messages2[0]['message_id'] == 4 assert messages2[0]['content'] == 'text param content' assert messages2[0]['title'] == 'title param title' # test missing parameter which is required by localization event3 = EventRequestSingle(user_id=1, event_id=3, pairs=None, timestamp=1442502890000) event_request_other3 = EventRequestBatch([event3]) messages3 = event_request_other3.build_messages() assert len(messages3) == 0
def test_build_messages_missing_event(setup): context.event_handler_manager = EventHandlerManager() single_request = EventRequestSingle(1, 5, {}, 1442502890000) request = EventRequestBatch([single_request]) assert len(request.build_messages()) == 0