jest-setup.js

"use strict";

const _ = require("lodash");
const basicChecks = require("./lib/testUtils/basicChecks");
const less = require("postcss-less");
const stylelint = require("./lib/standalone");

jest.mock("./lib/utils/getOsEol", () => () => "\n");

global.testRule = (rule, schema) => {
  expect.extend({
    toHaveMessage(testCase) {
      if (testCase.message === undefined) {
        return {
          message: () =>
            'Expected "reject" test case to have a "message" property',
          pass: false
        };
      }

      return {
        pass: true
      };
    }
  });

  describe(schema.ruleName, () => {
    const stylelintConfig = {
      rules: {
        [schema.ruleName]: schema.config
      }
    };

    let passingTestCases = schema.accept || [];

    if (!schema.skipBasicChecks) {
      passingTestCases = passingTestCases.concat(basicChecks);
    }

    if (passingTestCases && passingTestCases.length) {
      describe("accept", () => {
        passingTestCases.forEach(testCase => {
          const spec = testCase.only ? it.only : it;

          describe(JSON.stringify(schema.config, replacer), () => {
            describe(JSON.stringify(testCase.code), () => {
              spec(testCase.description || "no description", () => {
                const options = {
                  code: testCase.code,
                  config: stylelintConfig,
                  syntax: schema.syntax
                };

                return stylelint(options).then(output => {
                  expect(output.results[0].warnings).toEqual([]);
                  expect(output.results[0].parseErrors).toEqual([]);

                  if (!schema.fix) return;

                  // Check the fix
                  return stylelint(Object.assign({ fix: true }, options)).then(
                    output => {
                      const fixedCode = getOutputCss(output);

                      expect(fixedCode).toBe(testCase.code);
                    }
                  );
                });
              });
            });
          });
        });
      });
    }

    if (schema.reject && schema.reject.length) {
      describe("reject", () => {
        schema.reject.forEach(testCase => {
          const spec = testCase.only ? it.only : it;

          describe(JSON.stringify(schema.config, replacer), () => {
            describe(JSON.stringify(testCase.code), () => {
              spec(testCase.description || "no description", () => {
                const options = {
                  code: testCase.code,
                  config: stylelintConfig,
                  syntax: schema.syntax
                };

                return stylelint(options).then(output => {
                  const warning = output.results[0].warnings[0];

                  expect(output.results[0].parseErrors).toEqual([]);
                  expect(testCase).toHaveMessage();

                  if (testCase.message !== undefined) {
                    expect(_.get(warning, "text")).toBe(testCase.message);
                  }

                  if (testCase.line !== undefined) {
                    expect(_.get(warning, "line")).toBe(testCase.line);
                  }

                  if (testCase.column !== undefined) {
                    expect(_.get(warning, "column")).toBe(testCase.column);
                  }

                  if (!schema.fix) return;

                  if (
                    !testCase.fixed &&
                    testCase.fixed !== "" &&
                    !testCase.unfixable
                  ) {
                    throw new Error(
                      "If using { fix: true } in test schema, all reject cases must have { fixed: .. }"
                    );
                  }

                  // Check the fix
                  return stylelint(Object.assign({ fix: true }, options))
                    .then(output => {
                      const fixedCode = getOutputCss(output);

                      if (!testCase.unfixable) {
                        expect(fixedCode).toBe(testCase.fixed);
                        expect(fixedCode).not.toBe(testCase.code);
                      } else {
                        // can't fix
                        if (testCase.fixed) {
                          expect(fixedCode).toBe(testCase.fixed);
                        }

                        expect(fixedCode).toBe(testCase.code);
                      }

                      return {
                        fixedCode,
                        warnings: output.results[0].warnings
                      };
                    })
                    .then(({ fixedCode, warnings }) => {
                      // Checks whether only errors other than those fixed are reported.
                      return stylelint({
                        code: fixedCode,
                        config: stylelintConfig,
                        syntax: schema.syntax
                      }).then(output => ({
                        output,
                        warnings
                      }));
                    })
                    .then(({ output, warnings }) => {
                      expect(output.results[0].warnings).toEqual(warnings);
                      expect(output.results[0].parseErrors).toEqual([]);
                    });
                });
              });
            });
          });
        });
      });
    }
  });
};

function getOutputCss(output) {
  const result = output.results[0]._postcssResult;
  const css = result.root.toString(result.opts.syntax);

  if (result.opts.syntax === less) {
    // Less needs us to manually strip whitespace at the end of single-line comments ¯\_(ツ)_/¯
    return css.replace(/(\n?\s*\/\/.*?)[ \t]*(\r?\n)/g, "$1$2");
  }

  return css;
}

function replacer(key, value) {
  return value instanceof RegExp ? `[RegExp] ${value.toString()}` : value;
}