fleet/frontend/components/queries/QueryResultsTable/QueryResultsTable.tests.jsx

import React from 'react';
import expect, { createSpy, restoreSpies } from 'expect';
import { keys } from 'lodash';
import { mount } from 'enzyme';

import { fillInFormInput } from 'test/helpers';
import QueryResultsTable from 'components/queries/QueryResultsTable';

const host = {
  detail_updated_at: '2016-10-25T16:24:27.679472917-04:00',
  hostname: 'jmeller-mbp.local',
  id: 1,
  ip: '192.168.1.10',
  mac: '10:11:12:13:14:15',
  memory: 4145483776,
  os_version: 'Mac OS X 10.11.6',
  osquery_version: '2.0.0',
  platform: 'darwin',
  status: 'online',
  updated_at: '0001-01-01T00:00:00Z',
  uptime: 3600000000000,
  uuid: '1234-5678-9101',
};
const queryResult = {
  distributed_query_execution_id: 4,
  host,
  rows: [{ host_hostname: 'dfoihgsx', cwd: '/', directory: '/root' }],
};

const campaignWithNoQueryResults = {
  created_at: '0001-01-01T00:00:00Z',
  deleted: false,
  deleted_at: null,
  hosts_count: {
    failed: 0,
    successful: 0,
    total: 0,
  },
  id: 4,
  query_id: 12,
  status: 0,
  totals: {
    count: 3,
    online: 2,
  },
  updated_at: '0001-01-01T00:00:00Z',
  user_id: 1,
};
const campaignWithQueryResults = {
  ...campaignWithNoQueryResults,
  query_results: [
    { host_hostname: 'dfoihgsx', cwd: '/', directory: '/root' },
    { host_hostname: 'abc123', cwd: '/', directory: '/root' },
  ],
  hosts_count: {
    failed: 0,
    successful: 2,
    total: 2,
  },
};

describe('QueryResultsTable - component', () => {
  afterEach(restoreSpies);

  const componentWithoutQueryResults = mount(
    <QueryResultsTable campaign={campaignWithNoQueryResults} />,
  );
  const componentWithQueryResults = mount(
    <QueryResultsTable campaign={campaignWithQueryResults} />,
  );

  it('renders', () => {
    expect(componentWithoutQueryResults.length).toEqual(1);
    expect(componentWithQueryResults.length).toEqual(1);
  });

  it('renders a QueryProgressDetails component if Results is Fullscreen', () => {
    const component = mount(<QueryResultsTable campaign={campaignWithQueryResults} isQueryFullScreen />);
    const QueryProgressDetails = component.find('QueryProgressDetails');

    expect(QueryProgressDetails.length).toEqual(1, 'QueryProgressDetails did not render');
  });

  it('doesn\'t render a QueryProgressDetails component if Results isn\'t Fullscreen', () => {
    const QueryProgressDetails = componentWithQueryResults.find('QueryProgressDetails');

    expect(QueryProgressDetails.length).toEqual(0, 'QueryProgressDetails did not render');
  });

  it('sets the column headers to the keys of the query results', () => {
    const queryResultKeys = keys(queryResult.rows[0]);
    const tableHeaderText = componentWithQueryResults.find('thead').text();

    queryResultKeys.forEach((key) => {
      if (key === 'host_hostname') {
        expect(tableHeaderText).toInclude('hostname');
      } else {
        expect(tableHeaderText).toInclude(key);
      }
    });
  });

  it('filters by hostname', () => {
    const hostnameInputFilter = componentWithQueryResults.find('InputField').find({ name: 'hostname' });

    expect(componentWithQueryResults.find('QueryResultsRow').length).toEqual(2);

    fillInFormInput(hostnameInputFilter, 'abc123');

    expect(componentWithQueryResults.find('QueryResultsRow').length).toEqual(1);
  });

  it('calls the onExportQueryResults prop when the export button is clicked', () => {
    const spy = createSpy();
    const component = mount(<QueryResultsTable campaign={campaignWithQueryResults} onExportQueryResults={spy} />);

    const exportBtn = component.find('.query-results-table__export-btn');

    expect(spy).toNotHaveBeenCalled();

    exportBtn.hostNodes().simulate('click');

    expect(spy).toHaveBeenCalled();
  });
});