Fix function context at prompt (#1265)

yeoman · Jan 29, 2021 · cd1b557 · cd1b557
1 parent 337dddf
commit cd1b557
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/lib/index.js b/lib/index.js
@@ -525,7 +525,7 @@ class Generator extends EventEmitter {
 
     const storageForQuestion = {};
 
-    const getAnswerFromStorage = function(question) {
+    const getAnswerFromStorage = question => {
       let questionStorage = question.storage || storage;
       questionStorage =
         typeof questionStorage === 'string' ? this[questionStorage] : questionStorage;

diff --git a/test/base.js b/test/base.js
@@ -1786,6 +1786,17 @@ describe('Base', () => {
         });
     });
 
+    it('saves answers to config when specified as a property name', function() {
+      return this.dummy
+        .prompt([{ ...input1Prompt, storage: 'config' }, input2Prompt])
+        .then(answers => {
+          assert.equal(answers.prompt1, 'prompt1NewValue');
+          assert.equal(answers.prompt2, 'prompt2NewValue');
+          assert.equal(this.dummy.config.get('prompt1'), 'prompt1NewValue');
+          assert.equal(this.dummy.config.get('prompt2'), 'prompt2Value');
+        });
+    });
+
     it('saves answers to specific storage', function() {
       return this.dummy
         .prompt([{ ...input1Prompt, storage: this.dummy.config }, input2Prompt])