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

import {
  LensReportManager,
  setReportManager,
  stopReportManager,
  trackUiEvent,
  trackSuggestionEvent,
} from './factory';
import { coreMock } from 'src/core/public/mocks';
import { HttpSetup } from 'kibana/public';
import { IStorageWrapper } from 'src/plugins/kibana_utils/public';

jest.useFakeTimers();

const createMockStorage = () => {
  let lastData = { events: {}, suggestionEvents: {} };
  return {
    get: jest.fn().mockImplementation(() => lastData),
    set: jest.fn().mockImplementation((key, value) => {
      lastData = value;
    }),
    remove: jest.fn(),
    clear: jest.fn(),
  };
};

describe('Lens UI telemetry', () => {
  let storage: jest.Mocked<IStorageWrapper>;
  let http: jest.Mocked<HttpSetup>;
  let dateSpy: jest.SpyInstance;

  beforeEach(() => {
    dateSpy = jest
      .spyOn(Date, 'now')
      .mockImplementation(() => new Date(Date.UTC(2019, 9, 23)).valueOf());

    storage = createMockStorage();
    http = coreMock.createSetup().http;
    http.post.mockClear();
    const fakeManager = new LensReportManager({
      http,
      storage,
    });
    setReportManager(fakeManager);
  });

  afterEach(() => {
    stopReportManager();
    dateSpy.mockRestore();
  });

  it('should write immediately and track local state', () => {
    trackUiEvent('loaded');

    expect(storage.set).toHaveBeenCalledWith('lens-ui-telemetry', {
      events: expect.any(Object),
      suggestionEvents: {},
    });

    trackSuggestionEvent('reload');

    expect(storage.set).toHaveBeenLastCalledWith('lens-ui-telemetry', {
      events: expect.any(Object),
      suggestionEvents: expect.any(Object),
    });
  });

  it('should post the results after waiting 10 seconds, if there is data', async () => {
    jest.runOnlyPendingTimers();

    http.post.mockResolvedValue({});

    expect(http.post).not.toHaveBeenCalled();
    expect(storage.set).toHaveBeenCalledTimes(0);

    trackUiEvent('load');
    expect(storage.set).toHaveBeenCalledTimes(1);

    jest.runOnlyPendingTimers();

    expect(http.post).toHaveBeenCalledWith(`/api/lens/stats`, {
      body: JSON.stringify({
        events: {
          '2019-10-23': {
            load: 1,
          },
        },
        suggestionEvents: {},
      }),
    });
  });

  it('should keep its local state after an http error', () => {
    http.post.mockRejectedValue('http error');

    trackUiEvent('load');
    expect(storage.set).toHaveBeenCalledTimes(1);

    jest.runOnlyPendingTimers();

    expect(http.post).toHaveBeenCalled();
    expect(storage.set).toHaveBeenCalledTimes(1);
  });
});