comment-positioning-container.test.js

import React from 'react';
import {shallow} from 'enzyme';

import CommentPositioningContainer from '../../lib/containers/comment-positioning-container';
import File from '../../lib/models/patch/file';
import ObserveModel from '../../lib/views/observe-model';
import {cloneRepository, buildRepository} from '../helpers';
import {multiFilePatchBuilder} from '../builder/patch';

describe('CommentPositioningContainer', function() {
  let atomEnv, localRepository;

  beforeEach(async function() {
    atomEnv = global.buildAtomEnvironment();
    localRepository = await buildRepository(await cloneRepository());
  });

  afterEach(function() {
    atomEnv.destroy();
  });

  function buildApp(override = {}) {
    const props = {
      localRepository,
      multiFilePatch: multiFilePatchBuilder().build().multiFilePatch,
      commentThreads: [],
      prCommitSha: '0000000000000000000000000000000000000000',
      children: () => <div />,
      translateLinesGivenDiff: () => {},
      diffPositionToFilePosition: () => {},
      ...override,
    };

    return <CommentPositioningContainer {...props} />;
  }

  it('renders its child with null while loading positions', function() {
    const children = sinon.stub().returns(<div className="done" />);
    const wrapper = shallow(buildApp({children}))
      .find(ObserveModel).renderProp('children')(null);

    assert.isTrue(wrapper.exists('.done'));
    assert.isTrue(children.calledWith(null));
  });

  it('renders its child with a map of translated positions', function() {
    const children = sinon.stub().returns(<div className="done" />);

    const {multiFilePatch} = multiFilePatchBuilder()
      .addFilePatch(fp => {
        fp.setOldFile(f => f.path('file0.txt'));
      })
      .addFilePatch(fp => {
        fp.setOldFile(f => f.path('file1.txt'));
      })
      .build({preserveOriginal: true});

    const commentThreads = [
      {comments: [{path: 'file0.txt', position: 1}, {path: 'ignored.txt', position: 999}]},
      {comments: [{path: 'file0.txt', position: 5}]},
      {comments: [{path: 'file1.txt', position: 11}]},
    ];

    const diffPositionToFilePosition = (rawPositions, patch) => {
      if (patch.oldPath === 'file0.txt') {
        assert.sameMembers(Array.from(rawPositions), [1, 5]);
        return new Map([
          [1, 10],
          [5, 50],
        ]);
      } else if (patch.oldPath === 'file1.txt') {
        assert.sameMembers(Array.from(rawPositions), [11]);
        return new Map([
          [11, 16],
        ]);
      } else {
        throw new Error(`Unexpected patch: ${patch.oldPath}`);
      }
    };

    const wrapper = shallow(buildApp({children, multiFilePatch, commentThreads, diffPositionToFilePosition}))
      .find(ObserveModel).renderProp('children')({});

    assert.isTrue(wrapper.exists('.done'));
    const [translations] = children.lastCall.args;

    const file0 = translations.get('file0.txt');
    assert.sameDeepMembers(Array.from(file0.diffToFilePosition), [[1, 10], [5, 50]]);
    assert.isNull(file0.fileTranslations);

    const file1 = translations.get('file1.txt');
    assert.sameDeepMembers(Array.from(file1.diffToFilePosition), [[11, 16]]);
    assert.isNull(file1.fileTranslations);
  });

  it('uses a single content-change diff if one is available', function() {
    const children = sinon.stub().returns(<div className="done" />);

    const {multiFilePatch} = multiFilePatchBuilder()
      .addFilePatch(fp => {
        fp.setOldFile(f => f.path('file0.txt'));
      })
      .build({preserveOriginal: true});

    const commentThreads = [
      {comments: [{path: 'file0.txt', position: 1}]},
    ];

    const diffPositionToFilePosition = (rawPositions, patch) => {
      assert.strictEqual(patch.oldPath, 'file0.txt');
      assert.sameMembers(Array.from(rawPositions), [1]);
      return new Map([[1, 2]]);
    };

    const translateLinesGivenDiff = (translatedRows, diff) => {
      assert.sameMembers(Array.from(translatedRows), [2]);
      assert.strictEqual(diff, DIFF);
      return new Map([[2, 4]]);
    };

    const DIFF = Symbol('diff payload');

    const wrapper = shallow(buildApp({
      children,
      multiFilePatch,
      commentThreads,
      prCommitSha: '1111111111111111111111111111111111111111',
      diffPositionToFilePosition,
      translateLinesGivenDiff,
    })).find(ObserveModel).renderProp('children')({'file0.txt': [DIFF]});

    assert.isTrue(wrapper.exists('.done'));

    const [translations] = children.lastCall.args;

    const file0 = translations.get('file0.txt');
    assert.sameDeepMembers(Array.from(file0.diffToFilePosition), [[1, 2]]);
    assert.sameDeepMembers(Array.from(file0.fileTranslations), [[2, 4]]);
  });

  it('identifies the content change diff when two diffs are present', function() {
    const children = sinon.stub().returns(<div className="done" />);

    const {multiFilePatch} = multiFilePatchBuilder()
      .addFilePatch(fp => {
        fp.setOldFile(f => f.path('file0.txt'));
      })
      .build({preserveOriginal: true});

    const commentThreads = [
      {comments: [{path: 'file0.txt', position: 1}]},
    ];

    const diffs = [
      {oldPath: 'file0.txt', oldMode: File.modes.SYMLINK},
      {oldPath: 'file0.txt', oldMode: File.modes.NORMAL},
    ];

    const diffPositionToFilePosition = (rawPositions, patch) => {
      assert.strictEqual(patch.oldPath, 'file0.txt');
      assert.sameMembers(Array.from(rawPositions), [1]);
      return new Map([[1, 2]]);
    };

    const translateLinesGivenDiff = (translatedRows, diff) => {
      assert.sameMembers(Array.from(translatedRows), [2]);
      assert.strictEqual(diff, diffs[1]);
      return new Map([[2, 4]]);
    };

    const wrapper = shallow(buildApp({
      children,
      multiFilePatch,
      commentThreads,
      prCommitSha: '1111111111111111111111111111111111111111',
      diffPositionToFilePosition,
      translateLinesGivenDiff,
    })).find(ObserveModel).renderProp('children')({'file0.txt': diffs});

    assert.isTrue(wrapper.exists('.done'));

    const [translations] = children.lastCall.args;

    const file0 = translations.get('file0.txt');
    assert.sameDeepMembers(Array.from(file0.diffToFilePosition), [[1, 2]]);
    assert.sameDeepMembers(Array.from(file0.fileTranslations), [[2, 4]]);
  });

  it("finds the content diff if it's the first one", function() {
    const children = sinon.stub().returns(<div className="done" />);

    const {multiFilePatch} = multiFilePatchBuilder()
      .addFilePatch(fp => {
        fp.setOldFile(f => f.path('file0.txt'));
      })
      .build({preserveOriginal: true});

    const commentThreads = [
      {comments: [{path: 'file0.txt', position: 1}]},
    ];

    const diffs = [
      {oldPath: 'file0.txt', oldMode: File.modes.NORMAL},
      {oldPath: 'file0.txt', oldMode: File.modes.SYMLINK},
    ];

    const diffPositionToFilePosition = (rawPositions, patch) => {
      assert.strictEqual(patch.oldPath, 'file0.txt');
      assert.sameMembers(Array.from(rawPositions), [1]);
      return new Map([[1, 2]]);
    };

    const translateLinesGivenDiff = (translatedRows, diff) => {
      assert.sameMembers(Array.from(translatedRows), [2]);
      assert.strictEqual(diff, diffs[0]);
      return new Map([[2, 4]]);
    };

    const wrapper = shallow(buildApp({
      children,
      multiFilePatch,
      commentThreads,
      prCommitSha: '1111111111111111111111111111111111111111',
      diffPositionToFilePosition,
      translateLinesGivenDiff,
    })).find(ObserveModel).renderProp('children')({'file0.txt': diffs});

    assert.isTrue(wrapper.exists('.done'));

    const [translations] = children.lastCall.args;

    const file0 = translations.get('file0.txt');
    assert.sameDeepMembers(Array.from(file0.diffToFilePosition), [[1, 2]]);
    assert.sameDeepMembers(Array.from(file0.fileTranslations), [[2, 4]]);
  });
});