privatebin-mirror/js/test/PasteStatus.js

'use strict';
const common = require('../common');
const fc = require('fast-check');

function urlStrings(schema, longUrl, shortUrl) {
    longUrl.schema = schema;
    shortUrl.schema = schema;
    let longUrlString = common.urlToString(longUrl),
        shortUrlString = common.urlToString(shortUrl);
    // ensure the two random URLs actually are sorted as expected
    if (longUrlString.length <= shortUrlString.length) {
        if (longUrlString.length === shortUrlString.length) {
            longUrl.address.unshift('a');
            longUrlString = common.urlToString(longUrl);
        } else {
            [longUrlString, shortUrlString] = [shortUrlString, longUrlString];
        }
    }
    return [longUrlString, shortUrlString];
}

describe('PasteStatus', function () {
    describe('createPasteNotification', function () {
        this.timeout(30000);

        it('creates a notification after a successful document upload', function () {
            cleanup();
            document.body.innerHTML = '<a href="#" id="deletelink"><span></span></a><div id="pastelink"></div><div id="pastesuccess"></div>';
            PrivateBin.PasteStatus.init();
            const expected1 = 'https://example.com/long';
            const expected2 = 'https://example.com/short';
            PrivateBin.PasteStatus.createPasteNotification(expected1, expected2);

            assert.strictEqual(document.getElementById('pasteurl').href, expected1);
            assert.strictEqual(document.getElementById('deletelink').href, expected2);
            assert.ok(!document.getElementById('pastesuccess').classList.contains('hidden'));
        });

        it('creates a notification after a successful document upload (jsc)', () => {
            fc.assert(fc.asyncProperty(
                common.fcUrl(),
                common.fcUrl(false),
                async function (url1, url2) {
                    // sometimes the generator returns incomplete objects, bail out
                    if (!url1 || !url1.address || !url2 || !url2.address) {
                        return true;
                    }
                    const expected1 = common.urlToString(url1).replace(/&(gt|lt)$/, '&$1a'),
                        expected2 = common.urlToString(url2).replace(/&(gt|lt)$/, '&$1a');
                    globalThis.cleanup();
                    document.body.innerHTML = '<a href="#" id="deletelink"><span></span></a><div id="pastelink"></div><div id="pastesuccess"></div>';
                    PrivateBin.PasteStatus.init();
                    PrivateBin.PasteStatus.createPasteNotification(expected1, expected2);

                    assert.ok(!document.getElementById('pastesuccess').classList.contains('hidden'));

                    const result2 = document.getElementById('deletelink').href;
                    return document.getElementById('pasteurl').href === expected1 && result2 === expected2;
                }
            ));
        });
    });

    describe('extractUrl', function () {
        this.timeout(30000);

        it('extracts and updates IDN URLs found in given response', () => {
            fc.assert(fc.property(
                common.fcSchemas(false),
                fc.string({minLength: 1}),
                common.fcUrl(),
                function (schema, domain, url) {
                    domain = domain.replace(/\P{Letter}|[\u{AA}-\u{BA}]/gu, '').toLowerCase();
                    if (domain.length === 0) {
                        domain = 'a';
                    }
                    url.schema = schema;
                    url.address.unshift('.');
                    url.address = domain.split('').concat(url.address);
                    const urlString = common.urlToString(url),
                        expected = urlString.substring((schema + '://' + domain).length),
                        clean = globalThis.cleanup();

                    document.body.innerHTML = '<div><div id="pastelink"></div></div>';
                    PrivateBin.PasteStatus.init();
                    PrivateBin.PasteStatus.createPasteNotification('', '');
                    PrivateBin.PasteStatus.extractUrl(urlString);

                    const result = document.getElementById('pasteurl').href;
                    clean();

                    return result.endsWith(expected) && (
                        result.startsWith(schema + '://xn--') ||
                        result.startsWith(schema + '://' + domain)
                    );
                }
            ));
        });

        // YOURLS API samples from: https://yourls.org/readme.html#API;apireturn
        it('extracts and updates URLs found in YOURLS API JSON response', () => {
            fc.assert(fc.property(
                common.fcSchemas(false),
                common.fcUrl(),
                common.fcUrl(false),
                function (schema, longUrl, shortUrl) {
                    const [longUrlString, shortUrlString] = urlStrings(schema, longUrl, shortUrl),
                        yourlsResponse = {
                            url: {
                                keyword: longUrl.address.join(''),
                                url: longUrlString,
                                title: 'example title',
                                date: '2014-10-24 16:01:39',
                                ip: '127.0.0.1'
                            },
                            status: 'success',
                            message: longUrlString + ' added to database',
                            title: 'example title',
                            shorturl: shortUrlString,
                            statusCode: 200
                        },
                        clean = globalThis.cleanup();

                    document.body.innerHTML = '<div><div id="pastelink"></div></div>';
                    PrivateBin.PasteStatus.init();
                    PrivateBin.PasteStatus.createPasteNotification('', '');
                    PrivateBin.PasteStatus.extractUrl(JSON.stringify(yourlsResponse, undefined, 4));

                    const result = document.getElementById('pasteurl').href;
                    clean();

                    return result === shortUrlString;
                }
            ));
        });
        it('extracts and updates URLs found in YOURLS API XML response', () => {
            fc.assert(fc.property(
                common.fcSchemas(false),
                common.fcUrl(),
                common.fcUrl(false),
                function (schema, longUrl, shortUrl) {
                    const [longUrlString, shortUrlString] = urlStrings(schema, longUrl, shortUrl),
                        yourlsResponse = '<result>\n' +
                            '    <keyword>' + longUrl.address.join('') + '</keyword>\n' +
                            '    <shorturl>' + shortUrlString + '</shorturl>\n' +
                            '    <longurl>' + longUrlString + '</longurl>\n' +
                            '    <message>success</message>\n' +
                            '    <statusCode>200</statusCode>\n' +
                        '</result>',
                        clean = globalThis.cleanup();

                    document.body.innerHTML = '<div><div id="pastelink"></div></div>';
                    PrivateBin.PasteStatus.init();
                    PrivateBin.PasteStatus.createPasteNotification('', '');
                    PrivateBin.PasteStatus.extractUrl(yourlsResponse);

                    const result = document.getElementById('pasteurl').href;
                    clean();

                    return result === shortUrlString;
                }
            ));
        });
        it('extracts and updates URLs found in YOURLS proxy HTML response', () => {
            fc.assert(fc.property(
                common.fcSchemas(false),
                common.fcUrl(),
                common.fcUrl(false),
                function (schema, longUrl, shortUrl) {
                    const [_, shortUrlString] = urlStrings(schema, longUrl, shortUrl),
                        yourlsResponse = '<!DOCTYPE html>\n' +
                            '<html lang="en">\n' +
                            '\t<head>\n' +
                            '\t\t<meta charset="utf-8" />\n' +
                            '\t\t<meta name="robots" content="noindex" />\n' +
                            '\t\t<meta name="google" content="notranslate">\n' +
                            '\t\t<title>PrivateBin</title>\n' +
                            '\t</head>\n' +
                            '\t<body>\n' +
                            '\t\t<p>Your document is <a id="pasteurl" href="' + shortUrlString + '">' + shortUrlString + '</a> <span id="copyhint">(Hit <kbd>Ctrl</kbd>+<kbd>c</kbd> to copy)</span></p>\n' +
                            '\t</body>\n' +
                        '</html>',
                        clean = globalThis.cleanup();

                    document.body.innerHTML = '<div><div id="pastelink"></div></div>';
                    PrivateBin.PasteStatus.init();
                    PrivateBin.PasteStatus.createPasteNotification('', '');
                    PrivateBin.PasteStatus.extractUrl(yourlsResponse);

                    const result = document.getElementById('pasteurl').href;
                    clean();

                    return result === shortUrlString;
                }
            ));
        });
    });

    describe('showRemainingTime', function () {
        this.timeout(30000);

        it('shows burn after reading message or remaining time', () => {
            fc.assert(fc.property(
                fc.boolean(),
                fc.nat(),
                common.fcUrl(),
                function (burnafterreading, remainingTime, url) {
                    let clean = globalThis.cleanup('', {url: common.urlToString(url)}),
                        result;
                    document.body.innerHTML = '<div id="remainingtime" class="hidden"></div>';
                    PrivateBin.Alert.init();
                    PrivateBin.PasteStatus.init();
                    PrivateBin.PasteStatus.showRemainingTime(PrivateBin.Helper.PasteFactory({
                        'adata': [null, null, null, burnafterreading],
                        'v': 2,
                        'meta': {
                            'time_to_live': remainingTime
                        }
                    }));
                    if (burnafterreading) {
                        result = document.getElementById('remainingtime').classList.contains('foryoureyesonly') &&
                                !document.getElementById('remainingtime').classList.contains('hidden');
                    } else if (remainingTime) {
                        result =!document.getElementById('remainingtime').classList.contains('foryoureyesonly') &&
                                !document.getElementById('remainingtime').classList.contains('hidden');
                    } else {
                        result = document.getElementById('remainingtime').classList.contains('hidden') &&
                                !document.getElementById('remainingtime').classList.contains('foryoureyesonly');
                    }
                    clean();
                    return result;
                }
            ));
        });
    });

    describe('hideMessages', function () {
        it(
            'hides all messages',
            function() {
                document.body.innerHTML = (
                    '<div id="remainingtime"></div><div id="pastesuccess"></div>'
                );
                PrivateBin.PasteStatus.init();
                PrivateBin.PasteStatus.hideMessages();
                assert.ok(document.getElementById('remainingtime').classList.contains('hidden'));
                assert.ok(document.getElementById('pastesuccess').classList.contains('hidden'));
            }
        );
    });
});