es6-object-short-notation-visitors-test.js

/**
 * Copyright 2013 Facebook, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/**
 * @emails [email protected]
 */

/*jshint evil:true*/

require('mock-modules').autoMockOff();

describe('es6-object-short-notation-visitors', function() {
  var transformFn;

  var destructuringVisitors;
  var shortObjectsVisitors;

  var visitors;

  beforeEach(function() {
    require('mock-modules').dumpCache();
    transformFn = require('../../src/jstransform').transform;

    shortObjectsVisitors = require('../es6-object-short-notation-visitors').visitorList;
    destructuringVisitors = require('../es6-destructuring-visitors').visitorList;

    visitors = shortObjectsVisitors.concat(destructuringVisitors);
  });

  function transform(code) {
    return transformFn(visitors, code).code;
  }

  function expectTransform(code, result) {
    expect(transform(code)).toEqual(result);
  }

  // Functional tests.

  it('should transform short notation and return 5', function() {
    var code = transform([
      '(function(x, y) {',
      '  var data = {x, y};',
      '  return data.x + data.y;',
      '})(2, 3);'
    ].join('\n'));

    expect(eval(code)).toEqual(5);
  });

  it('should transform work with destructuring and return 10', function() {
    var code = transform([
      'var x = 5, y = 5;',
      '(function({x, y}) {',
      '  var data = {x, y};',
      '  return data.x + data.y;',
      '})({x, y});'
    ].join('\n'));

    expect(eval(code)).toEqual(10);
  });

  // Source code tests.
  it('should transform simple short notation', function() {

    // Should transform simple short notation.
    expectTransform(
      'function foo(x, y) { return {x, y}; }',
      'function foo(x, y) { return {x:x, y:y}; }'
    );

    // Should preserve lines transforming ugly code.
    expectTransform([
      'function',
      '',
      'foo    (',
      '    x,',
      '          y',
      '',
      ')',
      '',
      '        {',
      ' return         {',
      '          x,',
      '  y};',
      '}'
    ].join('\n'), [
      'function',
      '',
      'foo    (',
      '    x,',
      '          y',
      '',
      ')',
      '',
      '        {',
      ' return         {',
      '          x:x,',
      '  y:y};',
      '}'
    ].join('\n'));
  });

});