mirror of
				https://github.com/zulip/zulip.git
				synced 2025-11-04 05:53:43 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			489 lines
		
	
	
		
			17 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			489 lines
		
	
	
		
			17 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
"use strict";
 | 
						|
 | 
						|
const {strict: assert} = require("assert");
 | 
						|
const path = require("path");
 | 
						|
 | 
						|
require("css.escape");
 | 
						|
const puppeteer = require("puppeteer");
 | 
						|
 | 
						|
const {test_credentials} = require("../../var/puppeteer/test_credentials");
 | 
						|
 | 
						|
const root_dir = path.resolve(__dirname, "../../");
 | 
						|
const puppeteer_dir = path.join(root_dir, "var/puppeteer");
 | 
						|
 | 
						|
class CommonUtils {
 | 
						|
    constructor() {
 | 
						|
        this.browser = null;
 | 
						|
        this.screenshot_id = 0;
 | 
						|
        this.realm_url = "http://zulip.zulipdev.com:9981/";
 | 
						|
        this.pm_recipient = {
 | 
						|
            async set(page, recipient) {
 | 
						|
                // Without using the delay option here there seems to be
 | 
						|
                // a flake where the typeahead doesn't show up.
 | 
						|
                await page.type("#private_message_recipient", recipient, {delay: 100});
 | 
						|
 | 
						|
                // We use jQuery here because we need to use it's :visible
 | 
						|
                // pseudo selector to actually wait for typeahead item that
 | 
						|
                // is visible; there can be typeahead item with this selector
 | 
						|
                // that is invisible because it is meant for something else
 | 
						|
                // e.g. private message input typeahead is different from topic
 | 
						|
                // input typeahead but both can be present in the dom.
 | 
						|
                await page.waitForFunction(() => {
 | 
						|
                    const selector = ".typeahead-menu .active a:visible";
 | 
						|
                    return $(selector).length !== 0;
 | 
						|
                });
 | 
						|
 | 
						|
                await page.evaluate(() => {
 | 
						|
                    $(".typeahead-menu .active a:visible").click();
 | 
						|
                });
 | 
						|
            },
 | 
						|
 | 
						|
            async expect(page, expected) {
 | 
						|
                const actual_recipients = await page.evaluate(() =>
 | 
						|
                    compose_state.private_message_recipient(),
 | 
						|
                );
 | 
						|
                assert.equal(actual_recipients, expected);
 | 
						|
            },
 | 
						|
        };
 | 
						|
 | 
						|
        this.fullname = {
 | 
						|
            cordelia: "Cordelia Lear",
 | 
						|
            othello: "Othello, the Moor of Venice",
 | 
						|
            hamlet: "King Hamlet",
 | 
						|
        };
 | 
						|
 | 
						|
        this.window_size = {
 | 
						|
            width: 1400,
 | 
						|
            height: 1024,
 | 
						|
        };
 | 
						|
    }
 | 
						|
 | 
						|
    async ensure_browser() {
 | 
						|
        if (this.browser === null) {
 | 
						|
            const {window_size} = this;
 | 
						|
            this.browser = await puppeteer.launch({
 | 
						|
                args: [
 | 
						|
                    `--window-size=${window_size.width},${window_size.height}`,
 | 
						|
                    "--no-sandbox",
 | 
						|
                    "--disable-setuid-sandbox",
 | 
						|
                ],
 | 
						|
                defaultViewport: {width: 1280, height: 1024},
 | 
						|
                headless: true,
 | 
						|
            });
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    async get_page() {
 | 
						|
        await this.ensure_browser();
 | 
						|
        const page = await this.browser.newPage();
 | 
						|
        return page;
 | 
						|
    }
 | 
						|
 | 
						|
    async screenshot(page, name = null) {
 | 
						|
        if (name === null) {
 | 
						|
            name = `${this.screenshot_id}`;
 | 
						|
            this.screenshot_id += 1;
 | 
						|
        }
 | 
						|
 | 
						|
        const screenshot_path = path.join(puppeteer_dir, `${name}.png`);
 | 
						|
        await page.screenshot({
 | 
						|
            path: screenshot_path,
 | 
						|
        });
 | 
						|
    }
 | 
						|
 | 
						|
    async assert_selector_doesnt_exist(page, selector) {
 | 
						|
        await page.waitForFunction((selector) => $(selector === null), {}, selector);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * This function takes a params object whose fields
 | 
						|
     * are referenced by name attribute of an input field and
 | 
						|
     * the input as a key.
 | 
						|
     *
 | 
						|
     * For example to fill:
 | 
						|
     *  <form id="#demo">
 | 
						|
     *     <input type="text" name="username">
 | 
						|
     *     <input type="checkbox" name="terms">
 | 
						|
     *  </form>
 | 
						|
     *
 | 
						|
     * You can call:
 | 
						|
     * common.fill_form(page, '#demo', {
 | 
						|
     *     username: 'Iago',
 | 
						|
     *     terms: true
 | 
						|
     * });
 | 
						|
     */
 | 
						|
    async fill_form(page, form_selector, params) {
 | 
						|
        async function is_dropdown(page, name) {
 | 
						|
            return (await page.$(`select[name="${name}"]`)) !== null;
 | 
						|
        }
 | 
						|
        for (const name of Object.keys(params)) {
 | 
						|
            const name_selector = `${form_selector} [name="${name}"]`;
 | 
						|
            const value = params[name];
 | 
						|
            if (typeof value === "boolean") {
 | 
						|
                await page.$eval(name_selector, (el, value) => {
 | 
						|
                    if (el.checked !== value) {
 | 
						|
                        el.click();
 | 
						|
                    }
 | 
						|
                });
 | 
						|
            } else if (await is_dropdown(page, name)) {
 | 
						|
                await page.select(name_selector, params[name]);
 | 
						|
            } else {
 | 
						|
                // clear any existing text in the input field before filling.
 | 
						|
                await page.$eval(name_selector, (el) => {
 | 
						|
                    el.value = "";
 | 
						|
                });
 | 
						|
                await page.type(name_selector, params[name]);
 | 
						|
            }
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    async check_form_contents(page, form_selector, params) {
 | 
						|
        for (const name of Object.keys(params)) {
 | 
						|
            const name_selector = `${form_selector} [name="${name}"]`;
 | 
						|
            const expected_value = params[name];
 | 
						|
            if (typeof expected_value === "boolean") {
 | 
						|
                assert.equal(
 | 
						|
                    await page.$eval(name_selector, (el) => el.checked),
 | 
						|
                    expected_value,
 | 
						|
                    "Form content is not as expected.",
 | 
						|
                );
 | 
						|
            } else {
 | 
						|
                assert.equal(
 | 
						|
                    await page.$eval(name_selector, (el) => el.value),
 | 
						|
                    expected_value,
 | 
						|
                    "Form content is not as expected.",
 | 
						|
                );
 | 
						|
            }
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    async get_text_from_selector(page, selector) {
 | 
						|
        return await page.evaluate((selector) => $(selector).text().trim(), selector);
 | 
						|
    }
 | 
						|
 | 
						|
    async wait_for_text(page, selector, text) {
 | 
						|
        await page.waitForFunction(
 | 
						|
            (selector, text) => $(selector).text().includes(text),
 | 
						|
            {},
 | 
						|
            selector,
 | 
						|
            text,
 | 
						|
        );
 | 
						|
    }
 | 
						|
 | 
						|
    async get_stream_id(page, stream_name) {
 | 
						|
        return await page.evaluate(
 | 
						|
            (stream_name) => stream_data.get_stream_id(stream_name),
 | 
						|
            stream_name,
 | 
						|
        );
 | 
						|
    }
 | 
						|
 | 
						|
    async get_user_id_from_name(page, name) {
 | 
						|
        if (this.fullname[name] !== undefined) {
 | 
						|
            name = this.fullname[name];
 | 
						|
        }
 | 
						|
        return await page.evaluate((name) => {
 | 
						|
            const people = require("./static/js/people");
 | 
						|
            return people.get_user_id_from_name(name);
 | 
						|
        }, name);
 | 
						|
    }
 | 
						|
 | 
						|
    async get_internal_email_from_name(page, name) {
 | 
						|
        if (this.fullname[name] !== undefined) {
 | 
						|
            name = this.fullname[name];
 | 
						|
        }
 | 
						|
        return await page.evaluate((fullname) => {
 | 
						|
            const people = require("./static/js/people");
 | 
						|
            const user_id = people.get_user_id_from_name(fullname);
 | 
						|
            return people.get_by_user_id(user_id).email;
 | 
						|
        }, name);
 | 
						|
    }
 | 
						|
 | 
						|
    async log_in(page, credentials = null) {
 | 
						|
        console.log("Logging in");
 | 
						|
        await page.goto(this.realm_url + "login/");
 | 
						|
        assert.equal(this.realm_url + "login/", page.url());
 | 
						|
        if (credentials === null) {
 | 
						|
            credentials = test_credentials.default_user;
 | 
						|
        }
 | 
						|
        // fill login form
 | 
						|
        const params = {
 | 
						|
            username: credentials.username,
 | 
						|
            password: credentials.password,
 | 
						|
        };
 | 
						|
        await this.fill_form(page, "#login_form", params);
 | 
						|
        await page.$eval("#login_form", (form) => form.submit());
 | 
						|
 | 
						|
        await page.waitForSelector("#zhome .message_row", {visible: true});
 | 
						|
    }
 | 
						|
 | 
						|
    async log_out(page) {
 | 
						|
        await page.goto(this.realm_url);
 | 
						|
        const menu_selector = "#settings-dropdown";
 | 
						|
        const logout_selector = 'a[href="#logout"]';
 | 
						|
        console.log("Logging out");
 | 
						|
        await page.waitForSelector(menu_selector, {visible: true});
 | 
						|
        await page.click(menu_selector);
 | 
						|
        await page.waitForSelector(logout_selector);
 | 
						|
        await page.click(logout_selector);
 | 
						|
 | 
						|
        // Wait for a email input in login page so we know login
 | 
						|
        // page is loaded. Then check that we are at the login url.
 | 
						|
        await page.waitForSelector('input[name="username"]');
 | 
						|
        assert(page.url().includes("/login/"));
 | 
						|
    }
 | 
						|
 | 
						|
    async ensure_enter_does_not_send(page) {
 | 
						|
        await page.$eval("#enter_sends", (el) => {
 | 
						|
            if (el.checked) {
 | 
						|
                el.click();
 | 
						|
            }
 | 
						|
        });
 | 
						|
    }
 | 
						|
 | 
						|
    async assert_compose_box_content(page, expected_value) {
 | 
						|
        await page.waitForSelector("#compose-textarea");
 | 
						|
 | 
						|
        const compose_box_element = await page.$("#compose-textarea");
 | 
						|
        const compose_box_content = await page.evaluate(
 | 
						|
            (element) => element.value,
 | 
						|
            compose_box_element,
 | 
						|
        );
 | 
						|
        assert.equal(
 | 
						|
            compose_box_content,
 | 
						|
            expected_value,
 | 
						|
            `Compose box content did not match with the expected value '{${expected_value}}'`,
 | 
						|
        );
 | 
						|
    }
 | 
						|
 | 
						|
    async wait_for_fully_processed_message(page, content) {
 | 
						|
        await page.waitForFunction(
 | 
						|
            (content) => {
 | 
						|
                /*
 | 
						|
                The tricky part about making sure that
 | 
						|
                a message has actually been fully processed
 | 
						|
                is that we'll "locally echo" the message
 | 
						|
                first on the client.  Until the server
 | 
						|
                actually acks the message, the message will
 | 
						|
                have a temporary id and will not have all
 | 
						|
                the normal message controls.
 | 
						|
                For the Puppeteer tests, we want to avoid all
 | 
						|
                the edge cases with locally echoed messages.
 | 
						|
                In order to make sure a message is processed,
 | 
						|
                we use internals to determine the following:
 | 
						|
                    - has message_list even been updated with
 | 
						|
                      the message with out content?
 | 
						|
                    - has the locally_echoed flag been cleared?
 | 
						|
                But for the final steps we look at the
 | 
						|
                actual DOM (via JQuery):
 | 
						|
                    - is it visible?
 | 
						|
                    - does it look to have been
 | 
						|
                      re-rendered based on server info?
 | 
						|
            */
 | 
						|
                const last_msg = current_msg_list.last();
 | 
						|
                if (last_msg === undefined) {
 | 
						|
                    return false;
 | 
						|
                }
 | 
						|
 | 
						|
                if (last_msg.raw_content !== content) {
 | 
						|
                    return false;
 | 
						|
                }
 | 
						|
 | 
						|
                if (last_msg.locally_echoed) {
 | 
						|
                    return false;
 | 
						|
                }
 | 
						|
 | 
						|
                const row = rows.last_visible();
 | 
						|
                if (rows.id(row) !== last_msg.id) {
 | 
						|
                    return false;
 | 
						|
                }
 | 
						|
 | 
						|
                /*
 | 
						|
                Make sure the message is completely
 | 
						|
                re-rendered from its original "local echo"
 | 
						|
                version by looking for the star icon.  We
 | 
						|
                don't add the star icon until the server
 | 
						|
                responds.
 | 
						|
            */
 | 
						|
                return row.find(".star").length === 1;
 | 
						|
            },
 | 
						|
            {},
 | 
						|
            content,
 | 
						|
        );
 | 
						|
    }
 | 
						|
 | 
						|
    // Wait for any previous send to finish, then send a message.
 | 
						|
    async send_message(page, type, params) {
 | 
						|
        // If a message is outside the view, we do not need
 | 
						|
        // to wait for it to be processed later.
 | 
						|
        const outside_view = params.outside_view;
 | 
						|
        delete params.outside_view;
 | 
						|
 | 
						|
        // Compose box content should be empty before sending the message.
 | 
						|
        await this.assert_compose_box_content(page, "");
 | 
						|
 | 
						|
        if (type === "stream") {
 | 
						|
            await page.keyboard.press("KeyC");
 | 
						|
        } else if (type === "private") {
 | 
						|
            await page.keyboard.press("KeyX");
 | 
						|
            const recipients = params.recipient.split(", ");
 | 
						|
            for (const recipient of recipients) {
 | 
						|
                await this.pm_recipient.set(page, recipient);
 | 
						|
            }
 | 
						|
            delete params.recipient;
 | 
						|
        } else {
 | 
						|
            assert.fail("`send_message` got invalid message type");
 | 
						|
        }
 | 
						|
 | 
						|
        if (params.stream) {
 | 
						|
            params.stream_message_recipient_stream = params.stream;
 | 
						|
            delete params.stream;
 | 
						|
        }
 | 
						|
 | 
						|
        if (params.topic) {
 | 
						|
            params.stream_message_recipient_topic = params.topic;
 | 
						|
            delete params.topic;
 | 
						|
        }
 | 
						|
 | 
						|
        await this.fill_form(page, 'form[action^="/json/messages"]', params);
 | 
						|
        await this.assert_compose_box_content(page, params.content);
 | 
						|
        await this.ensure_enter_does_not_send(page);
 | 
						|
        await page.waitForSelector("#compose-send-button", {visible: true});
 | 
						|
        await page.click("#compose-send-button");
 | 
						|
 | 
						|
        // Sending should clear compose box content.
 | 
						|
        this.assert_compose_box_content(page, "");
 | 
						|
 | 
						|
        if (!outside_view) {
 | 
						|
            await this.wait_for_fully_processed_message(page, params.content);
 | 
						|
        }
 | 
						|
 | 
						|
        // Close the compose box after sending the message.
 | 
						|
        await page.evaluate(() => {
 | 
						|
            compose_actions.cancel();
 | 
						|
        });
 | 
						|
        // Make sure the compose box is closed.
 | 
						|
        await page.waitForSelector("#compose-textarea", {hidden: true});
 | 
						|
    }
 | 
						|
 | 
						|
    async send_multiple_messages(page, msgs) {
 | 
						|
        for (const msg of msgs) {
 | 
						|
            await this.send_message(page, msg.stream !== undefined ? "stream" : "private", msg);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * This method returns a array, which is formmated as:
 | 
						|
     *  [
 | 
						|
     *    ['stream > topic', ['message 1', 'message 2']],
 | 
						|
     *    ['You and Cordelia Lear', ['message 1', 'message 2']]
 | 
						|
     *  ]
 | 
						|
     *
 | 
						|
     * The messages are sorted chronologically.
 | 
						|
     */
 | 
						|
    async get_rendered_messages(page, table = "zhome") {
 | 
						|
        return await page.evaluate((table) => {
 | 
						|
            const $recipient_rows = $(`#${CSS.escape(table)}`).find(".recipient_row");
 | 
						|
            return $recipient_rows.toArray().map((element) => {
 | 
						|
                const $el = $(element);
 | 
						|
                const stream_name = $el.find(".stream_label").text().trim();
 | 
						|
                const topic_name = $el.find(".stream_topic a").text().trim();
 | 
						|
 | 
						|
                let key = stream_name;
 | 
						|
                if (topic_name !== "") {
 | 
						|
                    // If topic_name is '' then this is PMs, so only
 | 
						|
                    // append > topic_name if we are not in PMs or Group PMs.
 | 
						|
                    key = `${stream_name} > ${topic_name}`;
 | 
						|
                }
 | 
						|
 | 
						|
                const messages = $el
 | 
						|
                    .find(".message_row .message_content")
 | 
						|
                    .toArray()
 | 
						|
                    .map((message_row) => message_row.textContent.trim());
 | 
						|
 | 
						|
                return [key, messages];
 | 
						|
            });
 | 
						|
        }, table);
 | 
						|
    }
 | 
						|
 | 
						|
    // This method takes in page, table to fetch the messages
 | 
						|
    // from, and expected messages. The format of expected
 | 
						|
    // message is { "stream > topic": [messages] }.
 | 
						|
    // The method will only check that all the messages in the
 | 
						|
    // messages array passed exist in the order they are passed.
 | 
						|
    async check_messages_sent(page, table, messages) {
 | 
						|
        await page.waitForSelector(`#${CSS.escape(table)}`, {visible: true});
 | 
						|
        const rendered_messages = await this.get_rendered_messages(page, table);
 | 
						|
 | 
						|
        // We only check the last n messages because if we run
 | 
						|
        // the test with --interactive there will be duplicates.
 | 
						|
        const last_n_messages = rendered_messages.slice(-messages.length);
 | 
						|
        assert.deepStrictEqual(last_n_messages, messages);
 | 
						|
    }
 | 
						|
 | 
						|
    async manage_organization(page) {
 | 
						|
        const menu_selector = "#settings-dropdown";
 | 
						|
        await page.waitForSelector(menu_selector, {visible: true});
 | 
						|
        await page.click(menu_selector);
 | 
						|
 | 
						|
        const organization_settings = 'a[href="#organization"]';
 | 
						|
        await page.click(organization_settings);
 | 
						|
        await page.waitForSelector("#settings_overlay_container.show", {visible: true});
 | 
						|
 | 
						|
        const url = page.url();
 | 
						|
        assert(/^http:\/\/[^/]+\/#organization/.test(url), "Unexpected manage organization URL");
 | 
						|
 | 
						|
        const organization_settings_data_section = "li[data-section='organization-settings']";
 | 
						|
        await page.click(organization_settings_data_section);
 | 
						|
    }
 | 
						|
 | 
						|
    async select_item_via_typeahead(page, field_selector, str, item) {
 | 
						|
        console.log(`Looking in ${field_selector} to select ${str}, ${item}`);
 | 
						|
        await page.evaluate(
 | 
						|
            (field_selector, str, item) => {
 | 
						|
                // Set the value and then send a bogus keyup event to trigger
 | 
						|
                // the typeahead.
 | 
						|
                $(field_selector)
 | 
						|
                    .trigger("focus")
 | 
						|
                    .val(str)
 | 
						|
                    .trigger(new $.Event("keyup", {which: 0}));
 | 
						|
 | 
						|
                // Trigger the typeahead.
 | 
						|
                // Reaching into the guts of Bootstrap Typeahead like this is not
 | 
						|
                // great, but I found it very hard to do it any other way.
 | 
						|
 | 
						|
                const tah = $(field_selector).data().typeahead;
 | 
						|
                tah.mouseenter({
 | 
						|
                    currentTarget: $(`.typeahead:visible li:contains("${CSS.escape(item)}")`)[0],
 | 
						|
                });
 | 
						|
                tah.select();
 | 
						|
            },
 | 
						|
            field_selector,
 | 
						|
            str,
 | 
						|
            item,
 | 
						|
        );
 | 
						|
    }
 | 
						|
 | 
						|
    async run_test(test_function) {
 | 
						|
        // Pass a page instance to test so we can take
 | 
						|
        // a screenshot of it when the test fails.
 | 
						|
        const page = await this.get_page();
 | 
						|
        try {
 | 
						|
            await test_function(page);
 | 
						|
        } catch (error) {
 | 
						|
            console.log(error);
 | 
						|
 | 
						|
            // Take a screenshot, and increment the screenshot_id.
 | 
						|
            await this.screenshot(page, `failure-${this.screenshot_id}`);
 | 
						|
            this.screenshot_id += 1;
 | 
						|
 | 
						|
            await this.browser.close();
 | 
						|
            process.exit(1);
 | 
						|
        } finally {
 | 
						|
            this.browser.close();
 | 
						|
        }
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
const common = new CommonUtils();
 | 
						|
module.exports = common;
 |