Add task for test dependency

Author: yashpandit

generator.js

@@ -9,6 +9,7 @@ module.exports = app => {
 
   app.use(require('generate-project'));
   app.register('create-react-app', require('./lib/generators/create-react-app'));
+  app.register('custom-test', require('./lib/generators/custom-tests'));
 
   /**
    * Generate a `index.js` file to the current working directory. Learn how to [customize

lib/generators/custom-tests.js

@@ -0,0 +1,24 @@
+const spawn = require('../spawn');
+const path = require('path');
+const { prompt } = require('enquirer');
+const { YARN, ENZYME, JEST, ADD } = require('../constants');
+
+module.exports = app => {
+  app.task('custom-test', async () => {
+    const answers = prompt({
+      type: 'select',
+      name: 'test',
+      message:
+        'React provides a built-in testing suite. Would you like to add any of the following testing frameworks?',
+      choices: ['Enzyme', 'Jest', 'No']
+    });
+    const { test } = await answers;
+    if (test !== 'No' && test === 'Enzyme') {
+      await spawn(YARN, [ADD, ENZYME]);
+    } else if (test !== 'No' && test === 'Jest') {
+      await spawn(YARN, [ADD, JEST]);
+    }
+  });
+
+  app.task('default', ['custom-test']);
+};