Use.test.ts

import {ppath, xfs, npath}                from '@yarnpkg/fslib';
import process                            from 'node:process';
import {parseEnv}                         from 'node:util';
import {describe, beforeEach, it, expect} from 'vitest';

import {runCli}                           from './_runCli';

beforeEach(async () => {
  const home = await xfs.mktempPromise();

  // `process.env` is reset after each tests in setupTests.js.
  process.env.COREPACK_HOME = npath.fromPortablePath(home);
  process.env.COREPACK_DEFAULT_TO_LATEST = `0`;

  return async () => {
    await xfs.removePromise(home, {recursive: true});
  };
});

describe(`UseCommand`, () => {
  describe(`should set the package manager in the current project`, () => {
    it(`With an existing 'packageManager' field`, async () => {
      await xfs.mktempPromise(async cwd => {
        await xfs.writeJsonPromise(ppath.join(cwd, `package.json`), {
          packageManager: `yarn@1.0.0`,
          license: `MIT`,
        });

        await expect(runCli(cwd, [`use`, `yarn@1.22.4`])).resolves.toMatchObject({
          exitCode: 0,
          stdout: expect.stringMatching(/^Installing yarn@1\.22\.4 in the project\.\.\.\n\n/),
          stderr: ``,
        });

        await expect(xfs.readJsonPromise(ppath.join(cwd, `package.json`))).resolves.toMatchObject({
          packageManager: `yarn@1.22.4+sha512.a1833b862fe52169bd6c2a033045a07df5bc6a23595c259e675fed1b2d035ab37abe6ce309720abb6636d68f03615054b6292dc0a70da31c8697fda228b50d18`,
        });

        await expect(runCli(cwd, [`yarn`, `--version`])).resolves.toMatchObject({
          exitCode: 0,
          stdout: `1.22.4\n`,
          stderr: ``,
        });
      });
    });
    it(`with 'devEngines.packageManager' field`, async () => {
      await xfs.mktempPromise(async cwd => {
        process.env.NO_COLOR = `1`;
        const devEngines = {packageManager: {name: `yarn`, version: `2.x`}};
        await xfs.writeJsonPromise(ppath.join(cwd, `package.json`), {
          devEngines,
        });

        // Should refuse to install an incompatible version:
        await expect(runCli(cwd, [`use`, `yarn@1.22.4`])).resolves.toMatchObject({
          exitCode: 1,
          stderr: ``,
          stdout: `Installing yarn@1.22.4 in the project...\nUsage Error: The requested version of yarn@1.22.4+sha512.a1833b862fe52169bd6c2a033045a07df5bc6a23595c259e675fed1b2d035ab37abe6ce309720abb6636d68f03615054b6292dc0a70da31c8697fda228b50d18 does not match the devEngines specification (yarn@2.x)\n\n$ corepack use <pattern>\n`,
        });

        // Should accept setting to a compatible version:
        await expect(runCli(cwd, [`use`, `yarn@2.4.3`])).resolves.toMatchObject({
          exitCode: 0,
          stderr: ``,
          stdout: expect.stringMatching(/^Installing yarn@2\.4\.3 in the project\.\.\.\n\n/),
        });

        await expect(xfs.readJsonPromise(ppath.join(cwd, `package.json`))).resolves.toMatchObject({
          devEngines,
          packageManager: `yarn@2.4.3+sha512.8dd9fedc5451829619e526c56f42609ad88ae4776d9d3f9456d578ac085115c0c2f0fb02bb7d57fd2e1b6e1ac96efba35e80a20a056668f61c96934f67694fd0`,
        });

        await expect(runCli(cwd, [`yarn`, `--version`])).resolves.toMatchObject({
          exitCode: 0,
          stdout: `2.4.3\n`,
          stderr: ``,
        });
      });
    });

    it(`with 'devEngines.packageManager' and 'packageManager' fields`, async () => {
      await xfs.mktempPromise(async cwd => {
        process.env.NO_COLOR = `1`;
        const devEngines = {packageManager: {name: `yarn`, version: `1.x || 2.x`}};
        await xfs.writeJsonPromise(ppath.join(cwd, `package.json`), {
          devEngines,
          packageManager: `yarn@1.1.0`,
          license: `MIT`,
        });

        // Should refuse to install an incompatible version:
        await expect(runCli(cwd, [`use`, `yarn@1.22.4`])).resolves.toMatchObject({
          exitCode: 0,
          stderr: ``,
          stdout: expect.stringMatching(/^Installing yarn@1\.22\.4 in the project\.\.\.\n\n/),
        });

        // Should accept setting to a compatible version:
        await expect(runCli(cwd, [`use`, `yarn@2.4.3`])).resolves.toMatchObject({
          exitCode: 0,
          stderr: ``,
          stdout: expect.stringMatching(/^Installing yarn@2\.4\.3 in the project\.\.\.\n\n/),
        });

        await expect(xfs.readJsonPromise(ppath.join(cwd, `package.json`))).resolves.toMatchObject({
          devEngines,
          packageManager: `yarn@2.4.3+sha512.8dd9fedc5451829619e526c56f42609ad88ae4776d9d3f9456d578ac085115c0c2f0fb02bb7d57fd2e1b6e1ac96efba35e80a20a056668f61c96934f67694fd0`,
        });

        await expect(runCli(cwd, [`yarn`, `--version`])).resolves.toMatchObject({
          exitCode: 0,
          stdout: `2.4.3\n`,
          stderr: ``,
        });
      });
    });
  });

  it(`should update .corepack.env if present and contains definition for pm version`, async t => {
    // Skip that test on Node.js 18.x as it lacks support for .env files.
    if (process.version.startsWith(`v18.`)) t.skip();

    await Promise.all([
      `COREPACK_DEV_ENGINES_YARN=1.1.0\n`,
      `\nCOREPACK_DEV_ENGINES_YARN=1.1.0\n`,
      `COREPACK_DEV_ENGINES_YARN=1.1.0`,
      `\nCOREPACK_DEV_ENGINES_YARN=1.1.0`,
      `FOO=bar\nCOREPACK_DEV_ENGINES_YARN=1.1.0\n`,
      `FOO=bar\nCOREPACK_DEV_ENGINES_YARN=1.1.0`,
    ].map(originalEnv => xfs.mktempPromise(async cwd => {
      const pJSONContent = {
        devEngines: {packageManager: {name: `yarn`, version: `1.x`}},
        license: `MIT`,
      };
      await xfs.writeJsonPromise(ppath.join(cwd, `package.json`), pJSONContent);
      await xfs.writeFilePromise(ppath.join(cwd, `.corepack.env`), originalEnv);

      await expect(runCli(cwd, [`use`, `yarn@1.22.4`])).resolves.toMatchObject({
        exitCode: 0,
        stdout: expect.stringContaining(`Installing yarn@1.22.4 in the project...`),
        stderr: ``,
      });

      try {
        await expect(xfs.readFilePromise(ppath.join(cwd, `.corepack.env`), `utf-8`).then(parseEnv)).resolves.toMatchObject({
          COREPACK_DEV_ENGINES_YARN: `1.22.4+sha512.a1833b862fe52169bd6c2a033045a07df5bc6a23595c259e675fed1b2d035ab37abe6ce309720abb6636d68f03615054b6292dc0a70da31c8697fda228b50d18`,
        });
      } catch (cause) {
        throw new Error(JSON.stringify(originalEnv), {cause});
      }
      // It should not have touched package.json.
      await expect(xfs.readJsonPromise(ppath.join(cwd, `package.json`))).resolves.toStrictEqual(pJSONContent);

      await expect(runCli(cwd, [`yarn`, `--version`])).resolves.toMatchObject({
        exitCode: 0,
        stdout: `1.22.4\n`,
      });
    })));
  });

  it(`should update .other.env if present`, async t => {
    // Skip that test on Node.js 18.x as it lacks support for .env files.
    if (process.version.startsWith(`v18.`)) t.skip();

    await Promise.all([
      `COREPACK_DEV_ENGINES_YARN=1.1.0\n`,
      `\nCOREPACK_DEV_ENGINES_YARN=1.1.0\n`,
      `COREPACK_DEV_ENGINES_YARN=1.1.0`,
      `\nCOREPACK_DEV_ENGINES_YARN=1.1.0`,
      `FOO=bar\nCOREPACK_DEV_ENGINES_YARN=1.1.0\n`,
      `FOO=bar\nCOREPACK_DEV_ENGINES_YARN=1.1.0`,
    ].map(originalEnv => xfs.mktempPromise(async cwd => {
      await xfs.writeJsonPromise(ppath.join(cwd, `package.json`), {
        devEngines: {packageManager: {name: `yarn`, version: `1.x`}},
      });
      await xfs.writeFilePromise(ppath.join(cwd, `.other.env`), `COREPACK_DEV_ENGINES_YARN=1.0.0\n`);

      process.env.COREPACK_ENV_FILE = `.other.env`;
      await expect(runCli(cwd, [`use`, `yarn@1.22.4`])).resolves.toMatchObject({
        exitCode: 0,
      });

      try {
        await expect(xfs.readFilePromise(ppath.join(cwd, `.other.env`), `utf-8`).then(parseEnv)).resolves.toMatchObject({
          COREPACK_DEV_ENGINES_YARN: `1.22.4+sha512.a1833b862fe52169bd6c2a033045a07df5bc6a23595c259e675fed1b2d035ab37abe6ce309720abb6636d68f03615054b6292dc0a70da31c8697fda228b50d18`,
        });
      } catch (cause) {
        throw new Error(JSON.stringify(originalEnv), {cause});
      }

      await expect(runCli(cwd, [`yarn`, `--version`])).resolves.toMatchObject({
        exitCode: 0,
        stdout: `1.22.4\n`,
      });
    })));
  });

  it(`should create a package.json if absent`, async () => {
    await xfs.mktempPromise(async cwd => {
      await expect(runCli(cwd, [`use`, `yarn@1.22.4`])).resolves.toMatchObject({
        exitCode: 0,
        stderr: `warning package.json: No license field\nwarning No license field\n`,
      });

      await expect(xfs.readJsonPromise(ppath.join(cwd, `package.json`))).resolves.toMatchObject({
        packageManager: `yarn@1.22.4+sha512.a1833b862fe52169bd6c2a033045a07df5bc6a23595c259e675fed1b2d035ab37abe6ce309720abb6636d68f03615054b6292dc0a70da31c8697fda228b50d18`,
      });

      await expect(runCli(cwd, [`yarn`, `--version`])).resolves.toMatchObject({
        exitCode: 0,
        stdout: `1.22.4\n`,
        stderr: ``,
      });

      // Ensure Corepack is able to detect package.json in parent directory
      const subfolder = ppath.join(cwd, `subfolder`);
      await xfs.mkdirPromise(subfolder);

      await expect(runCli(subfolder, [`use`, `yarn@2.2.2`])).resolves.toMatchObject({
        exitCode: 0,
        stderr: ``,
      });
      await expect(runCli(cwd, [`yarn`, `--version`])).resolves.toMatchObject({
        exitCode: 0,
        stdout: `2.2.2\n`,
        stderr: ``,
      });
    });
  });

  describe(`should not care if packageManager is set to an invalid value`, () => {
    for (const {description, packageManager} of [
      {
        description: `when a version range is given`,
        packageManager: `yarn@1.x`,
      },
      {
        description: `when only the pm name is given`,
        packageManager: `yarn`,
      },
      {
        description: `when the version is missing`,
        packageManager: `yarn@`,
      },
      {
        description: `when the field is not a string`,
        packageManager: [],
      },
    ]) {
      it(description, async () => {
        await xfs.mktempPromise(async cwd => {
          await xfs.writeJsonPromise(ppath.join(cwd, `package.json`), {
            packageManager,
            license: `MIT`, // To avoid warning
          });

          await expect(runCli(cwd, [`use`, `yarn@1.22.4`])).resolves.toMatchObject({
            exitCode: 0,
            stderr: ``,
            stdout: expect.stringMatching(/^Installing yarn@1\.22\.4 in the project\.\.\.\n\n/),
          });

          await expect(xfs.readJsonPromise(ppath.join(cwd, `package.json`))).resolves.toMatchObject({
            packageManager: `yarn@1.22.4+sha512.a1833b862fe52169bd6c2a033045a07df5bc6a23595c259e675fed1b2d035ab37abe6ce309720abb6636d68f03615054b6292dc0a70da31c8697fda228b50d18`,
          });

          await expect(runCli(cwd, [`yarn`, `--version`])).resolves.toMatchObject({
            exitCode: 0,
            stdout: `1.22.4\n`,
            stderr: ``,
          });
        });
      });
    }
  });

  it(`should update the ".corepack.env" file from the current project`, async t => {
    // Skip that test on Node.js 18.x as it lacks support for .env files.
    if (process.version.startsWith(`v18.`)) t.skip();
    await Promise.all([
      `COREPACK_DEV_ENGINES_YARN=2.1.0\n`,
      `\nCOREPACK_DEV_ENGINES_YARN=2.1.0\n`,
      `COREPACK_DEV_ENGINES_YARN=2.1.0`,
      `\nCOREPACK_DEV_ENGINES_YARN=2.1.0`,
      `FOO=bar\nCOREPACK_DEV_ENGINES_YARN=2.1.0\n`,
      `FOO=bar\nCOREPACK_DEV_ENGINES_YARN=2.1.0`,
    ].map(originalEnv => xfs.mktempPromise(async cwd => {
      await xfs.writeJsonPromise(ppath.join(cwd, `package.json`), {
        devEngines: {packageManager: {name: `yarn`, version: `1.x || 2.x`}},
        license: `MIT`, // To avoid Yarn warning.
      });
      await xfs.writeFilePromise(ppath.join(cwd, `.corepack.env`), originalEnv);

      await expect(runCli(cwd, [`use`, `yarn@1.22.4`])).resolves.toMatchObject({
        exitCode: 0,
        stderr: ``,
        stdout: expect.stringMatching(/^Installing yarn@1\.22\.4 in the project\.\.\.\n\n/),
      });

      try {
        await expect(xfs.readFilePromise(ppath.join(cwd, `.corepack.env`), `utf-8`).then(parseEnv)).resolves.toMatchObject({
          COREPACK_DEV_ENGINES_YARN: `1.22.4+sha512.a1833b862fe52169bd6c2a033045a07df5bc6a23595c259e675fed1b2d035ab37abe6ce309720abb6636d68f03615054b6292dc0a70da31c8697fda228b50d18`,
        });
      } catch (cause) {
        throw new Error(JSON.stringify(originalEnv), {cause});
      }

      await expect(runCli(cwd, [`yarn`, `--version`])).resolves.toMatchObject({
        exitCode: 0,
        stdout: `1.22.4\n`,
        stderr: ``,
      });
    })));
  });
});