fleet/frontend/kolide/helpers.tests.js

import { omit } from 'lodash';

import { configStub, scheduledQueryStub } from 'test/stubs';
import helpers from 'kolide/helpers';

const label1 = { id: 1, target_type: 'labels' };
const label2 = { id: 2, target_type: 'labels' };
const host1 = { id: 6, target_type: 'hosts' };
const host2 = { id: 5, target_type: 'hosts' };

describe('Kolide API - helpers', () => {
  describe('#labelSlug', () => {
    it('creates a slug for the label', () => {
      expect(helpers.labelSlug({ name: 'All Hosts' })).toEqual('all-hosts');
      expect(helpers.labelSlug({ id: 7 })).toEqual('labels/7');
    });
  });

  describe('#formatConfigDataForServer', () => {
    const { formatConfigDataForServer } = helpers;
    const config = {
      org_name: 'Kolide',
      org_logo_url: '0.0.0.0:8080/logo.png',
      kolide_server_url: '',
      configured: false,
      domain: '',
      smtp_enabled: true,
      sender_address: '',
      server: '',
      port: 587,
      authentication_type: 'authtype_username_password',
      user_name: '',
      password: '',
      enable_ssl_tls: true,
      authentication_method: 'authmethod_plain',
      verify_ssl_certs: true,
      enable_start_tls: true,
      host_expiry_enabled: false,
      host_expiry_window: 0,
      live_query_disabled: false,
    };

    it('splits config into categories for the server', () => {
      expect(formatConfigDataForServer(config)).toEqual(omit(configStub, ['smtp_settings.configured']));
      expect(formatConfigDataForServer({ org_name: 'The Gnar Co' })).toEqual({
        org_info: { org_name: 'The Gnar Co' },
      });
      expect(
        formatConfigDataForServer({ org_name: 'The Gnar Co', kolide_server_url: 'https://example.com' }),
      ).toEqual({
        org_info: { org_name: 'The Gnar Co' },
        server_settings: { kolide_server_url: 'https://example.com' },
      });
      expect(
        formatConfigDataForServer({ domain: 'https://kolide.co' }),
      ).toEqual({
        smtp_settings: { domain: 'https://kolide.co' },
      });
      expect(
        formatConfigDataForServer({ host_expiry_window: '12' }),
      ).toEqual({
        host_expiry_settings: {
          host_expiry_window: 12,
        },
      });
    });
  });

  describe('#formatScheduledQueryForServer', () => {
    const { formatScheduledQueryForServer } = helpers;
    const scheduledQuery = {
      ...scheduledQueryStub,
      logging_type: 'snapshot',
      pack_id: '3',
      platform: 'all',
      query_id: '1',
      shard: '12',
    };

    it('sets the correct attributes for the server', () => {
      expect(formatScheduledQueryForServer(scheduledQuery)).toEqual({
        ...scheduledQueryStub,
        pack_id: 3,
        platform: '',
        query_id: 1,
        shard: 12,
        snapshot: true,
        removed: false,
      });
    });
  });

  describe('#formatScheduledQueryForClient', () => {
    const { formatScheduledQueryForClient } = helpers;
    const scheduledQuery = {
      ...scheduledQueryStub,
      platform: '',
      snapshot: true,
    };

    it('sets the correct attributes for the server', () => {
      expect(formatScheduledQueryForClient(scheduledQuery)).toEqual({
        ...scheduledQueryStub,
        logging_type: 'snapshot',
        platform: 'all',
      });
    });

    it('sets the logging_type attribute', () => {
      expect(formatScheduledQueryForClient({ ...scheduledQueryStub, removed: true, snapshot: false })).toEqual({
        ...scheduledQueryStub,
        logging_type: 'differential',
        removed: true,
        snapshot: false,
      });

      expect(formatScheduledQueryForClient({ ...scheduledQueryStub, snapshot: true })).toEqual({
        ...scheduledQueryStub,
        logging_type: 'snapshot',
        snapshot: true,
      });

      expect(formatScheduledQueryForClient({ ...scheduledQueryStub, snapshot: false, removed: false })).toEqual({
        ...scheduledQueryStub,
        logging_type: 'differential_ignore_removals',
        removed: false,
        snapshot: false,
      });
    });
  });

  describe('#formatSelectedTargetsForApi', () => {
    const { formatSelectedTargetsForApi } = helpers;

    it('splits targets into labels and hosts', () => {
      const targets = [host1, host2, label1, label2];

      expect(formatSelectedTargetsForApi(targets)).toEqual({
        hosts: [6, 5],
        labels: [1, 2],
      });
    });

    it('appends `_id` when appendID is specified', () => {
      const targets = [host1, host2, label1, label2];

      expect(formatSelectedTargetsForApi(targets, true)).toEqual({
        host_ids: [6, 5],
        label_ids: [1, 2],
      });
    });
  });

  describe('#setupData', () => {
    const formData = {
      email: 'hi@gnar.dog',
      name: 'Gnar Dog',
      kolide_server_url: 'https://gnar.kolide.co',
      org_logo_url: 'https://thegnar.co/assets/logo.png',
      org_name: 'The Gnar Co.',
      password: 'p@ssw0rd',
      password_confirmation: 'p@ssw0rd',
      username: 'gnardog',
    };

    it('formats the form data to send to the server', () => {
      expect(helpers.setupData(formData)).toEqual({
        kolide_server_url: 'https://gnar.kolide.co',
        org_info: {
          org_logo_url: 'https://thegnar.co/assets/logo.png',
          org_name: 'The Gnar Co.',
        },
        admin: {
          admin: true,
          email: 'hi@gnar.dog',
          name: 'Gnar Dog',
          password: 'p@ssw0rd',
          password_confirmation: 'p@ssw0rd',
          username: 'gnardog',
        },
      });
    });
  });
});