def test_notification_proto_without_login_id(mock_log):
'''Test that a notification proto without login_id fails validation'''
notification = NotificationRequestSingle(None, "Msg title",
"Text of a message.",
"some_screen_id")
assert not NotificationRequestBatch([notification
]).validate_single(notification)
def test_valid_notification_proto(mock_log):
'''Test that a valid notification proto is validated correctly.'''
notification = NotificationRequestSingle(1338, "Msg title",
"Text of a message.",
"some_screen_id")
assert NotificationRequestBatch([notification
]).validate_single(notification)
def create_request(self, requests):
valid_requests = []
validator = ProtoNotificationValidator()
for request in requests:
if validator.validate_single(request):
valid_requests.append(
NotificationRequestSingle(request.login_id, request.title,
request.content))
return NotificationRequestBatch(valid_requests)
def test_notification_proto_empty_screen_process(mocker, mock_log):
'''Test that a valid notification proto without screen can be processed.'''
mocker.patch('pushkin.database.database.get_raw_messages')
notification = NotificationRequestSingle(1338, "Msg title",
"Text of a message.")
NotificationRequestBatch([notification]).process_single(notification)
database.get_raw_messages.assert_called_with(1338, "Msg title",
"Text of a message.", "",
config.game, config.world_id,
config.dry_run)
def create_request(self, requests):
valid_requests = []
validator = JsonNotificationValidator()
for request in requests:
if validator.validate_single(request):
valid_requests.append(
NotificationRequestSingle(request['login_id'],
request['title'],
request['content']))
return NotificationRequestBatch(valid_requests)
def test_notification_proto_empty_content(mock_log):
'''Test that a notification proto with empty content fails validation'''
notification = NotificationRequestSingle(1338, "Msg title", "",
"some_screen_id")
assert not NotificationRequestBatch([notification
]).validate_single(notification)
def test_notification_proto_empty_title(mock_log):
'''Test that a notification proto with empty title fails validation'''
notification = NotificationRequestSingle(1338, "", "Text of a message.",
"some_screen_id")
assert not NotificationRequestBatch([notification
]).validate_single(notification)