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

import React from 'react';
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',
  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', () => {
  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).toContain('hostname');
      } else {
        expect(tableHeaderText).toContain(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 = jest.fn();
      const component = mount(<QueryResultsTable campaign={campaignWithQueryResults} onExportQueryResults={spy} />);

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

      expect(spy).not.toHaveBeenCalled();

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

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