mirror of
https://github.com/beekeeper-studio/beekeeper-studio.git
synced 2026-03-13 10:12:54 +08:00
19eafa6c0b
When a test or beforeEach fails before electronApp is assigned, the afterEach would crash trying to call .close() on undefined, masking the real error. Add a null check in all 9 e2e test files. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
82 lines
2.9 KiB
TypeScript
82 lines
2.9 KiB
TypeScript
import { _electron as electron } from 'playwright';
|
|
import { test, expect, beforeEach, afterEach } from '@playwright/test';
|
|
import { QueryTab } from '../pageComponents/QueryTab';
|
|
import { QueryResultPane } from '../pageComponents/QueryResultPane';
|
|
import { userActions } from "../pageActions/index";
|
|
import { POSTGRES_CONFIG } from './config/postgresDbConfig';
|
|
|
|
let electronApp;
|
|
let window;
|
|
let queryTab;
|
|
let resultPane;
|
|
let userAttemptsTo;
|
|
const testQueryPrefix = `SELECT * FROM actor`;
|
|
|
|
test.describe("Postgres query execution", () => {
|
|
|
|
beforeEach(async () => {
|
|
electronApp = await electron.launch({ args: ['dist/main.js'] });
|
|
window = await electronApp.firstWindow();
|
|
queryTab = new QueryTab(window);
|
|
resultPane = new QueryResultPane(window);
|
|
userAttemptsTo = userActions(window);
|
|
});
|
|
|
|
afterEach(async () => {
|
|
if (electronApp) {
|
|
await electronApp.close();
|
|
}
|
|
});
|
|
|
|
test("perform a Postgres query", async () => {
|
|
const postgresQuery = `${testQueryPrefix} limit 1;`;
|
|
|
|
await userAttemptsTo.selectNewConnection(POSTGRES_CONFIG.connectionType);
|
|
await userAttemptsTo.insertDatabaseDetails(POSTGRES_CONFIG);
|
|
await userAttemptsTo.connectWithDatabase();
|
|
|
|
await expect(queryTab.queryTabTextArea).toBeVisible();
|
|
|
|
await userAttemptsTo.writeAQuery(postgresQuery);
|
|
await userAttemptsTo.runQuery();
|
|
|
|
await expect(resultPane.resultFirstRow).toBeVisible();
|
|
});
|
|
|
|
test("postgres query with WHERE and 2 results", async () => {
|
|
const queryWithConditionals = `${testQueryPrefix} WHERE actor_id IN (1, 2);`
|
|
|
|
await userAttemptsTo.selectNewConnection(POSTGRES_CONFIG.connectionType);
|
|
await userAttemptsTo.insertDatabaseDetails(POSTGRES_CONFIG);
|
|
await userAttemptsTo.connectWithDatabase();
|
|
|
|
await expect(queryTab.queryTabTextArea).toBeVisible();
|
|
|
|
await userAttemptsTo.writeAQuery(queryWithConditionals);
|
|
await userAttemptsTo.runQuery();
|
|
|
|
await expect(resultPane.resultFirstRow).toBeVisible();
|
|
await expect(resultPane.resultSecondRow).toBeVisible();
|
|
await expect(resultPane.resultThridRow).not.toBeVisible();
|
|
});
|
|
|
|
test("runs valid query with no results", async () => {
|
|
const zeroResultsQuery = `${testQueryPrefix} actor WHERE actor_id = null;`
|
|
|
|
await userAttemptsTo.selectNewConnection(POSTGRES_CONFIG.connectionType);
|
|
await userAttemptsTo.insertDatabaseDetails(POSTGRES_CONFIG);
|
|
await userAttemptsTo.connectWithDatabase();
|
|
|
|
await expect(queryTab.queryTabTextArea).toBeVisible();
|
|
|
|
await userAttemptsTo.writeAQuery(zeroResultsQuery);
|
|
await userAttemptsTo.runQuery();
|
|
|
|
await expect(resultPane.resultFirstRow).not.toBeVisible();
|
|
await expect(resultPane.resultSecondRow).not.toBeVisible();
|
|
await expect(resultPane.resultThridRow).not.toBeVisible();
|
|
await expect(resultPane.noResults).toBeVisible();
|
|
});
|
|
|
|
});
|