beekeeper-studio/apps/studio/e2e/tests/copyResults.test.ts

import { _electron as electron } from 'playwright';
import { test, expect, beforeEach, afterEach } from '@playwright/test';
import { QueryTab } from '../pageComponents/QueryTab';
import { Footer } from '../pageComponents/Footer';
import { QueryResultPane } from '../pageComponents/QueryResultPane';
import { userActions } from "../pageActions/index";
import { POSTGRES_CONFIG } from './config/postgresDbConfig';

const POSTGRES_QUERY = 'SELECT * FROM actor WHERE actor_id IN (1, 2);';

let electronApp;
let window;
let queryTab;
let footer;
let resultPane;
let userAttemptsTo;

test.describe("Copy Results Verifications", () => {

    beforeEach(async () => {
        electronApp = await electron.launch({
            args: ['dist/main.js'],
        });
        window = await electronApp.firstWindow();
        queryTab = new QueryTab(window);
        resultPane = new QueryResultPane(window);
        footer = new Footer(window);
        userAttemptsTo = userActions(window);


        await userAttemptsTo.selectNewConnection(POSTGRES_CONFIG.connectionType);
        await userAttemptsTo.insertDatabaseDetails(POSTGRES_CONFIG);
        await userAttemptsTo.connectWithDatabase();

        await expect(queryTab.queryTabTextArea).toBeVisible();
        await userAttemptsTo.writeAQuery(POSTGRES_QUERY);
        await userAttemptsTo.runQuery();
        await expect(resultPane.resultSecondRow).toBeVisible();
    });

    afterEach(async () => {
        if (electronApp) {
            await electronApp.close();
        }
    });

    test("copy as TSV / Excel", async () => {
        const fileType = "TSV / Excel";

        await userAttemptsTo.clickOnDownloadMenu();

        const copyOption = await footer.copyFromDownloadMenu(fileType);

        await userAttemptsTo.clickOnCopyToClipboardOption(fileType);
        await expect(copyOption).toBeHidden();
        await expect(footer.copyToClipboardToast).toBeVisible();
    });

    test("copy as Markdown", async () => {
        const fileType = "Markdown";

        await userAttemptsTo.clickOnDownloadMenu();

        const copyOption = await footer.copyFromDownloadMenu(fileType);

        await userAttemptsTo.clickOnCopyToClipboardOption(fileType);
        await expect(copyOption).toBeHidden();
        await expect(footer.copyToClipboardToast).toBeVisible();
    });
});