kibana/x-pack/plugins/apm/server/lib/helpers/setup_request.test.ts

/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License;
 * you may not use this file except in compliance with the Elastic License.
 */

import { setupRequest } from './setup_request';

describe('setupRequest', () => {
  let callWithRequestSpy: jest.Mock;
  let mockReq: any;

  beforeEach(() => {
    callWithRequestSpy = jest.fn();
    mockReq = {
      params: {},
      query: {},
      server: {
        config: () => 'myConfig',
        plugins: {
          elasticsearch: {
            getCluster: () => ({ callWithRequest: callWithRequestSpy })
          }
        }
      },
      getUiSettingsService: jest.fn(() => ({
        get: jest.fn(() => Promise.resolve(false))
      }))
    };
  });

  it('should call callWithRequest with correct args', async () => {
    const setup = setupRequest(mockReq);
    await setup.client('myType', { body: { foo: 'bar' } });
    expect(callWithRequestSpy).toHaveBeenCalledWith(mockReq, 'myType', {
      body: {
        foo: 'bar',
        query: {
          bool: {
            filter: [{ range: { 'observer.version_major': { gte: 7 } } }]
          }
        }
      },
      ignore_throttled: true,
      rest_total_hits_as_int: true
    });
  });

  it('should set ignore_throttled to false if includeFrozen is true', async () => {
    // mock includeFrozen to return true
    mockReq.getUiSettingsService.mockImplementation(() => ({
      get: jest.fn(() => Promise.resolve(true))
    }));
    const setup = setupRequest(mockReq);
    await setup.client('myType', {});
    const params = callWithRequestSpy.mock.calls[0][2];
    expect(params.ignore_throttled).toBe(false);
  });

  it('should set filter if none exists', async () => {
    const setup = setupRequest(mockReq);
    await setup.client('myType', {});
    const params = callWithRequestSpy.mock.calls[0][2];
    expect(params.body).toEqual({
      query: {
        bool: { filter: [{ range: { 'observer.version_major': { gte: 7 } } }] }
      }
    });
  });

  it('should merge filters if one exists', async () => {
    const setup = setupRequest(mockReq);
    await setup.client('myType', {
      body: {
        query: { bool: { filter: [{ term: 'someTerm' }] } }
      }
    });
    const params = callWithRequestSpy.mock.calls[0][2];
    expect(params.body).toEqual({
      query: {
        bool: {
          filter: [
            { term: 'someTerm' },
            { range: { 'observer.version_major': { gte: 7 } } }
          ]
        }
      }
    });
  });
});