From d2c0926c7bb50607d9daa8aafb757bc56ad8eccd Mon Sep 17 00:00:00 2001
From: Brenton Simpson <appsforartists@google.com>
Date: Fri, 12 Jan 2024 15:52:25 -0800
Subject: [PATCH] WIP add test for #960

---
 src/test/basic.test.ts | 105 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 105 insertions(+)

diff --git a/src/test/basic.test.ts b/src/test/basic.test.ts
index b03731b40..3ed76a3d1 100644
--- a/src/test/basic.test.ts
+++ b/src/test/basic.test.ts
@@ -6,8 +6,10 @@
 
 import {suite} from 'uvu';
 import * as assert from 'uvu/assert';
+import {dedent} from 'ts-dedent';
 import {rigTest} from './util/rig-test.js';
 import {IS_WINDOWS} from '../util/windows.js';
+import {injectYarnBerryToRig} from './util/yarn-berry.js';
 import {NODE_MAJOR_VERSION} from './util/node-version.js';
 import {checkScriptOutput} from './util/check-script-output.js';
 
@@ -880,6 +882,109 @@ test(
   }),
 );
 
+test(
+  'commands can be run locally in yarn berry',
+  rigTest(async ({rig}) => {
+    await injectYarnBerryToRig(rig);
+
+    const rootCmd = await rig.newCommand();
+    const innerCmd = await rig.newCommand();
+
+    const originalLock = await rig.read('yarn.lock');
+
+    await rig.write({
+      'package.json': {
+        private: true,
+        workspaces: ['packages/*'],
+        scripts: {
+          cmd: 'yarn run -B wireit',
+        },
+        wireit: {
+          cmd: {
+            command: rootCmd.command,
+          },
+        },
+        devDependencies: {
+          wireit: '*',
+        },
+        resolutions: {
+          wireit: `portal:${process.cwd()}`,
+        },
+      },
+
+      'packages/inner/package.json': {
+        scripts: {
+          cmd: 'yarn run -TB wireit',
+        },
+        wireit: {
+          cmd: {
+            command: innerCmd.command,
+          },
+        },
+      },
+
+      // On a real system, `yarn` should do this automatically; however, in this
+      // test rig, a stale copy (`originalLock`) remains, even if we manually
+      // run `yarn`.  Therefore, we manually update the lockfile here to include
+      // the inner package.
+      'yarn.lock':
+        originalLock +
+        dedent`
+
+        "inner-d81e84@workspace:packages/inner":
+          version: 0.0.0-use.local
+          resolution: "inner-d81e84@workspace:packages/inner"
+          languageName: unknown
+          linkType: soft
+
+        "wireit@portal:${process.cwd()}::locator=root-workspace-0b6124%40workspace%3A.":
+          version: 0.0.0-use.local
+          resolution: "wireit@portal:${process.cwd()}::locator=root-workspace-0b6124%40workspace%3A."
+          dependencies:
+            braces: "npm:^3.0.2"
+            chokidar: "npm:^3.5.3"
+            dedent: "npm:^1.5.1"
+            fast-glob: "npm:^3.2.11"
+            jsonc-parser: "npm:^3.0.0"
+            proper-lockfile: "npm:^4.1.2"
+            ts-dedent: "npm:^2.2.0"
+          bin:
+            wireit: bin/wireit.js
+          languageName: node
+          linkType: soft
+      `,
+    });
+
+    // this is what should modify the lockfile, so we don't have to do it
+    // manually
+    rig.exec('yarn');
+
+    // logging: can be deleted once this works
+    //
+    // since the `.bin` is already set up by the test rig, IDK if we should need to
+    // modify the lockfile to include wireit.  however, it's saying
+    //
+    //     Usage Error: Couldn't find a script name "wireit" in the top-level (used by inner-d81e84@workspace:packages/inner).
+    //
+    // so i'm trying to make it work
+    rig.exec('ls -la node_modules/.bin');
+    rig.read('yarn.lock').then(console.log);
+
+    {
+      const exec = rig.exec('yarn run cmd', {cwd: 'packages/inner'});
+      (
+        await Promise.race([
+          innerCmd.nextInvocation(),
+          rootCmd.nextInvocation(),
+        ])
+      ).exit(0);
+      assert.equal((await exec.exit).code, 0);
+      assert.equal(innerCmd.numInvocations, 1);
+      assert.equal(rootCmd.numInvocations, 0);
+    }
+  }),
+);
+
 test(
   'commands run under pnpm workspaces',
   rigTest(async ({rig}) => {