snyk-request-manager/test/lib/requestManager/retries.test.ts at 211169c08cd801a768c8a99c46030651725de6b1 · snyk-labs/snyk-request-manager · 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
import { requestsManager } from '../../../src/lib/request/requestManager';
import * as fs from 'fs';
import * as nock from 'nock';
import * as path from 'path';
import { RequestsManagerApiError } from '../../../src/lib/customErrors/requestManagerErrors';

const fixturesFolderPath = path.resolve(__dirname, '../..') + '/fixtures/';

const requestManager = new requestsManager();

describe('Testing Request Retries', () => {
  it('Retry on 500 - success after 1 retry', async () => {
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(500, '500');
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(200, () => {
        return fs.readFileSync(
          fixturesFolderPath + 'apiResponses/general-doc.json',
        );
      });

    try {
      const response = await requestManager.request({
        verb: 'POST',
        url: '/apierror',
      });
      expect(response.data).toEqual(
        JSON.parse(
          fs
            .readFileSync(fixturesFolderPath + 'apiResponses/general-doc.json')
            .toString(),
        ),
      );
    } catch (err) {
      console.log(err);
    }
  });

  it('Retry on 500 - success after 4 retries', async () => {
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(500, '500');
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(500, '500');
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(500, '500');
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(500, '500');
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(200, () => {
        return fs.readFileSync(
          fixturesFolderPath + 'apiResponses/general-doc.json',
        );
      });

    try {
      const response = await requestManager.request({
        verb: 'POST',
        url: '/apierror',
      });
      expect(response.data).toEqual(
        JSON.parse(
          fs
            .readFileSync(fixturesFolderPath + 'apiResponses/general-doc.json')
            .toString(),
        ),
      );
    } catch (err) {
      console.log(err);
    }
  });

  it('Retry on 500 - fail after 5 retries', async () => {
    let hasReached5thTime = false;
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(500, '500');
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(500, '500');
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(500, '500');
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(500, '500');
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(500, '500');
    nock('https://api.snyk.io')
      .post(/\/apierror/)
      .reply(500, () => {
        hasReached5thTime = true;
        return '500';
      });

    try {
      const response = await requestManager.request({
        verb: 'POST',
        url: '/apierror',
      });
      expect(response.data).toThrowError();
    } catch (err) {
      expect(err).toBeInstanceOf(RequestsManagerApiError);
      expect(hasReached5thTime).toBeTruthy();
    }
  });
});