feat(adapter): add addAnswers support (#96)

mshima · web-flow · commit 126aa8379699 · 2025-04-14T20:41:38.000-03:00
diff --git a/workspaces/adapter/src/testing/test-adapter.ts b/workspaces/adapter/src/testing/test-adapter.ts
@@ -48,8 +48,8 @@ const isValueSet = (type: string, answer: any) => {
 };
 
 const createDummyPrompt = (options: DummyPromptOptions = {}) => {
+  const { mockedAnswers = {}, callback = answer => answer, throwOnMissingAnswer = false } = options;
   return createPrompt<any, TestQuestion>((config, done) => {
-    const { mockedAnswers = {}, callback = answer => answer, throwOnMissingAnswer = false } = options;
     let answer = mockedAnswers[config.name!];
 
     if (!isValueSet(config.type, answer)) {
@@ -68,7 +68,7 @@ const createDummyPrompt = (options: DummyPromptOptions = {}) => {
         answer = true;
       }
     }
-    done(callback(answer, { question: config, answers: { ...mockedAnswers, [config.name]: answer } }));
+    done(callback(answer, { question: config, answers: { [config.name]: answer } }));
 
     return config.message;
   });
@@ -86,14 +86,18 @@ export class TestAdapter<LogType extends Logger = Logger, SpyType = any> impleme
   diff: any & SpyType;
   log: LogType & SpyType;
   registerDummyPrompt: (promptName: string, customPromptOptions?: DummyPromptOptions) => PromptModule;
+  readonly mockedAnswers: PromptAnswers;
+
   private abortController = new AbortController();
   private readonly spyFactory: SpyFactory<SpyType>;
 
   constructor(options: TestAdapterOptions<SpyType> = {}) {
     const {
       spyFactory = defaultConfig.spyFactory ?? (spyOptions => () => spyOptions.returns),
       log = defaultConfig.log ?? createLogger(),
-      ...promptOptions
+      mockedAnswers,
+      callback,
+      throwOnMissingAnswer,
     } = options;
 
     this.spyFactory = spyFactory;
@@ -104,10 +108,16 @@ export class TestAdapter<LogType extends Logger = Logger, SpyType = any> impleme
       signal: this.abortController.signal,
     });
 
+    this.mockedAnswers = {};
+    this.addAnswers(mockedAnswers ?? {});
+
     const actualRegisterPrompt = this.promptModule.registerPrompt.bind(this.promptModule);
 
     this.registerDummyPrompt = (promptModuleName: string, customPromptOptions?: DummyPromptOptions) =>
-      actualRegisterPrompt(promptModuleName, createDummyPrompt(customPromptOptions ?? promptOptions));
+      actualRegisterPrompt(
+        promptModuleName,
+        createDummyPrompt({ callback, mockedAnswers: this.mockedAnswers, throwOnMissingAnswer, ...customPromptOptions }),
+      );
 
     this.promptModule.registerPrompt = (name: string) => this.registerDummyPrompt(name);
 
@@ -151,4 +161,17 @@ export class TestAdapter<LogType extends Logger = Logger, SpyType = any> impleme
   ): Promise<A> {
     return this.promptModule(questions, initialAnswers);
   }
+
+  /**
+   * Add answers to the mocked answers.
+   */
+  addAnswers(answers: PromptAnswers): void {
+    // Copy properties using Object.getOwnPropertyDescriptor to preserve getters and setters
+    for (const key in answers) {
+      const descriptors = Object.getOwnPropertyDescriptor(answers, key);
+      if (descriptors) {
+        Object.defineProperty(this.mockedAnswers, key, descriptors);
+      }
+    }
+  }
 }
diff --git a/workspaces/adapter/test/test-adapter.test.ts b/workspaces/adapter/test/test-adapter.test.ts
@@ -70,6 +70,28 @@ describe('TestAdapter', () => {
         respuesta: null,
       });
     });
+    it('addAnswers adds answers to mockedAnswers', async () => {
+      const adapter = new TestAdapter();
+
+      adapter.addAnswers({ respuesta: 'foo' });
+
+      await expect(adapter.prompt([{ name: 'respuesta', message: 'foo', type: 'list' }])).resolves.toMatchObject({ respuesta: 'foo' });
+      expect(adapter.mockedAnswers).toMatchObject(expect.objectContaining({ respuesta: 'foo' }));
+    });
+    it('addAnswers supports getters', async () => {
+      const adapter = new TestAdapter();
+
+      adapter.addAnswers({
+        _orderedRespuestas: ['foo', 'bar'],
+        get respuesta() {
+          return this._orderedRespuestas.shift();
+        },
+      });
+
+      await expect(adapter.prompt([{ name: 'respuesta', message: 'foo', type: 'list' }])).resolves.toMatchObject({ respuesta: 'foo' });
+      await expect(adapter.prompt([{ name: 'respuesta', message: 'foo', type: 'list' }])).resolves.toMatchObject({ respuesta: 'bar' });
+      await expect(adapter.prompt([{ name: 'respuesta', message: 'foo', type: 'list' }])).resolves.toMatchObject({ respuesta: undefined });
+    });
   });
   describe('#queue()', () => {
     it('should execute the callback', async () => {
