zulip/frontend_tests/node_tests/notifications.js

// Dependencies
set_global('$', global.make_zjquery({
    silent: true,
}));
set_global('blueslip', global.make_zblueslip());
set_global('document', {
    hasFocus: function () {
        return true;
    },
});
set_global('page_params', {
    is_admin: false,
    realm_users: [],
    enable_desktop_notifications: true,
    enable_sounds: true,
});
const _navigator = {
    userAgent: 'Mozilla/5.0 AppleWebKit/537.36 Chrome/64.0.3282.167 Safari/537.36',
};
set_global('navigator', _navigator);

zrequire('alert_words');
zrequire('muting');
zrequire('stream_data');
zrequire('people');
zrequire('ui');
zrequire('util');

zrequire('notifications');

// Not muted streams
const general = {
    subscribed: true,
    name: 'general',
    stream_id: 10,
    is_muted: false,
};

// Muted streams
const muted = {
    subscribed: true,
    name: 'muted',
    stream_id: 20,
    is_muted: true,
};

stream_data.add_sub('general', general);
stream_data.add_sub('muted', muted);

muting.add_muted_topic(general.stream_id, 'muted topic');

run_test('message_is_notifiable', () => {
    // Case 1: If the message was sent by this user,
    //  DO NOT notify the user
    // In this test, all other circumstances should trigger notification
    // EXCEPT sent_by_me, which should trump them
    let message = {
        id: muted.stream_id,
        content: 'message number 1',
        sent_by_me: true,
        notification_sent: false,
        mentioned: true,
        mentioned_me_directly: true,
        type: 'stream',
        stream: 'general',
        stream_id: general.stream_id,
        topic: 'whatever',
    };
    assert.equal(notifications.should_send_desktop_notification(message), true);
    assert.equal(notifications.should_send_audible_notification(message), true);
    // Not notifiable because it was sent by the current user
    assert.equal(notifications.message_is_notifiable(message), false);

    // Case 2: If the user has already been sent a notificaton about this message,
    //  DO NOT notify the user
    // In this test, all other circumstances should trigger notification
    // EXCEPT notification_sent, which should trump them
    // (ie: it mentions user, it's not muted, etc)
    message = {
        id: general.stream_id,
        content: 'message number 2',
        sent_by_me: false,
        notification_sent: true,
        mentioned: true,
        mentioned_me_directly: true,
        type: 'stream',
        stream: 'general',
        stream_id: general.stream_id,
        topic: 'whatever',
    };
    assert.equal(notifications.should_send_desktop_notification(message), true);
    assert.equal(notifications.should_send_audible_notification(message), true);
    assert.equal(notifications.message_is_notifiable(message), false);

    // Case 3: If a message mentions the user directly,
    //  DO notify the user
    // Mentioning trumps muting
    message = {
        id: 30,
        content: 'message number 3',
        sent_by_me: false,
        notification_sent: false,
        mentioned: true,
        mentioned_me_directly: true,
        type: 'stream',
        stream: 'muted',
        stream_id: muted.stream_id,
        topic: 'topic_three',
    };
    assert.equal(notifications.should_send_desktop_notification(message), true);
    assert.equal(notifications.should_send_audible_notification(message), true);
    assert.equal(notifications.message_is_notifiable(message), true);

    // Case 4:
    // Mentioning should trigger notification in unmuted topic
    message = {
        id: 40,
        content: 'message number 4',
        sent_by_me: false,
        notification_sent: false,
        mentioned: true,
        mentioned_me_directly: true,
        type: 'stream',
        stream: 'general',
        stream_id: general.stream_id,
        topic: 'vanilla',
    };
    assert.equal(notifications.should_send_desktop_notification(message), true);
    assert.equal(notifications.should_send_audible_notification(message), true);
    assert.equal(notifications.message_is_notifiable(message), true);

    // Case 5: If a message is in a muted stream
    //  and does not mention the user DIRECTLY,
    //  DO NOT notify the user
    message = {
        id: 50,
        content: 'message number 5',
        sent_by_me: false,
        notification_sent: false,
        mentioned: true,
        mentioned_me_directly: false,
        type: 'stream',
        stream: 'muted',
        stream_id: muted.stream_id,
        topic: 'whatever',
    };
    assert.equal(notifications.should_send_desktop_notification(message), true);
    assert.equal(notifications.should_send_audible_notification(message), true);
    assert.equal(notifications.message_is_notifiable(message), false);

    // Case 6: If a message is in a muted topic
    //  and does not mention the user DIRECTLY,
    //  DO NOT notify the user
    message = {
        id: 50,
        content: 'message number 6',
        sent_by_me: false,
        notification_sent: false,
        mentioned: true,
        mentioned_me_directly: false,
        type: 'stream',
        stream: 'general',
        stream_id: general.stream_id,
        topic: 'muted topic',
    };
    assert.equal(notifications.should_send_desktop_notification(message), true);
    assert.equal(notifications.should_send_audible_notification(message), true);
    assert.equal(notifications.message_is_notifiable(message), false);

    // Case 7
    // If none of the above cases apply
    // (ie: topic is not muted, message does not mention user,
    //  no notification sent before, message not sent by user),
    // return true to pass it to notifications settings
    message = {
        id: 60,
        content: 'message number 7',
        sent_by_me: false,
        notification_sent: false,
        mentioned: true,
        mentioned_me_directly: false,
        type: 'stream',
        stream: 'general',
        stream_id: general.stream_id,
        topic: 'whatever',
    };
    assert.equal(notifications.should_send_desktop_notification(message), true);
    assert.equal(notifications.should_send_audible_notification(message), true);
    assert.equal(notifications.message_is_notifiable(message), true);
});


run_test('basic_notifications', () => {

    let n; // Object for storing all notification data for assertions.
    let last_closed_message_id = null;
    let last_shown_message_id = null;

    // Notifications API stub
    notifications.set_notification_api({
        createNotification: function createNotification(icon, title, content, tag) {
            const notification_object = {icon: icon, body: content, tag: tag};
            // properties for testing.
            notification_object.tests = {
                shown: false,
            };
            notification_object.show = function () {
                last_shown_message_id = this.tag;
            };
            notification_object.close = function () {
                last_closed_message_id = this.tag;
            };
            notification_object.cancel = function () { notification_object.close(); };
            return notification_object;
        },
    });

    const message_1 = {
        id: 1000,
        content: '@-mentions the user',
        avatar_url: 'url',
        sent_by_me: false,
        sender_full_name: 'Jesse Pinkman',
        notification_sent: false,
        mentioned_me_directly: true,
        type: 'stream',
        stream: 'general',
        stream_id: muted.stream_id,
        topic: 'whatever',
    };

    const message_2 = {
        id: 1500,
        avatar_url: 'url',
        content: '@-mentions the user',
        sent_by_me: false,
        sender_full_name: 'Gus Fring',
        notification_sent: false,
        mentioned_me_directly: true,
        type: 'stream',
        stream: 'general',
        stream_id: muted.stream_id,
        topic: 'lunch',
    };

    // Send notification.
    notifications.process_notification({message: message_1, desktop_notify: true});
    n = notifications.get_notifications();
    assert.equal('Jesse Pinkman to general > whatever' in n, true);
    assert.equal(Object.keys(n).length, 1);
    assert.equal(last_shown_message_id, message_1.id);

    // Remove notification.
    notifications.close_notification(message_1);
    n = notifications.get_notifications();
    assert.equal('Jesse Pinkman to general > whatever' in n, false);
    assert.equal(Object.keys(n).length, 0);
    assert.equal(last_closed_message_id, message_1.id);

    // Send notification.
    message_1.id = 1001;
    notifications.process_notification({message: message_1, desktop_notify: true});
    n = notifications.get_notifications();
    assert.equal('Jesse Pinkman to general > whatever' in n, true);
    assert.equal(Object.keys(n).length, 1);
    assert.equal(last_shown_message_id, message_1.id);

    // Process same message again. Notification count shouldn't increase.
    message_1.id = 1002;
    notifications.process_notification({message: message_1, desktop_notify: true});
    n = notifications.get_notifications();
    assert.equal('Jesse Pinkman to general > whatever' in n, true);
    assert.equal(Object.keys(n).length, 1);
    assert.equal(last_shown_message_id, message_1.id);

    // Send another message. Notification count should increase.
    notifications.process_notification({message: message_2, desktop_notify: true});
    n = notifications.get_notifications();
    assert.equal('Gus Fring to general > lunch' in n, true);
    assert.equal('Jesse Pinkman to general > whatever' in n, true);
    assert.equal(Object.keys(n).length, 2);
    assert.equal(last_shown_message_id, message_2.id);

    // Remove notifications.
    notifications.close_notification(message_1);
    notifications.close_notification(message_2);
    n = notifications.get_notifications();
    assert.equal('Jesse Pinkman to general > whatever' in n, false);
    assert.equal(Object.keys(n).length, 0);
    assert.equal(last_closed_message_id, message_2.id);
});