Mailspring/app/internal_packages/open-tracking/specs/open-tracking-icon-spec.jsx

import React from 'react';
import ReactDOM from 'react-dom';
import { findRenderedDOMComponentWithClass } from 'react-dom/test-utils';

import { Message, MailspringTestUtils } from 'mailspring-exports';
import OpenTrackingIcon from '../lib/open-tracking-icon';
import { PLUGIN_ID } from '../lib/open-tracking-constants';

const { renderIntoDocument } = MailspringTestUtils;

function makeIcon(thread, props = {}) {
  return renderIntoDocument(<OpenTrackingIcon {...props} thread={thread} />);
}

function find(component, className) {
  return ReactDOM.findDOMNode(findRenderedDOMComponentWithClass(component, className));
}

function addOpenMetadata(obj, openCount) {
  obj.directlyAttachMetadata(PLUGIN_ID, { open_count: openCount });
}

describe('Open tracking icon', function openTrackingIcon() {
  beforeEach(() => {
    this.thread = { __messages: [] };
  });

  it('shows no icon if the thread has no messages', () => {
    const icon = ReactDOM.findDOMNode(makeIcon(this.thread));
    expect(icon.children.length).toEqual(0);
  });

  it('shows no icon if the thread messages have no metadata', () => {
    this.thread.__messages.push(new Message());
    this.thread.__messages.push(new Message());
    const icon = ReactDOM.findDOMNode(makeIcon(this.thread));
    expect(icon.children.length).toEqual(0);
  });

  describe('With messages and metadata', () => {
    beforeEach(() => {
      this.messages = [new Message(), new Message(), new Message({ draft: true })];
      this.thread.__messages.push(...this.messages);
    });

    it('shows no icon if metadata is malformed', () => {
      this.messages[0].directlyAttachMetadata(PLUGIN_ID, { gar: 'bage' });
      const icon = ReactDOM.findDOMNode(makeIcon(this.thread));
      expect(icon.children.length).toEqual(0);
    });

    it('shows an unopened icon if last non draft message has metadata and is unopened', () => {
      addOpenMetadata(this.messages[0], 1);
      addOpenMetadata(this.messages[1], 0);
      const icon = find(makeIcon(this.thread), 'open-tracking-icon');
      expect(icon.children.length).toEqual(1);
      expect(icon.querySelector('img.unopened')).not.toBeNull();
      expect(icon.querySelector('img.opened')).toBeNull();
    });

    it('shows an opened icon if last non draft message with metadata is opened', () => {
      addOpenMetadata(this.messages[0], 0);
      addOpenMetadata(this.messages[1], 1);
      const icon = find(makeIcon(this.thread), 'open-tracking-icon');
      expect(icon.children.length).toEqual(1);
      expect(icon.querySelector('img.unopened')).toBeNull();
      expect(icon.querySelector('img.opened')).not.toBeNull();
    });
  });
});