mtcute/packages/test/src/client.test.ts

import { describe, expect, it } from 'vitest'

import { hexEncode } from '@mtcute/core/utils.js'

import { StubTelegramClient } from './client.js'
import { createStub } from './stub.js'

describe('client stub', () => {
    it('should correctly intercept rpc calls', async () => {
        const client = new StubTelegramClient()

        const stubConfig = createStub('config')
        client.respondWith('help.getConfig', () => stubConfig)

        await client.with(async () => {
            const result = await client.call({ _: 'help.getConfig' })
            expect(result).toEqual(stubConfig)
        })
    })

    it('should correctly decrypt intercepted raw messages', async () => {
        const log: string[] = []

        const client = new StubTelegramClient()

        client.onRawMessage((msg) => {
            log.push(`message ctor=${hexEncode(msg.subarray(0, 4))}`)
            client.close().catch(() => {})
        })

        await client.with(async () => {
            await client.call({ _: 'help.getConfig' }).catch(() => {}) // ignore "client closed" error

            expect(log).toEqual([
                'message ctor=dcf8f173', // msg_container
            ])
        })
    })
})