immich/cli/test/cli-test-utils.ts

import fs from 'node:fs';
import path from 'node:path';
import { ImmichApi } from 'src/services/api.service';

export const TEST_CONFIG_DIR = '/tmp/immich/';
export const TEST_AUTH_FILE = path.join(TEST_CONFIG_DIR, 'auth.yml');
export const TEST_IMMICH_INSTANCE_URL = 'https://test/api';
export const TEST_IMMICH_API_KEY = 'pNussssKSYo5WasdgalvKJ1n9kdvaasdfbluPg';

export const CLI_BASE_OPTIONS = { configDirectory: TEST_CONFIG_DIR };

export const setup = async () => {
  const api = new ImmichApi(process.env.IMMICH_INSTANCE_URL as string, '');
  await api.signUpAdmin({ email: 'cli@immich.app', password: 'password', name: 'Administrator' });
  const admin = await api.login({ email: 'cli@immich.app', password: 'password' });
  const apiKey = await api.createApiKey(
    { name: 'CLI Test' },
    { headers: { Authorization: `Bearer ${admin.accessToken}` } },
  );

  api.setApiKey(apiKey.secret);

  return api;
};

export const spyOnConsole = () => vi.spyOn(console, 'log').mockImplementation(() => {});

export const createTestAuthFile = async (contents: string) => {
  if (!fs.existsSync(TEST_CONFIG_DIR)) {
    // Create config folder if it doesn't exist
    const created = await fs.promises.mkdir(TEST_CONFIG_DIR, { recursive: true });
    if (!created) {
      throw new Error(`Failed to create config folder ${TEST_CONFIG_DIR}`);
    }
  }

  fs.writeFileSync(TEST_AUTH_FILE, contents);
};

export const readTestAuthFile = async (): Promise<string> => {
  return await fs.promises.readFile(TEST_AUTH_FILE, 'utf8');
};

export const deleteAuthFile = () => {
  try {
    fs.unlinkSync(TEST_AUTH_FILE);
  } catch (error: any) {
    if (error.code !== 'ENOENT') {
      throw error;
    }
  }
};