urllib/test/fetch.test.ts at 333b129e3c3af4ed9bebe2c48a619c07d1357820 · node-modules/urllib · 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
117
118
119
120
121
122
123
import assert from 'node:assert/strict';
import diagnosticsChannel from 'node:diagnostics_channel';
import { setTimeout as sleep } from 'node:timers/promises';
import { describe, it, beforeAll, afterAll } from 'vitest';
import { startServer } from './fixtures/server.js';
import {
  fetch, FetchDiagnosticsMessage, FetchFactory, FetchResponseDiagnosticsMessage,
} from '../src/fetch.js';
import { RequestDiagnosticsMessage, ResponseDiagnosticsMessage } from '../src/HttpClient.js';
import { Request } from 'undici';

describe('fetch.test.ts', () => {
  let close: any;
  let _url: string;
  beforeAll(async () => {
    const { closeServer, url } = await startServer();
    close = closeServer;
    _url = url;
  });

  afterAll(async () => {
    await close();
  });

  it('fetch should work', async () => {
    let requestDiagnosticsMessage: RequestDiagnosticsMessage;
    let responseDiagnosticsMessage: ResponseDiagnosticsMessage;
    let fetchDiagnosticsMessage: FetchDiagnosticsMessage;
    let fetchResponseDiagnosticsMessage: FetchResponseDiagnosticsMessage;
    diagnosticsChannel.subscribe('urllib:request', msg => {
      requestDiagnosticsMessage = msg as RequestDiagnosticsMessage;
    });
    diagnosticsChannel.subscribe('urllib:response', msg => {
      responseDiagnosticsMessage = msg as ResponseDiagnosticsMessage;
    });
    diagnosticsChannel.subscribe('urllib:fetch:request', msg => {
      fetchDiagnosticsMessage = msg as FetchDiagnosticsMessage;
    });
    diagnosticsChannel.subscribe('urllib:fetch:response', msg => {
      fetchResponseDiagnosticsMessage = msg as FetchResponseDiagnosticsMessage;
    });
    FetchFactory.setClientOptions({});

    let response = await fetch(`${_url}html`);

    assert(response);
    assert(requestDiagnosticsMessage!.request);
    assert(responseDiagnosticsMessage!.request);
    assert(responseDiagnosticsMessage!.response);
    assert(responseDiagnosticsMessage!.response.socket.localAddress);
    assert([ '127.0.0.1', '::1' ].includes(responseDiagnosticsMessage!.response.socket.localAddress));

    assert(fetchDiagnosticsMessage!.fetch);
    assert(fetchResponseDiagnosticsMessage!.fetch);
    assert(fetchResponseDiagnosticsMessage!.response);
    assert(fetchResponseDiagnosticsMessage!.timingInfo);

    await sleep(1);
    // again, keep alive
    response = await fetch(`${_url}html`);
    // console.log(responseDiagnosticsMessage!.response.socket);
    assert(responseDiagnosticsMessage!.response.socket.handledRequests > 1);
    assert(responseDiagnosticsMessage!.response.socket.handledResponses > 1);

    const stats = FetchFactory.getDispatcherPoolStats();
    assert(stats);
    assert(Object.keys(stats).length > 0);
  });

  it('fetch error should has socket info', async () => {
    let requestDiagnosticsMessage: RequestDiagnosticsMessage;
    let responseDiagnosticsMessage: ResponseDiagnosticsMessage;
    let fetchDiagnosticsMessage: FetchDiagnosticsMessage;
    let fetchResponseDiagnosticsMessage: FetchResponseDiagnosticsMessage;
    diagnosticsChannel.subscribe('urllib:request', msg => {
      requestDiagnosticsMessage = msg as RequestDiagnosticsMessage;
    });
    diagnosticsChannel.subscribe('urllib:response', msg => {
      responseDiagnosticsMessage = msg as ResponseDiagnosticsMessage;
    });
    diagnosticsChannel.subscribe('urllib:fetch:request', msg => {
      fetchDiagnosticsMessage = msg as FetchDiagnosticsMessage;
    });
    diagnosticsChannel.subscribe('urllib:fetch:response', msg => {
      fetchResponseDiagnosticsMessage = msg as FetchResponseDiagnosticsMessage;
    });
    FetchFactory.setClientOptions({});

    await assert.rejects(async () => {
      await fetch(`${_url}html?timeout=9999`, {
        signal: AbortSignal.timeout(100),
      });
    }, (err: any) => {
      assert.equal(err.name, 'TimeoutError');
      assert.equal(err.message, 'The operation was aborted due to timeout');
      return true;
    });

    assert(requestDiagnosticsMessage!.request);
    assert(responseDiagnosticsMessage!.request);
    assert(responseDiagnosticsMessage!.response);
    // console.log(responseDiagnosticsMessage!.response.socket);
    assert(responseDiagnosticsMessage!.response.socket.localAddress);
    assert([ '127.0.0.1', '::1' ].includes(responseDiagnosticsMessage!.response.socket.localAddress));

    assert(fetchDiagnosticsMessage!.fetch);
    assert(fetchResponseDiagnosticsMessage!.fetch);

    const stats = FetchFactory.getDispatcherPoolStats();
    assert(stats);
    assert(Object.keys(stats).length > 0, `dispatcher pool stats: ${JSON.stringify(stats)}`);
  });

  it('fetch request with post should work', async () => {
    await assert.doesNotReject(async () => {
      const request = new Request(_url, {
        method: 'POST',
        body: 'test-body',
      });
      await fetch(request);
    }, /Cannot construct a Request with a Request object that has already been used/);
  });
});