metamask-extension/app/scripts/lib/segment/analytics.test.js at 894bd090f9d44abb3a068082627a2f18d22c1eb7 · MetaMask/metamask-extension · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
import Analytics from './analytics';

const DUMMY_KEY = 'DUMMY_KEY';
const DUMMY_MESSAGE = {
  userId: 'userId',
  idValue: 'idValue',
  event: 'event',
};
const FLUSH_INTERVAL = 10000;

global.setImmediate = (arg) => {
  arg();
};

global.fetch = () =>
  Promise.resolve({
    ok: true,
    json: () => Promise.resolve({ success: true }),
  });

describe('Analytics', function () {
  let analytics;

  beforeEach(() => {
    analytics = new Analytics(DUMMY_KEY);
  });

  describe('#flush', function () {
    it('first message is immediately flushed', function () {
      const mock = jest.fn(analytics.flush);
      analytics.flush = mock;
      analytics.track(DUMMY_MESSAGE);
      expect(analytics.queue).toHaveLength(0);
      expect(mock).toHaveBeenCalledTimes(1);
    });

    it('after first message it is called when queue size equals flushAt value', function () {
      analytics = new Analytics(DUMMY_KEY, { flushAt: 3 });
      const mock = jest.fn(analytics.flush);
      analytics.flush = mock;
      analytics.track(DUMMY_MESSAGE);
      analytics.track(DUMMY_MESSAGE);
      analytics.track(DUMMY_MESSAGE);
      analytics.track(DUMMY_MESSAGE);
      expect(analytics.queue).toHaveLength(0);
      expect(mock).toHaveBeenCalledTimes(2);
    });

    it('except for first message it is called until queue size is less than flushAt value', function () {
      analytics = new Analytics(DUMMY_KEY, { flushAt: 3 });
      const mock = jest.fn(analytics.flush);
      analytics.flush = mock;
      analytics.track(DUMMY_MESSAGE);
      analytics.track(DUMMY_MESSAGE);
      analytics.track(DUMMY_MESSAGE);
      expect(analytics.queue).toHaveLength(2);
      expect(mock).toHaveBeenCalledTimes(1);
    });

    it('after first message it is called after flushInterval is elapsed', function () {
      jest.useFakeTimers();
      analytics = new Analytics(DUMMY_KEY, { flushInterval: FLUSH_INTERVAL });
      const mock = jest.fn(analytics.flush);
      analytics.flush = mock;
      analytics.track(DUMMY_MESSAGE);
      analytics.track(DUMMY_MESSAGE);
      jest.advanceTimersByTime(FLUSH_INTERVAL);
      expect(analytics.queue).toHaveLength(0);
      expect(mock).toHaveBeenCalledTimes(2);
    });

    it('after first message it is not called until flushInterval is elapsed', function () {
      jest.useFakeTimers();
      analytics = new Analytics(DUMMY_KEY, { flushInterval: FLUSH_INTERVAL });
      const mock = jest.fn(analytics.flush);
      analytics.flush = mock;
      analytics.track(DUMMY_MESSAGE);
      analytics.track(DUMMY_MESSAGE);
      jest.advanceTimersByTime(FLUSH_INTERVAL - 100);
      expect(analytics.queue).toHaveLength(1);
      expect(mock).toHaveBeenCalledTimes(1);
    });

    it('invokes callbacks', async function () {
      const callback = jest.fn();
      analytics.track(DUMMY_MESSAGE);
      analytics.track(DUMMY_MESSAGE, callback);
      await analytics.flush();
      expect(callback).toHaveBeenCalledTimes(1);
    });
  });

  describe('#track', function () {
    it('adds messages to ququq', function () {
      analytics.track(DUMMY_MESSAGE);
      analytics.track(DUMMY_MESSAGE);
      expect(analytics.queue).toHaveLength(1);
    });
  });

  describe('#page', function () {
    it('adds messages to ququq', function () {
      analytics.page(DUMMY_MESSAGE);
      analytics.page(DUMMY_MESSAGE);
      expect(analytics.queue).toHaveLength(1);
    });
  });

  describe('#identify', function () {
    it('adds messages to ququq', function () {
      analytics.identify(DUMMY_MESSAGE);
      analytics.identify(DUMMY_MESSAGE);
      expect(analytics.queue).toHaveLength(1);
    });
  });
});