react-apollo-testing-mock-gen/src/index.js

Mock generator for testing with MockedProvider.

const fs = require('fs');

const config = require('./config');

const { exec } = require('child_process');
const { mockTemplate } = require('./templates/mock-template');
const { getFileSource } = require('./file-parser');
const { capitalize } = require('./utils');
const buildFragmentStuctures = require('./fragment-parser');
const { filePathName, fileFolderPath, fileName, mockFilePath } = require('./path-parser');
const { buildSelectionStructure, getVariables } = require('./query-parser');
const { resolveSelectionValues } = require('./type-resolver');

const getOperationDefinition = (sourceJson) => {
  return sourceJson.definitions.find((def) => def.kind === 'OperationDefinition');
}

const generateMock = () => {
  const { sourceJson, content } = getFileSource(filePathName);
  const fragments = buildFragmentStuctures(content, sourceJson);
  const operationDefinition = getOperationDefinition(sourceJson);
  const variables = getVariables(operationDefinition.variableDefinitions);
  const requestObjectStructure = buildSelectionStructure(operationDefinition.selectionSet, fragments);
  const operationName = capitalize(operationDefinition.operation)
  const mockResponse = resolveSelectionValues(operationName, requestObjectStructure);
  const template = mockTemplate(fileName.replace('.gql', ''), variables, mockResponse, fileName);

  if (!fs.existsSync(`${fileFolderPath}/mocks`)) {
    try {
      fs.mkdirSync(`${fileFolderPath}/mocks`);
    } catch (e) {
      console.log('❌  Could not create mock folder.');
    }
  }

  fs.writeFile(mockFilePath, template, (error) => {
    if (error) {
      console.log('❌  Could not create mock file.');

      process.exit(1);
    }

    console.log('File created successfully. 🎉')
  });

  if (config.lint) {
    exec(config.lint(mockFilePath), (err) => {
      if (err) {
        console.log('❌  Could not execute lint command.');
        
        process.exit(1);
      }

      console.log('File has been formatted successfully. 💅');
    });
  }
}

module.exports = generateMock