fixup

Author: mrgrain

packages/@aws-cdk/toolkit-lib/test/_helpers/index.ts

@@ -4,8 +4,8 @@ import * as path from 'node:path';
 import type { AssemblyDirectoryProps, Toolkit } from '../../lib';
 import { ToolkitError } from '../../lib';
 
-export * from '../../lib/api/shared-private';
 export * from './test-cloud-assembly-source';
+export * from './test-io-host';
 
 function fixturePath(...parts: string[]): string {
   return path.normalize(path.join(__dirname, '..', '_fixtures', ...parts));

packages/@aws-cdk/toolkit-lib/test/_helpers/test-io-host.ts

@@ -0,0 +1,62 @@
+import type { IIoHost, IoMessage, IoMessageLevel, IoRequest } from '../../lib/api/io';
+import { asIoHelper, isMessageRelevantForLevel, type IoHelper } from '../../lib/api/shared-private';
+import { RequireApproval } from '../../lib/toolkit';
+
+/**
+ * A test implementation of IIoHost that does nothing but can be spied on.
+ *
+ * Includes a level to filter out irrelevant messages, defaults to `info`.
+ *
+ * Optionally set an approval level for code `CDK_TOOLKIT_I5060`.
+ *
+ * # How to use
+ *
+ * Configure and reset the `notifySpy` and `requestSpy` members as you would any
+ * mock function.
+ */
+export class TestIoHost implements IIoHost {
+  public readonly notifySpy: jest.Mock<any, any, any>;
+  public readonly requestSpy: jest.Mock<any, any, any>;
+
+  public requireDeployApproval = RequireApproval.NEVER;
+
+  constructor(public level: IoMessageLevel = 'info') {
+    this.notifySpy = jest.fn();
+    this.requestSpy = jest.fn();
+  }
+
+  public asHelper(action = 'synth'): IoHelper {
+    return asIoHelper(this, action as any);
+  }
+
+  public async notify(msg: IoMessage<unknown>): Promise<void> {
+    if (isMessageRelevantForLevel(msg, this.level)) {
+      this.notifySpy(msg);
+    }
+  }
+
+  public async requestResponse<T, U>(msg: IoRequest<T, U>): Promise<U> {
+    if (isMessageRelevantForLevel(msg, this.level) && this.needsApproval(msg)) {
+      this.requestSpy(msg);
+    }
+    return msg.defaultResponse;
+  }
+
+  private needsApproval(msg: IoRequest<any, any>): boolean {
+    // Return true if the code is unrelated to approval
+    if (!['CDK_TOOLKIT_I5060'].includes(msg.code)) {
+      return true;
+    }
+
+    switch (this.requireDeployApproval) {
+      case RequireApproval.NEVER:
+        return false;
+      case RequireApproval.ANY_CHANGE:
+        return true;
+      case RequireApproval.BROADENING:
+        return msg.data?.permissionChangeType === 'broadening';
+      default:
+        return true;
+    }
+  }
+}

packages/@aws-cdk/toolkit-lib/test/util/aws-cdk.ts

@@ -9,4 +9,4 @@ export {
   mockSTSClient,
   setDefaultSTSMocks,
   restoreSdkMocksToDefault,
-} from '../../../../aws-cdk/test/util/mock-sdk';
+} from '../../../../aws-cdk/test/_helpers/mock-sdk';