zulip-desktop/tests/setup.js

import fs from "node:fs";
import path from "node:path";
import process from "node:process";

import {_electron} from "playwright-core";

const testsPackage = JSON.parse(
  fs.readFileSync(new URL("package.json", import.meta.url), "utf8"),
);

// Runs Zulip Desktop.
// Returns a promise that resolves to an Electron Application once the app has loaded.
export function createApp() {
  return _electron.launch({
    args: [import.meta.dirname], // Ensure this dir has a package.json file with a 'main' entry point
  });
}

// Quit the app, end the test
export async function endTest(app) {
  await app.close();
}

function getAppDataDirectory() {
  let base;

  switch (process.platform) {
    case "darwin": {
      base = path.join(process.env.HOME, "Library", "Application Support");
      break;
    }

    case "linux": {
      base =
        process.env.XDG_CONFIG_HOME ?? path.join(process.env.HOME, ".config");
      break;
    }

    case "win32": {
      base = process.env.APPDATA;
      break;
    }

    default: {
      throw new Error("Could not detect app data dir base.");
    }
  }

  console.log("Detected App Data Dir base:", base);
  return path.join(base, testsPackage.productName);
}

// Resets the test directory, containing domain.json, window-state.json, etc
export function resetTestDataDirectory() {
  const appDataDirectory = getAppDataDirectory();
  fs.rmSync(appDataDirectory, {force: true, recursive: true});
}