From 02a5708fe62a3f4aa3f5cc2124a78bfe48e56bcb Mon Sep 17 00:00:00 2001
From: Kanad Gupta <git@kanad.dev>
Date: Thu, 17 Oct 2024 14:06:59 -0500
Subject: [PATCH] feat: add ability to override `NODE_ENV`

---
 README.md    | 1 +
 src/index.ts | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 73791dc..b5e3fbd 100644
--- a/README.md
+++ b/README.md
@@ -24,6 +24,7 @@ See the [V4 Migration Guide](./MIGRATION.md) if you are migrating from v3 or old
 
 - `print` - Print everything that goes to stdout and stderr.
 - `stripAnsi` - Strip ansi codes from everything that goes to stdout and stderr. Defaults to true.
+- `testNodeEnv` - Sets the `NODE_ENV` value when capturing output. Defaults to `'test'`.
 
 See the [tests](./test/capture-output.test.ts) for example usage.
 
diff --git a/src/index.ts b/src/index.ts
index 8ae63e6..66da1d0 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -8,6 +8,7 @@ const debug = makeDebug('oclif-test')
 type CaptureOptions = {
   print?: boolean
   stripAnsi?: boolean
+  testNodeEnv?: string
 }
 
 type CaptureResult<T> = {
@@ -77,6 +78,7 @@ function splitString(str: string): string[] {
 export async function captureOutput<T>(fn: () => Promise<unknown>, opts?: CaptureOptions): Promise<CaptureResult<T>> {
   const print = opts?.print ?? false
   const stripAnsi = opts?.stripAnsi ?? true
+  const testNodeEnv = opts?.testNodeEnv || 'test'
 
   const originals = {
     NODE_ENV: process.env.NODE_ENV,
@@ -112,7 +114,7 @@ export async function captureOutput<T>(fn: () => Promise<unknown>, opts?: Captur
 
   process.stdout.write = mock('stdout')
   process.stderr.write = mock('stderr')
-  process.env.NODE_ENV = 'test'
+  process.env.NODE_ENV = testNodeEnv
 
   try {
     const result = await fn()