File Explorer

1/*2 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.3 * SPDX-License-Identifier: Apache-2.0.4 */5 6import { HttpClientConnectionManager, HttpClientConnection, HttpHeader, HttpHeaders, HttpRequest } from "@awscrt/http";7import { ClientBootstrap, SocketOptions, SocketType, SocketDomain, ClientTlsContext, TlsConnectionOptions, InputStream } from "@awscrt/io";8import { PassThrough } from "stream";9 10jest.setTimeout(10000);11jest.retryTimes(3);12 13test('HTTP Headers', () => {14 15    const long_token = "AQoDYXdzEPT//////////wEXAMPLEtc764bNrC9SAPBSM22wDOk4x4HIZ8j4FZTwdQWLWsKWHGBuFqwAeMicRXmxfpSPfIeoIYRqTflfKD8YUuwthAx7mSEI/qkPpKPi/kMcGdQrmGdeehM4IC1NtBmUpp2wUE8phUZampKsburEDy0KPkyQDYwT7WZ0wq5VSXDvp75YU9HFvlRd8Tx6q6fE8YQcHNVXAkiY9q6d+xo0rKwT38xVqr7ZD0u0iPPkUL64lIZbqBAz+scqKmlzm8FDrypNC9Yjc8fPOLn9FX9KSYvKTr4rvx3iSIlTJabIQwj2ICCR/oLxBA==";16    const header_array: HttpHeader[] = [17        ['Host', 'www.amazon.com'],18        ['Content-Length', '42'],19        ["x-amz-security-token", long_token],20    ];21    let headers = new HttpHeaders(header_array);22    let request = new HttpRequest("", "", new HttpHeaders(header_array));23    <unknown>request;24 25    const iterator = headers[Symbol.iterator].call(headers);26    <unknown>iterator;27    const next = iterator.next.call(iterator);28    <unknown>next;29 30    let found_headers = 0;31    for (const header of headers) {32        expect(['Host', 'Content-Length', 'x-amz-security-token']).toContain(header[0]);33        expect(['www.amazon.com', '42', long_token]).toContain(header[1]);34        found_headers++;35    }36    expect(found_headers).toBe(3);37    // Upgrade header does not exist38    expect(headers.get('Upgrade')).toBeFalsy();39 40    // Make sure case doesn't matter41    expect(headers.get('HOST')).toBe('www.amazon.com');42    expect(headers.get('x-amz-security-token')).toBe(long_token);43 44    // Remove Content-Length, and make sure host is all that's left45    headers.remove('content-length');46    headers.remove('x-amz-security-token');47    found_headers = 0;48    for (const header of headers) {49        expect(header[0]).toBe('Host');50        expect(header[1]).toBe('www.amazon.com');51        found_headers++;52    }53    expect(found_headers).toBe(1);54 55    headers.clear();56    for (const header of headers) {57        // this should never be called58        expect(header).toBeNull();59    }60});61 62test('HTTP Request without body', () => {63    let request = new HttpRequest("GET", "/index.html");64 65    expect(request.method).toBe("GET");66    expect(request.path).toBe('/index.html');67    expect(request.headers.length).toBe(0);68 69    request.method = "POST";70    request.path = "/test.html"71 72    expect(request.method).toBe("POST");73    expect(request.path).toBe('/test.html');74 75    request.headers.add("Host", "www.amazon.com");76    expect(request.headers.length).toBe(1);77});78 79test('HTTP Request with body', () => {80 81    let stream = new PassThrough();82    stream.write("test");83    stream.end();84    let body_stream = new InputStream(stream);85    let request = new HttpRequest("POST", "/index.html", undefined, body_stream);86 87    expect(request.method).toBe("POST");88    expect(request.path).toBe('/index.html');89    expect(request.headers.length).toBe(0);90 91    // Body property for request is not readable92});93 94 95async function test_connection(host: string, port: number, tls_opts?: TlsConnectionOptions, bootstrap?: ClientBootstrap) {96    let setup_error_code: Number = -1;97    let shutdown_error_code: Number = -1;98    let connection_error: Error | undefined;99    const promise = new Promise((resolve, reject) => {100        let connection = new HttpClientConnection(101            bootstrap,102            host,103            port,104            new SocketOptions(SocketType.STREAM, SocketDomain.IPV4, 3000),105            tls_opts106        );107        connection.on('connect', () => {108            setup_error_code = 0;109            connection.close();110        });111        connection.on('close', () => {112            if (!connection_error) {113                shutdown_error_code = 0;114                resolve(true);115            }116        });117        connection.on('error', (error) => {118            connection_error = error;119            reject(error);120        });121    });122    await expect(promise).resolves.toBeTruthy();123 124    expect(setup_error_code).toEqual(0);125    expect(shutdown_error_code).toEqual(0);126    expect(connection_error).toBeUndefined();127}128 129test('HTTP Connection Create/Destroy', async () => {130    await test_connection("s3.amazonaws.com", 80, undefined, new ClientBootstrap());131});132 133test('HTTP Connection Create/Destroy Undef Bootstrap', async () => {134    await test_connection("s3.amazonaws.com", 80, undefined, undefined);135});136 137test('HTTPS Connection Create/Destroy', async () => {138    const host = "s3.amazonaws.com";139    await test_connection(host, 443, new TlsConnectionOptions(new ClientTlsContext(), host), new ClientBootstrap());140});141 142test('HTTPS Connection Create/Destroy Undef Bootstrap', async () => {143    const host = "s3.amazonaws.com";144    await test_connection(host, 443, new TlsConnectionOptions(new ClientTlsContext(), host), undefined);145});146 147async function test_stream(method: string, host: string, port: number, activate: boolean, tls_opts?: TlsConnectionOptions) {148    const promise = new Promise((resolve, reject) => {149        let connection = new HttpClientConnection(150            new ClientBootstrap(),151            host,152            port,153            new SocketOptions(SocketType.STREAM, SocketDomain.IPV4, 3000),154            tls_opts);155        connection.on('connect', () => {156            let request = new HttpRequest(157                method,158                '/',159                new HttpHeaders([160                    ['host', host],161                    ['user-agent', 'AWS CRT for NodeJS']162                ])163            );164            let stream = connection.request(request);165            stream.on('response', (status_code, headers) => {166                expect(status_code == 301 || status_code == 200).toBe(true);167                expect(headers).toBeDefined();168            });169            stream.on('data', (body_data) => {170                expect(body_data.byteLength).toBeGreaterThan(0);171            });172            stream.on('end', () => {173                connection.close();174            });175            stream.on('error', (error) => {176                connection.close();177                console.log(error);178                expect(error).toBeUndefined();179            });180            if (activate) {181                stream.activate();182            } else {183                resolve(true);184            }185        });186        connection.on('close', () => {187            resolve(true);188        });189        connection.on('error', (error) => {190            reject(error);191        });192    });193 194    await expect(promise).resolves.toBeTruthy();195}196 197test('HTTP Stream GET', async () => {198    await test_stream('GET', 'amazon.com', 80, true, undefined);199});200 201 202test('HTTPS Stream GET', async () => {203    const host = 'amazon.com';204    await test_stream('GET', host, 443, true, new TlsConnectionOptions(new ClientTlsContext(), host));205});206 207test('HTTP Stream UnActivated', async () => {208    await test_stream('GET', 'amazon.com', 80, false, undefined);209});210 211test('HTTP Connection Manager create/destroy', () => {212    const bootstrap = new ClientBootstrap();213    let connection_manager = new HttpClientConnectionManager(214        bootstrap,215        "s3.amazon.com",216        80,217        4,218        16 * 1024,219        new SocketOptions(SocketType.STREAM, SocketDomain.IPV4, 3000),220        undefined221    );222    expect(connection_manager).toBeDefined();223    connection_manager.close();224});225 226test('HTTP Connection Manager acquire/release', async () => {227    const bootstrap = new ClientBootstrap();228    let connection_manager = new HttpClientConnectionManager(229        bootstrap,230        "s3.amazon.com",231        80,232        4,233        16 * 1024,234        new SocketOptions(SocketType.STREAM, SocketDomain.IPV4, 3000),235        undefined236    );237    expect(connection_manager).toBeDefined();238 239    const connection = await connection_manager.acquire();240    expect(connection).toBeDefined();241    connection_manager.release(connection);242 243    connection_manager.close();244});245 246 247 248 249test('HTTP Connection Manager acquire/stream/release', async () => {250    const bootstrap = new ClientBootstrap();251    let connection_manager = new HttpClientConnectionManager(252        bootstrap,253        "amazon.com",254        80,255        4,256        16 * 1024,257        new SocketOptions(SocketType.STREAM, SocketDomain.IPV4, 3000),258        undefined259    );260    expect(connection_manager).toBeDefined();261 262    const connection = await connection_manager.acquire();263    expect(connection).toBeDefined();264 265    let request = new HttpRequest(266        'GET',267        '/',268        new HttpHeaders([269            ['host', 'amazon.com'],270            ['user-agent', 'AWS CRT for NodeJS']271        ])272    );273 274    let connection_error: Error | undefined;275 276    const promise = new Promise((resolve, reject) => {277        let stream = connection.request(request);278        stream.on('response', (status_code, headers) => {279            expect(status_code == 301 || status_code == 200).toBe(true);280            expect(headers).toBeDefined();281        });282        stream.on('data', (body_data) => {283            expect(body_data.byteLength).toBeGreaterThan(0);284        });285        stream.on('end', () => {286            connection_manager.release(connection);287            connection_manager.close();288            if (!connection_error) {289                resolve(true);290            }291        });292        stream.on('error', (error) => {293            connection_error = error;294            reject(error);295        });296        stream.activate();297    })298 299    await expect(promise).resolves.toBeTruthy();300    expect(connection_error).toBeUndefined();301});302 303