propTypeCompositionHandler-test.js

/*
 *  Copyright (c) 2015, Facebook, Inc.
 *  All rights reserved.
 *
 *  This source code is licensed under the BSD-style license found in the
 *  LICENSE file in the root directory of this source tree. An additional grant
 *  of patent rights can be found in the PATENTS file in the same directory.
 *
 */

/*global jest, describe, it, expect, beforeEach*/

jest.disableAutomock();
jest.mock('../../Documentation');

describe('propTypeCompositionHandler', () => {
  var statement, expression;
  var getPropTypeMock;
  var documentation;
  var propTypeCompositionHandler;

  beforeEach(() => {
    ({statement, expression} = require('../../../tests/utils'));
    getPropTypeMock = jest.genMockFunction().mockImplementation(() => ({}));
    jest.setMock('../../utils/getPropType', getPropTypeMock);
    jest.mock('../../utils/getPropType');

    documentation = new (require('../../Documentation'))();
    propTypeCompositionHandler = require('../propTypeCompositionHandler').default;
  });

  function test(getSrc, parse) {
    it('understands assignment from module', () => {
      var definition = parse(`
        ${getSrc('Foo.propTypes')}
        var Foo = require("Foo.react");
      `);

      propTypeCompositionHandler(documentation, definition);
      expect(documentation.composes).toEqual(['Foo.react']);

      documentation = new (require('../../Documentation'))();
      definition = parse(`
        ${getSrc('SharedProps')}
        var SharedProps = require("SharedProps");
      `);

      propTypeCompositionHandler(documentation, definition);
      expect(documentation.composes).toEqual(['SharedProps']);
    });

    it('understands the spread operator', () => {
      var definitionSrc = getSrc(`
        {
          ...Foo.propTypes,
          ...SharedProps,
        }
      `);
      var definition = parse(`
        ${definitionSrc}
        var Foo = require("Foo.react");
        var SharedProps = require("SharedProps");
      `);

      propTypeCompositionHandler(documentation, definition);
      expect(documentation.composes).toEqual(['Foo.react', 'SharedProps']);
    });

  }

  describe('React.createClass', () => {
    test(
      propTypesSrc => `({propTypes: ${propTypesSrc}})`,
      src => statement(src).get('expression')
    );
  });

  describe('class definition', () => {
    test(
      propTypesSrc => `
        class Component {
          static propTypes = ${propTypesSrc};
        }
      `,
      src => statement(src)
    );
  });

  it('does not error if propTypes cannot be found', () => {
    var definition = expression('{fooBar: 42}');
    expect(() => propTypeCompositionHandler(documentation, definition))
      .not.toThrow();

    definition = statement('class Foo {}');
    expect(() => propTypeCompositionHandler(documentation, definition))
      .not.toThrow();
  });
});