@tdengine/websocket 3.2.2 → 3.3.0

package/lib/test/tmq/tmq.test.js

@@ -0,0 +1,513 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const constant_1 = require("@src/tmq/constant");
+const wsTmq_1 = require("@src/tmq/wsTmq");
+const config_1 = require("@src/common/config");
+const wsSql_1 = require("@src/sql/wsSql");
+const utils_1 = require("@test-helpers/utils");
+const wsConnectorPool_1 = require("@src/client/wsConnectorPool");
+const log_1 = require("@src/common/log");
+(0, log_1.setLevel)("debug");
+const stable = "st";
+const db = "ws_tmq_test";
+const topics = ["topic_ws_bean"];
+const tokenTopic = "topic_token_test";
+let createTopic = `create topic if not exists ${topics[0]} as select * from ${db}.${stable}`;
+let dropTopic = `DROP TOPIC IF EXISTS ${topics[0]};`;
+let dsn = `ws://${(0, utils_1.testUsername)()}:${(0, utils_1.testPassword)()}@localhost:6041`;
+let tmqDsn = "ws://localhost:6041";
+beforeAll(async () => {
+    let conf = new config_1.WSConfig(dsn);
+    const createDB = `create database if not exists ${db} keep 3650`;
+    const dropDB = `drop database if exists ${db}`;
+    const useDB = `use ${db}`;
+    const stableTags = [
+        true,
+        -1,
+        -2,
+        -3,
+        BigInt(-4),
+        1,
+        2,
+        3,
+        BigInt(4),
+        parseFloat((3.1415).toFixed(5)),
+        parseFloat((3.14159265).toFixed(15)),
+        "varchar_tag_1",
+        "nchar_tag_1",
+    ];
+    const tableValues = [
+        [
+            BigInt(1656677710000),
+            0,
+            -1,
+            -2,
+            BigInt(-3),
+            0,
+            1,
+            2,
+            BigInt(3),
+            parseFloat((3.1415).toFixed(5)),
+            parseFloat((3.14159265).toFixed(15)),
+            "varchar_col_1",
+            "nchar_col_1",
+            true,
+            "NULL",
+            "POINT (4.0 8.0)",
+            "0x7661726332",
+        ],
+        [
+            BigInt(1656677720000),
+            -1,
+            -2,
+            -3,
+            BigInt(-4),
+            1,
+            2,
+            3,
+            BigInt(4),
+            parseFloat((3.1415 * 2).toFixed(5)),
+            parseFloat((3.14159265 * 2).toFixed(15)),
+            "varchar_col_2",
+            "nchar_col_2",
+            false,
+            "NULL",
+            "POINT (3.0 5.0)",
+            "0x7661726333",
+        ],
+        [
+            BigInt(1656677730000),
+            -2,
+            -3,
+            -4,
+            BigInt(-5),
+            2,
+            3,
+            4,
+            BigInt(5),
+            parseFloat((3.1415 * 3).toFixed(5)),
+            parseFloat((3.14159265 * 3).toFixed(15)),
+            "varchar_col_3",
+            "nchar_col_3",
+            true,
+            "NULL",
+            "LINESTRING (1.000000 1.000000, 2.000000 2.000000, 5.000000 5.000000)",
+            "0x7661726334",
+        ],
+        [
+            BigInt(1656677740000),
+            -3,
+            -4,
+            -5,
+            BigInt(-6),
+            3,
+            4,
+            5,
+            BigInt(6),
+            parseFloat((3.1415 * 4).toFixed(5)),
+            parseFloat((3.14159265 * 4).toFixed(15)),
+            "varchar_col_4",
+            "nchar_col_4",
+            false,
+            "NULL",
+            "POLYGON ((3.000000 6.000000, 5.000000 6.000000, 5.000000 8.000000, 3.000000 8.000000, 3.000000 6.000000))",
+            "0x7661726335",
+        ],
+        [
+            BigInt(1656677750000),
+            -4,
+            -5,
+            -6,
+            BigInt(-7),
+            4,
+            5,
+            6,
+            BigInt(7),
+            parseFloat((3.1415 * 5).toFixed(5)),
+            parseFloat((3.14159265 * 5).toFixed(15)),
+            "varchar_col_5",
+            "nchar_col_5",
+            true,
+            "NULL",
+            "POINT (7.0 9.0)",
+            "0x7661726335",
+        ],
+    ];
+    const tableCNValues = [
+        [
+            BigInt(1656677760000),
+            0,
+            -1,
+            -2,
+            BigInt(-3),
+            0,
+            1,
+            2,
+            BigInt(3),
+            parseFloat((3.1415).toFixed(5)),
+            parseFloat((3.14159265).toFixed(15)),
+            "varchar_列_壹",
+            "nchar_列_甲",
+            true,
+            "NULL",
+            "POINT (4.0 8.0)",
+            "0x7661726332",
+        ],
+        [
+            BigInt(1656677770000),
+            -1,
+            -2,
+            -3,
+            BigInt(-4),
+            1,
+            2,
+            3,
+            BigInt(4),
+            parseFloat((3.1415 * 2).toFixed(5)),
+            parseFloat((3.14159265 * 2).toFixed(15)),
+            "varchar_列_贰",
+            "nchar_列_乙",
+            false,
+            "NULL",
+            "POINT (3.0 5.0)",
+            "0x7661726333",
+        ],
+        [
+            BigInt(1656677780000),
+            -2,
+            -3,
+            -4,
+            BigInt(-5),
+            2,
+            3,
+            4,
+            BigInt(5),
+            parseFloat((3.1415 * 3).toFixed(5)),
+            parseFloat((3.14159265 * 3).toFixed(15)),
+            "varchar_列_叁",
+            "nchar_列_丙",
+            true,
+            "NULL",
+            "LINESTRING (1.000000 1.000000, 2.000000 2.000000, 5.000000 5.000000)",
+            "0x7661726334",
+        ],
+        [
+            BigInt(1656677790000),
+            -3,
+            -4,
+            -5,
+            BigInt(-6),
+            3,
+            4,
+            5,
+            BigInt(6),
+            parseFloat((3.1415 * 4).toFixed(5)),
+            parseFloat((3.14159265 * 4).toFixed(15)),
+            "varchar_列_肆",
+            "nchar_列_丁",
+            false,
+            "NULL",
+            "POLYGON ((3.000000 6.000000, 5.000000 6.000000, 5.000000 8.000000, 3.000000 8.000000, 3.000000 6.000000))",
+            "0x7661726335",
+        ],
+        [
+            BigInt(1656677800000),
+            -4,
+            -5,
+            -6,
+            BigInt(-7),
+            4,
+            5,
+            6,
+            BigInt(7),
+            parseFloat((3.1415 * 5).toFixed(5)),
+            parseFloat((3.14159265 * 5).toFixed(15)),
+            "varchar_列_伍",
+            "nchar_列_戊",
+            true,
+            "NULL",
+            "POINT (7.0 9.0)",
+            "0x7661726335",
+        ],
+    ];
+    let ws = await wsSql_1.WsSql.open(conf);
+    await ws.exec(dropTopic);
+    await ws.exec(dropDB);
+    await ws.exec(createDB);
+    await ws.exec(useDB);
+    await ws.exec((0, utils_1.createSTable)(stable));
+    await ws.exec(createTopic);
+    let insert = (0, utils_1.insertStable)(tableValues, stableTags, stable);
+    let insertRes = await ws.exec(insert);
+    insert = (0, utils_1.insertStable)(tableCNValues, stableTags, stable);
+    insertRes = await ws.exec(insert);
+    await ws.exec(`create topic if not exists ${tokenTopic} as select * from ${db}.${stable}`);
+    await ws.close();
+});
+describe("TDWebSocket.Tmq()", () => {
+    jest.setTimeout(20 * 1000);
+    let configMap = new Map([
+        [constant_1.TMQConstants.GROUP_ID, "gId"],
+        [constant_1.TMQConstants.CONNECT_USER, (0, utils_1.testUsername)()],
+        [constant_1.TMQConstants.CONNECT_PASS, (0, utils_1.testPassword)()],
+        [constant_1.TMQConstants.AUTO_OFFSET_RESET, "earliest"],
+        [constant_1.TMQConstants.CLIENT_ID, "test_tmq_client"],
+        [constant_1.TMQConstants.WS_URL, tmqDsn],
+        [constant_1.TMQConstants.ENABLE_AUTO_COMMIT, "true"],
+        [constant_1.TMQConstants.AUTO_COMMIT_INTERVAL_MS, "1000"],
+        ["session.timeout.ms", "10000"],
+        ["max.poll.interval.ms", "30000"],
+        ["msg.with.table.name", "true"],
+    ]);
+    console.log(configMap);
+    test("normal connect", async () => {
+        let consumer = await wsTmq_1.WsConsumer.newConsumer(configMap);
+        await consumer.close();
+    });
+    test("connect error", async () => {
+        expect.assertions(1);
+        let consumer = null;
+        let errConfigMap = new Map([
+            [constant_1.TMQConstants.GROUP_ID, "test"],
+            [constant_1.TMQConstants.CONNECT_USER, "root"],
+            [constant_1.TMQConstants.CONNECT_PASS, "test"],
+            [constant_1.TMQConstants.AUTO_OFFSET_RESET, "earliest1"],
+            [constant_1.TMQConstants.CLIENT_ID, "test_tmq_client"],
+            [constant_1.TMQConstants.WS_URL, tmqDsn],
+            [constant_1.TMQConstants.ENABLE_AUTO_COMMIT, "true"],
+            [constant_1.TMQConstants.AUTO_COMMIT_INTERVAL_MS, "1000"],
+            ["session.timeout.ms", "10000"],
+            ["max.poll.interval.ms", "30000"],
+            ["msg.with.table.name", "true"],
+        ]);
+        try {
+            consumer = await wsTmq_1.WsConsumer.newConsumer(errConfigMap);
+            await consumer.subscribe(topics);
+        }
+        catch (e) {
+            console.log(e);
+            expect([104]).toContain(e.code);
+        }
+        finally {
+            if (consumer) {
+                await consumer.close();
+            }
+        }
+    });
+    test("normal Subscribe", async () => {
+        let consumer = await wsTmq_1.WsConsumer.newConsumer(configMap);
+        await consumer.subscribe(topics);
+        let assignment = await consumer.assignment();
+        console.log(assignment);
+        let counts = 0;
+        let useTime = [];
+        for (let i = 0; i < 5; i++) {
+            let startTime = new Date().getTime();
+            let res = await consumer.poll(500);
+            let currTime = new Date().getTime();
+            useTime.push(Math.abs(currTime - startTime));
+            for (let [key, value] of res) {
+                console.log(key, value.getMeta());
+                let data = value.getData();
+                if (data == null || data.length == 0) {
+                    break;
+                }
+                for (let record of data) {
+                    console.log("-----===>>", record);
+                }
+            }
+        }
+        await consumer.seekToBeginning(assignment);
+        for (let i = 0; i < 5; i++) {
+            let startTime = new Date().getTime();
+            let res = await consumer.poll(500);
+            let currTime = new Date().getTime();
+            useTime.push(Math.abs(currTime - startTime));
+            for (let [key, value] of res) {
+                console.log(key, value.getMeta());
+                let data = value.getData();
+                if (data == null || data.length == 0) {
+                    break;
+                }
+                counts += data.length;
+            }
+        }
+        let topicArray = await consumer.subscription();
+        expect(topics.length).toEqual(topicArray.length);
+        for (let index = 0; index < topicArray.length; index++) {
+            expect(topics[index]).toEqual(topicArray[index]);
+        }
+        assignment = await consumer.commit();
+        console.log(assignment);
+        assignment = await consumer.committed(assignment);
+        assignment = await consumer.commitOffsets(assignment);
+        console.log(assignment);
+        await consumer.unsubscribe();
+        await consumer.close();
+        console.log("------------->", useTime);
+        console.log("------------->", counts);
+        expect(counts).toEqual(10);
+    });
+    test("Topic not exist", async () => {
+        let consumer = await wsTmq_1.WsConsumer.newConsumer(configMap);
+        try {
+            await consumer.subscribe(["aaa"]);
+        }
+        catch (e) {
+            expect(e.message).toMatch("Topic not exist");
+        }
+        await consumer.close();
+    });
+    test("normal seek", async () => {
+        let consumer = await wsTmq_1.WsConsumer.newConsumer(configMap);
+        await consumer.subscribe(topics);
+        let assignment = await consumer.assignment();
+        console.log("------START--------", assignment);
+        await consumer.seekToEnd(assignment);
+        await consumer.seekToBeginning(assignment);
+        await consumer.seekToEnd(assignment);
+        assignment = await consumer.assignment();
+        console.log("------END--------", assignment);
+        for (let i = 0; i < assignment.length; i++) {
+            expect(assignment[i].offset).toEqual(assignment[i].end);
+        }
+        await consumer.unsubscribe();
+        await consumer.close();
+    });
+    (0, utils_1.testEnterprise)("connect with token", async () => {
+        const conf = new config_1.WSConfig(dsn);
+        const wsSql = await wsSql_1.WsSql.open(conf);
+        await wsSql.exec("drop token if exists test_tmq_token");
+        const wsRows = await wsSql.query(`create token test_tmq_token from user ${(0, utils_1.testUsername)()}`);
+        await wsRows.next();
+        const token = wsRows.getData()?.[0];
+        expect(token).toBeTruthy();
+        await wsRows.close();
+        const tmqConf = new Map([
+            [constant_1.TMQConstants.WS_URL, "ws://localhost:6041"],
+            [constant_1.TMQConstants.CONNECT_USER, "invalid_user"],
+            [constant_1.TMQConstants.CONNECT_PASS, "invalid_pass"],
+            [constant_1.TMQConstants.CONNECT_TOKEN, token],
+            [constant_1.TMQConstants.GROUP_ID, "g1101"],
+            [constant_1.TMQConstants.CLIENT_ID, "c1101"],
+            [constant_1.TMQConstants.AUTO_OFFSET_RESET, "earliest"],
+            [constant_1.TMQConstants.ENABLE_AUTO_COMMIT, "false"],
+            [constant_1.TMQConstants.AUTO_COMMIT_INTERVAL_MS, "1000"],
+        ]);
+        const consumer = await wsTmq_1.WsConsumer.newConsumer(tmqConf);
+        await consumer.subscribe([tokenTopic]);
+        let count = 0;
+        for (let i = 0; i < 5; i++) {
+            const res = await consumer.poll(500);
+            for (const [, value] of res) {
+                const data = value.getData();
+                if (data == null || data.length == 0) {
+                    break;
+                }
+                count += data.length;
+            }
+        }
+        expect(count).toEqual(10);
+        await (0, utils_1.Sleep)(3000);
+        await consumer.unsubscribe();
+        await consumer.close();
+        await wsSql.exec("drop token if exists test_tmq_token");
+        await wsSql.close();
+    });
+    (0, utils_1.testEnterprise)("connect with token url", async () => {
+        const conf = new config_1.WSConfig(dsn);
+        const wsSql = await wsSql_1.WsSql.open(conf);
+        await wsSql.exec("drop token if exists test_tmq_token_url");
+        const wsRows = await wsSql.query(`create token test_tmq_token_url from user ${(0, utils_1.testUsername)()}`);
+        await wsRows.next();
+        const token = wsRows.getData()?.[0];
+        expect(token).toBeTruthy();
+        await wsRows.close();
+        const tmqConf = new Map([
+            [constant_1.TMQConstants.WS_URL, `ws://localhost:6041?bearer_token=${token}`],
+            [constant_1.TMQConstants.CONNECT_USER, "invalid_user"],
+            [constant_1.TMQConstants.CONNECT_PASS, "invalid_pass"],
+            [constant_1.TMQConstants.GROUP_ID, "g1103"],
+            [constant_1.TMQConstants.CLIENT_ID, "c1103"],
+            [constant_1.TMQConstants.AUTO_OFFSET_RESET, "earliest"],
+            [constant_1.TMQConstants.ENABLE_AUTO_COMMIT, false],
+            [constant_1.TMQConstants.AUTO_COMMIT_INTERVAL_MS, 1000],
+        ]);
+        const consumer = await wsTmq_1.WsConsumer.newConsumer(tmqConf);
+        await consumer.subscribe([tokenTopic]);
+        let count = 0;
+        for (let i = 0; i < 5; i++) {
+            const res = await consumer.poll(500);
+            for (const [, value] of res) {
+                const data = value.getData();
+                if (data == null || data.length == 0) {
+                    break;
+                }
+                count += data.length;
+            }
+        }
+        expect(count).toEqual(10);
+        await (0, utils_1.Sleep)(3000);
+        await consumer.unsubscribe();
+        await consumer.close();
+        await wsSql.exec("drop token if exists test_tmq_token_url");
+        await wsSql.close();
+    });
+    (0, utils_1.testEnterprise)("connect with invalid token", async () => {
+        const conf = new Map([
+            [constant_1.TMQConstants.GROUP_ID, "g1102"],
+            [constant_1.TMQConstants.CLIENT_ID, "c1102"],
+            [constant_1.TMQConstants.WS_URL, "ws://localhost:6041?bearer_token=invalid_token"],
+        ]);
+        await expect(wsTmq_1.WsConsumer.newConsumer(conf)).rejects.toMatchObject({
+            message: expect.stringMatching(/invalid token/i),
+        });
+        conf.set(constant_1.TMQConstants.WS_URL, "ws://localhost:6041");
+        conf.set(constant_1.TMQConstants.CONNECT_TOKEN, "invalid_token1");
+        await expect(wsTmq_1.WsConsumer.newConsumer(conf)).rejects.toMatchObject({
+            message: expect.stringMatching(/invalid token/i),
+        });
+        conf.set(constant_1.TMQConstants.WS_URL, "ws://localhost:6041?bearer_token=");
+        conf.delete(constant_1.TMQConstants.CONNECT_TOKEN);
+        await expect(wsTmq_1.WsConsumer.newConsumer(conf)).rejects.toMatchObject({
+            message: expect.stringMatching(/invalid url/i),
+        });
+        conf.set(constant_1.TMQConstants.WS_URL, "ws://localhost:6041");
+        conf.set(constant_1.TMQConstants.CONNECT_TOKEN, "");
+        await expect(wsTmq_1.WsConsumer.newConsumer(conf)).rejects.toMatchObject({
+            message: expect.stringMatching(/invalid url/i),
+        });
+    });
+    (0, utils_1.testNon3360)("connector version info", async () => {
+        const consumer = await wsTmq_1.WsConsumer.newConsumer(configMap);
+        await consumer.subscribe(topics);
+        const conf = new config_1.WSConfig("ws://localhost:6041");
+        conf.setUser((0, utils_1.testUsername)());
+        conf.setPwd((0, utils_1.testPassword)());
+        const wsSql = await wsSql_1.WsSql.open(conf);
+        await (0, utils_1.Sleep)(2000);
+        const wsRows = await wsSql.query("show connections");
+        let count = 0;
+        while (await wsRows.next()) {
+            const data = wsRows.getData();
+            if (Array.isArray(data) && data.some(v => typeof v === "string" && v.includes("nodejs-ws"))) {
+                count++;
+            }
+        }
+        expect(count).toBeGreaterThanOrEqual(2);
+        await wsRows.close();
+        await wsSql.close();
+        await consumer.unsubscribe();
+        await consumer.close();
+    });
+});
+afterAll(async () => {
+    const dropDB = `drop database if exists ${db}`;
+    const conf = new config_1.WSConfig(dsn);
+    const ws = await wsSql_1.WsSql.open(conf);
+    await ws.exec(dropTopic);
+    await ws.exec(`drop topic if exists ${tokenTopic}`);
+    await ws.exec(dropDB);
+    await ws.close();
+    wsConnectorPool_1.WebSocketConnectionPool.instance().destroyed();
+});

package/lib/test/unit/connectionManager.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=connectionManager.test.d.ts.map

package/lib/test/unit/connectionManager.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"connectionManager.test.d.ts","sourceRoot":"","sources":["../../../test/unit/connectionManager.test.ts"],"names":[],"mappings":""}

package/lib/test/unit/connectionManager.test.js

@@ -0,0 +1,91 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const wsConnectionManager_1 = require("../../src/client/wsConnectionManager");
+function makeParsedUrl(hosts) {
+    return {
+        hosts,
+        scheme: "ws",
+        username: "root",
+        password: "taosdata",
+        database: "testdb",
+        params: new Map(),
+    };
+}
+function makeAuthInfo() {
+    return {
+        username: "root",
+        password: "taosdata",
+        database: "testdb",
+    };
+}
+describe("ConnectionManager", () => {
+    describe("constructor", () => {
+        test("initializes with CLOSED state", () => {
+            const parsed = makeParsedUrl([{ host: "host1", port: 6041 }]);
+            const cm = new wsConnectionManager_1.ConnectionManager(parsed, makeAuthInfo());
+            expect(cm.state).toBe(wsConnectionManager_1.ConnectionState.CLOSED);
+        });
+        test("random initial host selection is within range", () => {
+            const hosts = [
+                { host: "host1", port: 6041 },
+                { host: "host2", port: 6042 },
+                { host: "host3", port: 6043 },
+            ];
+            const parsed = makeParsedUrl(hosts);
+            // Run multiple times to verify randomness stays in bounds
+            for (let i = 0; i < 20; i++) {
+                const cm = new wsConnectionManager_1.ConnectionManager(parsed, makeAuthInfo());
+                const current = cm.currentHost;
+                expect(current).toBeDefined();
+                expect(hosts).toContainEqual(current);
+            }
+        });
+        test("applies custom retry options", () => {
+            const parsed = makeParsedUrl([{ host: "host1", port: 6041 }]);
+            const opts = {
+                retries: 5,
+                retryBackoffMs: 500,
+                retryBackoffMaxMs: 10000,
+                resendWrite: true,
+            };
+            const cm = new wsConnectionManager_1.ConnectionManager(parsed, makeAuthInfo(), opts);
+            expect(cm.state).toBe(wsConnectionManager_1.ConnectionState.CLOSED);
+        });
+    });
+    describe("state management", () => {
+        test("cannot connect when not in CLOSED state", async () => {
+            const parsed = makeParsedUrl([{ host: "localhost", port: 1 }]);
+            const cm = new wsConnectionManager_1.ConnectionManager(parsed, makeAuthInfo(), { retries: 0 }, 2000);
+            // First connect will fail (no server), putting us back to CLOSED
+            try {
+                await cm.connect();
+            }
+            catch (e) {
+                // expected
+            }
+            expect(cm.state).toBe(wsConnectionManager_1.ConnectionState.CLOSED);
+        }, 15000);
+        test("close on already closed is a no-op", async () => {
+            const parsed = makeParsedUrl([{ host: "host1", port: 6041 }]);
+            const cm = new wsConnectionManager_1.ConnectionManager(parsed, makeAuthInfo());
+            await cm.close(); // Should not throw
+            expect(cm.state).toBe(wsConnectionManager_1.ConnectionState.CLOSED);
+        });
+        test("getReadyState returns -1 when not connected", () => {
+            const parsed = makeParsedUrl([{ host: "host1", port: 6041 }]);
+            const cm = new wsConnectionManager_1.ConnectionManager(parsed, makeAuthInfo());
+            expect(cm.getReadyState()).toBe(-1);
+        });
+    });
+    describe("connection failure", () => {
+        test("throws ERR_ALL_HOSTS_EXHAUSTED after all retries", async () => {
+            const parsed = makeParsedUrl([
+                { host: "nonexistent1", port: 1 },
+                { host: "nonexistent2", port: 2 },
+            ]);
+            const cm = new wsConnectionManager_1.ConnectionManager(parsed, makeAuthInfo(), { retries: 0, retryBackoffMs: 1, retryBackoffMaxMs: 1 });
+            await expect(cm.connect()).rejects.toThrow("All hosts exhausted");
+            expect(cm.state).toBe(wsConnectionManager_1.ConnectionState.CLOSED);
+        }, 30000);
+    });
+});

package/lib/test/utils.d.ts

@@ -15,4 +15,8 @@ export declare function expectStableData(rows: Array<Array<any>>, tags: Array<an
 export declare function hexToBytes(hex: string): ArrayBuffer;
 export declare function compareUint8Arrays(arr1: Uint8Array, arr2: Uint8Array): boolean;
 export declare function Sleep(ms: number): Promise<void>;
+export declare function testUsername(): string;
+export declare function testPassword(): string;
+export declare const testEnterprise: jest.It;
+export declare const testNon3360: jest.It;
 //# sourceMappingURL=utils.d.ts.map

package/lib/test/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../test/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAExD,wBAAgB,aAAa,CACzB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,EAAE,EAAE,MAAM,EACV,MAAM,EAAE,MAAM,GACf,MAAM,CAcR;AAED,wBAAgB,YAAY,CACxB,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EACzB,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,EAChB,MAAM,EAAE,MAAM,EACd,KAAK,GAAE,MAAgB,GACxB,MAAM,CAkCR;AAED,wBAAgB,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAqB7E;AAED,eAAO,MAAM,SAAS,EAAE,KAAK,CAAC,YAAY,CAsFzC,CAAC;AAEF,eAAO,MAAM,QAAQ,EAAE,KAAK,CAAC,YAAY,CAMxC,CAAC;AACF,eAAO,MAAM,OAAO,EAAE,KAAK,CAAC,YAAY,CAkEvC,CAAC;AAEF,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAKvD;AACD,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAK3D;AACD,wBAAgB,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAErD;AAED,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAKnD;AACD,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAKvD;AACD,wBAAgB,WAAW,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAEjD;AAED,wBAAgB,gBAAgB,CAC5B,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EACvB,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,GACjB,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAMnB;AAED,wBAAgB,UAAU,CAAC,GAAG,EAAE,MAAM,GAAG,WAAW,CAQnD;AAoBD,wBAAgB,kBAAkB,CAC9B,IAAI,EAAE,UAAU,EAChB,IAAI,EAAE,UAAU,GACjB,OAAO,CAYT;AAED,wBAAgB,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE/C"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../test/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAExD,wBAAgB,aAAa,CACzB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,EAAE,EAAE,MAAM,EACV,MAAM,EAAE,MAAM,GACf,MAAM,CAcR;AAED,wBAAgB,YAAY,CACxB,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EACzB,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,EAChB,MAAM,EAAE,MAAM,EACd,KAAK,GAAE,MAAgB,GACxB,MAAM,CAkCR;AAED,wBAAgB,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAqB7E;AAED,eAAO,MAAM,SAAS,EAAE,KAAK,CAAC,YAAY,CAsFzC,CAAC;AAEF,eAAO,MAAM,QAAQ,EAAE,KAAK,CAAC,YAAY,CAMxC,CAAC;AAEF,eAAO,MAAM,OAAO,EAAE,KAAK,CAAC,YAAY,CAkEvC,CAAC;AAEF,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAKvD;AACD,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAK3D;AACD,wBAAgB,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAErD;AAED,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAKnD;AACD,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAKvD;AACD,wBAAgB,WAAW,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAEjD;AAED,wBAAgB,gBAAgB,CAC5B,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EACvB,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,GACjB,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAMnB;AAED,wBAAgB,UAAU,CAAC,GAAG,EAAE,MAAM,GAAG,WAAW,CAQnD;AAED,wBAAgB,kBAAkB,CAC9B,IAAI,EAAE,UAAU,EAChB,IAAI,EAAE,UAAU,GACjB,OAAO,CAYT;AAED,wBAAgB,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE/C;AAED,wBAAgB,YAAY,IAAI,MAAM,CAErC;AAED,wBAAgB,YAAY,IAAI,MAAM,CAErC;AAED,eAAO,MAAM,cAAc,SAA2E,CAAC;AACvG,eAAO,MAAM,WAAW,SAAqE,CAAC"}

package/lib/test/utils.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.tagMeta = exports.jsonMeta = exports.tableMeta = void 0;
+exports.testNon3360 = exports.testEnterprise = exports.tagMeta = exports.jsonMeta = exports.tableMeta = void 0;
 exports.getInsertBind = getInsertBind;
 exports.insertStable = insertStable;
 exports.insertNTable = insertNTable;
@@ -17,6 +17,8 @@ exports.expectStableData = expectStableData;
 exports.hexToBytes = hexToBytes;
 exports.compareUint8Arrays = compareUint8Arrays;
 exports.Sleep = Sleep;
+exports.testUsername = testUsername;
+exports.testPassword = testPassword;
 const log_1 = __importDefault(require("../src/common/log"));
 function getInsertBind(valuesLen, tagsLen, db, stable) {
     let sql = `insert into ? using ${db}.${stable} tags ( ?`;
@@ -289,22 +291,6 @@ function hexToBytes(hex) {
     }
     return a.buffer;
 }
-// export function createStmtData(varbinary:string = "ab",
-//     geoHex:string = "0101000020E6100000000000000000F03F0000000000000040"):Array<Array<any>> {
-//     let multi:any[][] = [
-//         [1709183268567, 1709183268568, 1709183268569],
-//         [10.2, 10.3, 10.4],
-//         [292, 293, 294],
-//         [0.32, 0.33, 0.34],
-//         ];
-//     let res = hexToBytes(geoHex)
-//     let geom = Array.from(new Uint8Array(res))
-//     multi.push([geom, geom, geom])
-//     res = new TextEncoder().encode(varbinary)
-//     let binary = Array.from(new Uint8Array(res))
-//     multi.push([binary, binary, binary])
-//     return multi
-// }
 function compareUint8Arrays(arr1, arr2) {
     if (arr1.length !== arr2.length) {
         log_1.default.debug(`${arr1.length} !== ${arr2.length}`);
@@ -321,3 +307,11 @@ function compareUint8Arrays(arr1, arr2) {
 function Sleep(ms) {
     return new Promise((resolve) => setTimeout(resolve, ms));
 }
+function testUsername() {
+    return process.env.TDENGINE_TEST_USERNAME || "root";
+}
+function testPassword() {
+    return process.env.TDENGINE_TEST_PASSWORD || "taosdata";
+}
+exports.testEnterprise = process.env.TEST_ENTERPRISE?.toLowerCase() === "true" ? test : test.skip;
+exports.testNon3360 = process.env.TEST_3360?.toLowerCase() === "true" ? test.skip : test;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tdengine/websocket",
-  "version": "3.2.2",
+  "version": "3.3.0",
   "description": "The websocket Node.js connector for TDengine. TDengine versions 3.3.2.0 and above are recommended to use this connector.",
   "source": "index.ts",
   "main": "lib/index.js",