@mtcute/core 0.1.1 → 0.1.2

package/esm/network/persistent-connection.test.js

@@ -0,0 +1,155 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+import { createStub, defaultTestCryptoProvider, StubTelegramTransport } from '@mtcute/test';
+import { LogManager } from '../utils/index.js';
+import { PersistentConnection } from './persistent-connection.js';
+import { defaultReconnectionStrategy } from './reconnection.js';
+class FakePersistentConnection extends PersistentConnection {
+    constructor(params) {
+        const log = new LogManager();
+        log.level = 0;
+        super(params, log);
+    }
+    onConnected() {
+        this.onConnectionUsable();
+    }
+    onError() { }
+    onMessage() { }
+}
+describe('PersistentConnection', () => {
+    beforeEach(() => void vi.useFakeTimers());
+    afterEach(() => void vi.useRealTimers());
+    const create = async (params) => {
+        return new FakePersistentConnection({
+            crypto: await defaultTestCryptoProvider(),
+            transportFactory: () => new StubTelegramTransport({}),
+            dc: createStub('dcOption'),
+            testMode: false,
+            reconnectionStrategy: defaultReconnectionStrategy,
+            ...params,
+        });
+    };
+    it('should set up listeners on transport', async () => {
+        const transportFactory = vi.fn().mockImplementation(() => {
+            const transport = new StubTelegramTransport({});
+            vi.spyOn(transport, 'on');
+            return transport;
+        });
+        await create({ transportFactory });
+        const transport = transportFactory.mock.results[0].value;
+        expect(transport.on).toHaveBeenCalledWith('ready', expect.any(Function));
+        expect(transport.on).toHaveBeenCalledWith('message', expect.any(Function));
+        expect(transport.on).toHaveBeenCalledWith('error', expect.any(Function));
+        expect(transport.on).toHaveBeenCalledWith('close', expect.any(Function));
+    });
+    it('should properly reset old transport', async () => {
+        const transportFactory = vi.fn().mockImplementation(() => {
+            const transport = new StubTelegramTransport({});
+            vi.spyOn(transport, 'close');
+            return transport;
+        });
+        const pc = await create({ transportFactory });
+        const transport = transportFactory.mock.results[0].value;
+        pc.changeTransport(transportFactory);
+        expect(transport.close).toHaveBeenCalledOnce();
+    });
+    it('should buffer unsent packages', async () => {
+        const transportFactory = vi.fn().mockImplementation(() => {
+            const transport = new StubTelegramTransport({});
+            const transportConnect = transport.connect;
+            vi.spyOn(transport, 'connect').mockImplementation((dc, test) => {
+                setTimeout(() => {
+                    transportConnect.call(transport, dc, test);
+                }, 100);
+            });
+            vi.spyOn(transport, 'send');
+            return transport;
+        });
+        const pc = await create({ transportFactory });
+        const transport = transportFactory.mock.results[0].value;
+        const data1 = new Uint8Array([1, 2, 3]);
+        const data2 = new Uint8Array([4, 5, 6]);
+        await pc.send(data1);
+        await pc.send(data2);
+        expect(transport.send).toHaveBeenCalledTimes(0);
+        await vi.advanceTimersByTimeAsync(150);
+        expect(transport.send).toHaveBeenCalledTimes(2);
+        expect(transport.send).toHaveBeenCalledWith(data1);
+        expect(transport.send).toHaveBeenCalledWith(data2);
+    });
+    it('should reconnect on close', async () => {
+        const reconnectionStrategy = vi.fn().mockImplementation(() => 1000);
+        const transportFactory = vi.fn().mockImplementation(() => new StubTelegramTransport({}));
+        const pc = await create({
+            reconnectionStrategy,
+            transportFactory,
+        });
+        const transport = transportFactory.mock.results[0].value;
+        pc.connect();
+        await vi.waitFor(() => expect(pc.isConnected).toBe(true));
+        transport.close();
+        expect(reconnectionStrategy).toHaveBeenCalledOnce();
+        expect(pc.isConnected).toBe(false);
+        await vi.advanceTimersByTimeAsync(1000);
+        expect(pc.isConnected).toBe(true);
+    });
+    describe('inactivity timeout', () => {
+        it('should disconnect on inactivity (passed in constructor)', async () => {
+            const pc = await create({
+                inactivityTimeout: 1000,
+            });
+            pc.connect();
+            await vi.waitFor(() => expect(pc.isConnected).toBe(true));
+            vi.advanceTimersByTime(1000);
+            await vi.waitFor(() => expect(pc.isConnected).toBe(false));
+        });
+        it('should disconnect on inactivity (set up with setInactivityTimeout)', async () => {
+            const pc = await create();
+            pc.connect();
+            pc.setInactivityTimeout(1000);
+            await vi.waitFor(() => expect(pc.isConnected).toBe(true));
+            vi.advanceTimersByTime(1000);
+            await vi.waitFor(() => expect(pc.isConnected).toBe(false));
+        });
+        it('should not disconnect on inactivity if disabled', async () => {
+            const pc = await create({
+                inactivityTimeout: 1000,
+            });
+            pc.connect();
+            pc.setInactivityTimeout(undefined);
+            await vi.waitFor(() => expect(pc.isConnected).toBe(true));
+            vi.advanceTimersByTime(1000);
+            await vi.waitFor(() => expect(pc.isConnected).toBe(true));
+        });
+        it('should reconnect after inactivity before sending', async () => {
+            const transportFactory = vi.fn().mockImplementation(() => {
+                const transport = new StubTelegramTransport({});
+                vi.spyOn(transport, 'connect');
+                vi.spyOn(transport, 'send');
+                return transport;
+            });
+            const pc = await create({
+                inactivityTimeout: 1000,
+                transportFactory,
+            });
+            const transport = transportFactory.mock.results[0].value;
+            pc.connect();
+            vi.advanceTimersByTime(1000);
+            await vi.waitFor(() => expect(pc.isConnected).toBe(false));
+            vi.mocked(transport.connect).mockClear();
+            await pc.send(new Uint8Array([1, 2, 3]));
+            expect(transport.connect).toHaveBeenCalledOnce();
+            expect(transport.send).toHaveBeenCalledOnce();
+        });
+        it('should propagate errors', async () => {
+            const transportFactory = vi.fn().mockImplementation(() => new StubTelegramTransport({}));
+            const pc = await create({ transportFactory });
+            const transport = transportFactory.mock.results[0].value;
+            pc.connect();
+            await vi.waitFor(() => expect(pc.isConnected).toBe(true));
+            const onErrorSpy = vi.spyOn(pc, 'onError');
+            transport.emit('error', new Error('test error'));
+            expect(onErrorSpy).toHaveBeenCalledOnce();
+        });
+    });
+});
+//# sourceMappingURL=persistent-connection.test.js.map

package/esm/network/persistent-connection.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"persistent-connection.test.js","sourceRoot":"","sources":["../../../src/network/persistent-connection.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAExE,OAAO,EAAE,UAAU,EAAE,yBAAyB,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAA;AAE3F,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAA;AAC9C,OAAO,EAAE,oBAAoB,EAA8B,MAAM,4BAA4B,CAAA;AAC7F,OAAO,EAAE,2BAA2B,EAAE,MAAM,mBAAmB,CAAA;AAE/D,MAAM,wBAAyB,SAAQ,oBAAoB;IACvD,YAAY,MAAkC;QAC1C,MAAM,GAAG,GAAG,IAAI,UAAU,EAAE,CAAA;QAC5B,GAAG,CAAC,KAAK,GAAG,CAAC,CAAA;QACb,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;IACtB,CAAC;IAED,WAAW;QACP,IAAI,CAAC,kBAAkB,EAAE,CAAA;IAC7B,CAAC;IACD,OAAO,KAAI,CAAC;IACZ,SAAS,KAAI,CAAC;CACjB;AAED,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IAClC,UAAU,CAAC,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,aAAa,EAAE,CAAC,CAAA;IACzC,SAAS,CAAC,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,aAAa,EAAE,CAAC,CAAA;IAExC,MAAM,MAAM,GAAG,KAAK,EAAE,MAA4C,EAAE,EAAE;QAClE,OAAO,IAAI,wBAAwB,CAAC;YAChC,MAAM,EAAE,MAAM,yBAAyB,EAAE;YACzC,gBAAgB,EAAE,GAAG,EAAE,CAAC,IAAI,qBAAqB,CAAC,EAAE,CAAC;YACrD,EAAE,EAAE,UAAU,CAAC,UAAU,CAAC;YAC1B,QAAQ,EAAE,KAAK;YACf,oBAAoB,EAAE,2BAA2B;YACjD,GAAG,MAAM;SACZ,CAAC,CAAA;IACN,CAAC,CAAA;IAED,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE;YACrD,MAAM,SAAS,GAAG,IAAI,qBAAqB,CAAC,EAAE,CAAC,CAAA;YAE/C,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,CAAC,CAAA;YAEzB,OAAO,SAAS,CAAA;QACpB,CAAC,CAAC,CAAA;QACF,MAAM,MAAM,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAA;QAElC,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAA8B,CAAA;QAEjF,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAA;QACxE,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAA;QAC1E,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAA;QACxE,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAA;IAC5E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE;YACrD,MAAM,SAAS,GAAG,IAAI,qBAAqB,CAAC,EAAE,CAAC,CAAA;YAE/C,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAA;YAE5B,OAAO,SAAS,CAAA;QACpB,CAAC,CAAC,CAAA;QACF,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAA;QAE7C,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAA8B,CAAA;QAEjF,EAAE,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAA;QAEpC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,oBAAoB,EAAE,CAAA;IAClD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE;YACrD,MAAM,SAAS,GAAG,IAAI,qBAAqB,CAAC,EAAE,CAAC,CAAA;YAE/C,MAAM,gBAAgB,GAAG,SAAS,CAAC,OAAO,CAAA;YAC1C,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,kBAAkB,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE;gBAC3D,UAAU,CAAC,GAAG,EAAE;oBACZ,gBAAgB,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,EAAE,IAAI,CAAC,CAAA;gBAC9C,CAAC,EAAE,GAAG,CAAC,CAAA;YACX,CAAC,CAAC,CAAA;YACF,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;YAE3B,OAAO,SAAS,CAAA;QACpB,CAAC,CAAC,CAAA;QACF,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAA;QAE7C,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAA8B,CAAA;QAEjF,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;QACvC,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;QAEvC,MAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACpB,MAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAEpB,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAE/C,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAA;QAEtC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAC/C,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAA;QAClD,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAA;IACtD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,oBAAoB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAA;QACnE,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,IAAI,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAA;QAExF,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC;YACpB,oBAAoB;YACpB,gBAAgB;SACnB,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAA8B,CAAA;QAEjF,EAAE,CAAC,OAAO,EAAE,CAAA;QAEZ,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QAEzD,SAAS,CAAC,KAAK,EAAE,CAAA;QAEjB,MAAM,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,EAAE,CAAA;QACnD,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAElC,MAAM,EAAE,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAA;QAEvC,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACrC,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC;gBACpB,iBAAiB,EAAE,IAAI;aAC1B,CAAC,CAAA;YAEF,EAAE,CAAC,OAAO,EAAE,CAAA;YAEZ,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;YAEzD,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAA;YAE5B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;QAC9D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAChF,MAAM,EAAE,GAAG,MAAM,MAAM,EAAE,CAAA;YAEzB,EAAE,CAAC,OAAO,EAAE,CAAA;YACZ,EAAE,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAA;YAE7B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;YAEzD,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAA;YAE5B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;QAC9D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC;gBACpB,iBAAiB,EAAE,IAAI;aAC1B,CAAC,CAAA;YAEF,EAAE,CAAC,OAAO,EAAE,CAAA;YACZ,EAAE,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAA;YAElC,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;YAEzD,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAA;YAE5B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QAC7D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACrD,MAAM,SAAS,GAAG,IAAI,qBAAqB,CAAC,EAAE,CAAC,CAAA;gBAE/C,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,SAAS,CAAC,CAAA;gBAC9B,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;gBAE3B,OAAO,SAAS,CAAA;YACpB,CAAC,CAAC,CAAA;YAEF,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC;gBACpB,iBAAiB,EAAE,IAAI;gBACvB,gBAAgB;aACnB,CAAC,CAAA;YACF,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAA8B,CAAA;YAEjF,EAAE,CAAC,OAAO,EAAE,CAAA;YAEZ,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAA;YAE5B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;YAE1D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAA;YAExC,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YAExC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,oBAAoB,EAAE,CAAA;YAChD,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,oBAAoB,EAAE,CAAA;QACjD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,IAAI,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAA;YAExF,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAA;YAC7C,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAA8B,CAAA;YAEjF,EAAE,CAAC,OAAO,EAAE,CAAA;YAEZ,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;YAEzD,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,CAAA;YAE1C,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAA;YAEhD,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,EAAE,CAAA;QAC7C,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA","sourcesContent":["import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'\n\nimport { createStub, defaultTestCryptoProvider, StubTelegramTransport } from '@mtcute/test'\n\nimport { LogManager } from '../utils/index.js'\nimport { PersistentConnection, PersistentConnectionParams } from './persistent-connection.js'\nimport { defaultReconnectionStrategy } from './reconnection.js'\n\nclass FakePersistentConnection extends PersistentConnection {\n    constructor(params: PersistentConnectionParams) {\n        const log = new LogManager()\n        log.level = 0\n        super(params, log)\n    }\n\n    onConnected() {\n        this.onConnectionUsable()\n    }\n    onError() {}\n    onMessage() {}\n}\n\ndescribe('PersistentConnection', () => {\n    beforeEach(() => void vi.useFakeTimers())\n    afterEach(() => void vi.useRealTimers())\n\n    const create = async (params?: Partial<PersistentConnectionParams>) => {\n        return new FakePersistentConnection({\n            crypto: await defaultTestCryptoProvider(),\n            transportFactory: () => new StubTelegramTransport({}),\n            dc: createStub('dcOption'),\n            testMode: false,\n            reconnectionStrategy: defaultReconnectionStrategy,\n            ...params,\n        })\n    }\n\n    it('should set up listeners on transport', async () => {\n        const transportFactory = vi.fn().mockImplementation(() => {\n            const transport = new StubTelegramTransport({})\n\n            vi.spyOn(transport, 'on')\n\n            return transport\n        })\n        await create({ transportFactory })\n\n        const transport = transportFactory.mock.results[0].value as StubTelegramTransport\n\n        expect(transport.on).toHaveBeenCalledWith('ready', expect.any(Function))\n        expect(transport.on).toHaveBeenCalledWith('message', expect.any(Function))\n        expect(transport.on).toHaveBeenCalledWith('error', expect.any(Function))\n        expect(transport.on).toHaveBeenCalledWith('close', expect.any(Function))\n    })\n\n    it('should properly reset old transport', async () => {\n        const transportFactory = vi.fn().mockImplementation(() => {\n            const transport = new StubTelegramTransport({})\n\n            vi.spyOn(transport, 'close')\n\n            return transport\n        })\n        const pc = await create({ transportFactory })\n\n        const transport = transportFactory.mock.results[0].value as StubTelegramTransport\n\n        pc.changeTransport(transportFactory)\n\n        expect(transport.close).toHaveBeenCalledOnce()\n    })\n\n    it('should buffer unsent packages', async () => {\n        const transportFactory = vi.fn().mockImplementation(() => {\n            const transport = new StubTelegramTransport({})\n\n            const transportConnect = transport.connect\n            vi.spyOn(transport, 'connect').mockImplementation((dc, test) => {\n                setTimeout(() => {\n                    transportConnect.call(transport, dc, test)\n                }, 100)\n            })\n            vi.spyOn(transport, 'send')\n\n            return transport\n        })\n        const pc = await create({ transportFactory })\n\n        const transport = transportFactory.mock.results[0].value as StubTelegramTransport\n\n        const data1 = new Uint8Array([1, 2, 3])\n        const data2 = new Uint8Array([4, 5, 6])\n\n        await pc.send(data1)\n        await pc.send(data2)\n\n        expect(transport.send).toHaveBeenCalledTimes(0)\n\n        await vi.advanceTimersByTimeAsync(150)\n\n        expect(transport.send).toHaveBeenCalledTimes(2)\n        expect(transport.send).toHaveBeenCalledWith(data1)\n        expect(transport.send).toHaveBeenCalledWith(data2)\n    })\n\n    it('should reconnect on close', async () => {\n        const reconnectionStrategy = vi.fn().mockImplementation(() => 1000)\n        const transportFactory = vi.fn().mockImplementation(() => new StubTelegramTransport({}))\n\n        const pc = await create({\n            reconnectionStrategy,\n            transportFactory,\n        })\n\n        const transport = transportFactory.mock.results[0].value as StubTelegramTransport\n\n        pc.connect()\n\n        await vi.waitFor(() => expect(pc.isConnected).toBe(true))\n\n        transport.close()\n\n        expect(reconnectionStrategy).toHaveBeenCalledOnce()\n        expect(pc.isConnected).toBe(false)\n\n        await vi.advanceTimersByTimeAsync(1000)\n\n        expect(pc.isConnected).toBe(true)\n    })\n\n    describe('inactivity timeout', () => {\n        it('should disconnect on inactivity (passed in constructor)', async () => {\n            const pc = await create({\n                inactivityTimeout: 1000,\n            })\n\n            pc.connect()\n\n            await vi.waitFor(() => expect(pc.isConnected).toBe(true))\n\n            vi.advanceTimersByTime(1000)\n\n            await vi.waitFor(() => expect(pc.isConnected).toBe(false))\n        })\n\n        it('should disconnect on inactivity (set up with setInactivityTimeout)', async () => {\n            const pc = await create()\n\n            pc.connect()\n            pc.setInactivityTimeout(1000)\n\n            await vi.waitFor(() => expect(pc.isConnected).toBe(true))\n\n            vi.advanceTimersByTime(1000)\n\n            await vi.waitFor(() => expect(pc.isConnected).toBe(false))\n        })\n\n        it('should not disconnect on inactivity if disabled', async () => {\n            const pc = await create({\n                inactivityTimeout: 1000,\n            })\n\n            pc.connect()\n            pc.setInactivityTimeout(undefined)\n\n            await vi.waitFor(() => expect(pc.isConnected).toBe(true))\n\n            vi.advanceTimersByTime(1000)\n\n            await vi.waitFor(() => expect(pc.isConnected).toBe(true))\n        })\n\n        it('should reconnect after inactivity before sending', async () => {\n            const transportFactory = vi.fn().mockImplementation(() => {\n                const transport = new StubTelegramTransport({})\n\n                vi.spyOn(transport, 'connect')\n                vi.spyOn(transport, 'send')\n\n                return transport\n            })\n\n            const pc = await create({\n                inactivityTimeout: 1000,\n                transportFactory,\n            })\n            const transport = transportFactory.mock.results[0].value as StubTelegramTransport\n\n            pc.connect()\n\n            vi.advanceTimersByTime(1000)\n\n            await vi.waitFor(() => expect(pc.isConnected).toBe(false))\n\n            vi.mocked(transport.connect).mockClear()\n\n            await pc.send(new Uint8Array([1, 2, 3]))\n\n            expect(transport.connect).toHaveBeenCalledOnce()\n            expect(transport.send).toHaveBeenCalledOnce()\n        })\n\n        it('should propagate errors', async () => {\n            const transportFactory = vi.fn().mockImplementation(() => new StubTelegramTransport({}))\n\n            const pc = await create({ transportFactory })\n            const transport = transportFactory.mock.results[0].value as StubTelegramTransport\n\n            pc.connect()\n\n            await vi.waitFor(() => expect(pc.isConnected).toBe(true))\n\n            const onErrorSpy = vi.spyOn(pc, 'onError')\n\n            transport.emit('error', new Error('test error'))\n\n            expect(onErrorSpy).toHaveBeenCalledOnce()\n        })\n    })\n})\n"]}

package/esm/network/session-connection.js

@@ -1353,7 +1353,6 @@ export class SessionConnection extends PersistentConnection {
             });
         }
         const result = writer.result();
-        // probably the easiest way lol
         this.log.debug('sending %d messages: size = %db, acks = %d (msg_id = %s), ping = %s (msg_id = %s), state_req = %s (msg_id = %s), resend = %s (msg_id = %s), cancels = %s (msg_id = %s), rpc = %s, container = %s, root msg_id = %l', messageCount, packetSize, ackMsgIds?.length || 'false', ackMsgIds?.map((it) => it.toString()), Boolean(pingRequest), pingMsgId, getStateMsgIds?.map((it) => it.toString()) || 'false', getStateMsgId, resendMsgIds?.map((it) => it.toString()) || 'false', cancelRpcs, cancelRpcs?.map((it) => it.toString()) || 'false', resendMsgId, rpcToSend.map((it) => it.method), useContainer, rootMsgId);
         const enc = this._session.encryptMessage(result);
         this.send(enc).catch((err) => {
@@ -1362,7 +1361,9 @@ export class SessionConnection extends PersistentConnection {
             if (ackMsgIds) {
                 this._session.queuedAcks.splice(0, 0, ...ackMsgIds);
             }
-            this._onMessageFailed(rootMsgId, 'unknown error');
+            if (rootMsgId) {
+                this._onMessageFailed(rootMsgId, 'unknown error');
+            }
         });
     }
 }