mtcute/packages/core/tests/fuzz/fuzz-session.spec.ts

import { expect } from 'chai'
import { randomBytes } from 'crypto'
import { describe, it } from 'mocha'

import { sleep } from '../../src'
import { createTestTelegramClient } from './utils'

// eslint-disable-next-line @typescript-eslint/no-var-requires
require('dotenv-flow').config()

describe('fuzz : session', async function () {
    this.timeout(45000)

    it('random auth_key', async () => {
        const client = createTestTelegramClient()

        // random key
        const initKey = randomBytes(256)
        await client.storage.setAuthKeyFor(2, initKey)

        // client is supposed to handle this and generate a new key

        const errors: unknown[] = []

        const errorHandler = (err: unknown) => {
            errors.push(err)
        }

        client.onError(errorHandler)

        await client.connect()

        await sleep(10000)

        await client.close()

        expect(errors.length).eq(0)

        expect((await client.storage.getAuthKeyFor(2))?.toString('hex')).not.eq(
            initKey.toString('hex'),
        )
    })

    it('random auth_key for other dc', async () => {
        const client = createTestTelegramClient()

        // random key for dc1
        const initKey = randomBytes(256)
        await client.storage.setAuthKeyFor(1, initKey)

        // client is supposed to handle this and generate a new key

        const errors: unknown[] = []

        const errorHandler = (err: unknown) => {
            errors.push(err)
        }

        client.onError(errorHandler)

        await client.connect()
        await client.waitUntilUsable()

        const conn = await client.createAdditionalConnection(1)
        await conn.sendRpc({ _: 'help.getConfig' })

        await sleep(10000)

        await client.close()

        expect(errors.length).eq(0)

        expect((await client.storage.getAuthKeyFor(1))?.toString('hex')).not.eq(
            initKey.toString('hex'),
        )
    })
})