urllib/test/user-agent.test.ts at e2d80e33c3da7ddaa4df6d40f6a901c999fe2c3c · 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
import { strict as assert } from 'node:assert';
import { describe, it, beforeAll, afterAll } from 'vitest';
import urllib from '../src/index.js';
import { startServer } from './fixtures/server.js';

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

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

  it('should return default user agent', async () => {
    const response = await urllib.request(_url, {
      dataType: 'json',
    });
    assert.equal(response.status, 200);
    // console.log(response.data.headers);
    assert.match(response.data.headers['user-agent'], /^node-urllib\/VERSION Node\.js\/\d+\.\d+\.\d+ \(/);
  });

  it('should return no user agent if user-agent header is set to empty string', async () => {
    const response = await urllib.request(_url, {
      dataType: 'json',
      headers: { 'user-agent': '' },
    });
    assert.equal(response.status, 200);
    // console.log(response.data.headers);
    assert.equal(response.data.headers['user-agent'], undefined);
  });

  it('should return no user agent if user-agent header is set undefined', async () => {
    const response = await urllib.request(_url, {
      dataType: 'json',
      headers: { 'user-agent': undefined },
    });
    assert.equal(response.status, 200);
    // console.log(response.data.headers);
    assert.equal(response.data.headers['user-agent'], undefined);
  });

  it('should return mock user agent', async () => {
    let response = await urllib.request(_url, {
      dataType: 'json',
      headers: { 'user-agent': 'mock agent' },
    });
    assert.equal(response.status, 200);
    // console.log(response.data.headers);
    assert.equal(response.data.headers['user-agent'], 'mock agent');

    response = await urllib.request(_url, {
      dataType: 'json',
      headers: { 'User-Agent': 'mock agent upper case' },
    });
    assert.equal(response.status, 200);
    // console.log(response.data.headers);
    assert.equal(response.data.headers['user-agent'], 'mock agent upper case');
  });
});