File Explorer

1/*2 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.3 * SPDX-License-Identifier: Apache-2.0.4 */5 6import * as test_env from "@test/test_env"7import * as retry from "@test/retry"8import * as mqtt311 from "./mqtt";9import * as aws_iot_mqtt311 from "./aws_iot";10import * as io from "./io"11import { v4 as uuid } from 'uuid';12import {once} from "events";13 14jest.setTimeout(30000);15 16test_env.conditional_test(test_env.AWS_IOT_ENV.mqtt311_is_valid_custom_auth_unsigned())('Aws Iot Core Mqtt over websockets with Non-Signing Custom Auth - Connection Success', async () => {17    await retry.networkTimeoutRetryWrapper( async () => {18        let builder = aws_iot_mqtt311.AwsIotMqttConnectionConfigBuilder.new_default_builder();19        builder.with_custom_authorizer(20            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_UNSIGNED_USERNAME,21            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_UNSIGNED_NAME,22            "",23            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_UNSIGNED_PASSWORD24        )25        builder.with_endpoint(test_env.AWS_IOT_ENV.MQTT311_HOST);26        builder.with_client_id(`node-mqtt-unit-test-${uuid()}`)27        let config = builder.build();28        let client = new mqtt311.MqttClient();29        let connection = client.new_connection(config);30        await connection.connect();31        await connection.disconnect();32    })33});34 35test_env.conditional_test(test_env.AWS_IOT_ENV.mqtt311_is_valid_custom_auth_signed())('Aws Iot Core Mqtt over websockets with Signing Custom Auth - Connection Success', async () => {36    await retry.networkTimeoutRetryWrapper( async () => {37        let builder = aws_iot_mqtt311.AwsIotMqttConnectionConfigBuilder.new_default_builder();38        builder.with_custom_authorizer(39            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_USERNAME,40            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_NAME,41            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_SIGNATURE,42            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_PASSWORD,43            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_KEY_NAME,44            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_TOKEN,45        )46        builder.with_endpoint(test_env.AWS_IOT_ENV.MQTT311_HOST);47        builder.with_client_id(`node-mqtt-unit-test-${uuid()}`)48        let config = builder.build();49        let client = new mqtt311.MqttClient();50        let connection = client.new_connection(config);51        await connection.connect();52        await connection.disconnect();53    })54});55 56test_env.conditional_test(test_env.AWS_IOT_ENV.mqtt311_is_valid_custom_auth_signed())('Aws Iot Core Mqtt over websockets with Signing Custom Auth Unencoded Signature - Connection Success', async () => {57    await retry.networkTimeoutRetryWrapper( async () => {58        let builder = aws_iot_mqtt311.AwsIotMqttConnectionConfigBuilder.new_default_builder();59        builder.with_custom_authorizer(60            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_USERNAME,61            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_NAME,62            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_SIGNATURE_UNENCODED,63            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_PASSWORD,64            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_KEY_NAME,65            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_TOKEN,66        )67        builder.with_endpoint(test_env.AWS_IOT_ENV.MQTT311_HOST);68        builder.with_client_id(`node-mqtt-unit-test-${uuid()}`)69        let config = builder.build();70        let client = new mqtt311.MqttClient();71        let connection = client.new_connection(config);72        await connection.connect();73        await connection.disconnect();74    })75});76 77test_env.conditional_test(test_env.AWS_IOT_ENV.mqtt311_is_valid_cred())('MQTT Browser Websocket Connect/Disconnect', async () => {78    await retry.networkTimeoutRetryWrapper( async () => {79        let builder = aws_iot_mqtt311.AwsIotMqttConnectionConfigBuilder.new_with_websockets();80        builder.with_endpoint(test_env.AWS_IOT_ENV.MQTT311_HOST);81        builder.with_client_id(`node-mqtt-unit-test-${uuid()}`)82        builder.with_credentials(83            test_env.AWS_IOT_ENV.MQTT311_REGION,84            test_env.AWS_IOT_ENV.MQTT311_CRED_ACCESS_KEY,85            test_env.AWS_IOT_ENV.MQTT311_CRED_SECRET_ACCESS_KEY,86            test_env.AWS_IOT_ENV.MQTT311_CRED_SESSION_TOKEN87        );88        let config = builder.build();89        let client = new mqtt311.MqttClient(new io.ClientBootstrap());90        let connection = client.new_connection(config);91 92        const connectionSuccess = once(connection, "connection_success");93        await connection.connect();94 95        let connectionSuccessEvent: mqtt311.OnConnectionSuccessResult = (await connectionSuccess)[0];96        expect(connectionSuccessEvent.session_present).toBeFalsy();97        expect(connectionSuccessEvent.reason_code).toBeUndefined();98 99        const closed = once(connection, "closed");100        await connection.disconnect();101        await closed;102    })103});104 105test_env.conditional_test(test_env.AWS_IOT_ENV.mqtt311_is_valid_cred())('MQTT Browser Websocket Connect/Disconnect No Bootstrap', async () => {106    await retry.networkTimeoutRetryWrapper( async () => {107        let builder = aws_iot_mqtt311.AwsIotMqttConnectionConfigBuilder.new_with_websockets();108        builder.with_endpoint(test_env.AWS_IOT_ENV.MQTT311_HOST);109        builder.with_client_id(`node-mqtt-unit-test-${uuid()}`)110        builder.with_credentials(111            test_env.AWS_IOT_ENV.MQTT311_REGION,112            test_env.AWS_IOT_ENV.MQTT311_CRED_ACCESS_KEY,113            test_env.AWS_IOT_ENV.MQTT311_CRED_SECRET_ACCESS_KEY,114            test_env.AWS_IOT_ENV.MQTT311_CRED_SESSION_TOKEN115        );116        let config = builder.build();117 118        let client = new mqtt311.MqttClient();119        let connection = client.new_connection(config);120 121        const connectionSuccess = once(connection, "connection_success");122        await connection.connect();123 124        let connectionSuccessEvent: mqtt311.OnConnectionSuccessResult = (await connectionSuccess)[0];125        expect(connectionSuccessEvent.session_present).toBeFalsy();126        expect(connectionSuccessEvent.reason_code).toBeUndefined();127 128        const closed = once(connection, "closed");129        await connection.disconnect();130        await closed;131    })132});133 134test_env.conditional_test(test_env.AWS_IOT_ENV.mqtt311_is_valid_custom_auth_signed())('MQTT Browser Websocket Connect/Disconnect - Connection Failure', async () => {135    await retry.networkTimeoutRetryWrapper( async () => {136        let builder = aws_iot_mqtt311.AwsIotMqttConnectionConfigBuilder.new_default_builder();137        builder.with_custom_authorizer(138            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_USERNAME,139            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_NAME,140            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_SIGNATURE,141            "Thisisnotthepassword",142            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_KEY_NAME,143            test_env.AWS_IOT_ENV.MQTT311_CUSTOM_AUTH_SIGNED_TOKEN,144        )145        builder.with_endpoint(test_env.AWS_IOT_ENV.MQTT311_HOST);146        builder.with_client_id(`node-mqtt-unit-test-${uuid()}`)147        let config = builder.build();148        let client = new mqtt311.MqttClient();149        let connection = client.new_connection(config);150 151        const connectionFailure = once(connection, "connection_failure")152        try {153            connection.connect();154        } catch (error) {155            // Skip - this is expected because we are intentionally using a bad password156        }157 158        let connectionFailedEvent: mqtt311.OnConnectionFailedResult = (await connectionFailure)[0];159        expect(connectionFailedEvent).toBeDefined();160        expect(connectionFailedEvent.error).toBeDefined();161 162        // Disconnect to stop trying to reconnect163        connection.disconnect();164    })165});166