mtcute/e2e/deno/tests/01.auth.ts

import { assertEquals } from 'https://deno.land/std@0.223.0/assert/mod.ts'

import { MtcuteError, tl } from '@mtcute/core'
import { BaseTelegramClient, TelegramClient } from '@mtcute/core/client.js'

import { getApiParams } from '../utils.ts'

const getAccountId = () =>
    Math.floor(Math.random() * 10000)
        .toString()
        .padStart(4, '0')

Deno.test('1. authorization', { sanitizeResources: false }, async (t) => {
    await t.step('should authorize in default dc', async () => {
        const base = new BaseTelegramClient(getApiParams('dc2.session'))
        const tg = new TelegramClient({ client: base })

        // reset storage just in case
        await base.mt.storage.load()
        await base.storage.clear(true)

        while (true) {
            const phone = `999662${getAccountId()}`
            let user

            try {
                user = await tg.start({
                    phone,
                    code: () => '22222',
                })
            } catch (e) {
                if (
                    (e instanceof MtcuteError && e.message.match(/Signup is no longer supported|2FA is enabled/)) ||
                    tl.RpcError.is(e, 'SESSION_PASSWORD_NEEDED')
                ) {
                    // retry with another number
                    continue
                } else {
                    await tg.close()
                    throw e
                }
            }

            await tg.close()

            assertEquals(user.isSelf, true)
            assertEquals(user.phoneNumber, phone)
            break
        }
    })

    await t.step('should authorize in dc 1', async () => {
        const base = new BaseTelegramClient(getApiParams('dc1.session'))
        const tg = new TelegramClient({ client: base })

        // reset storage just in case
        await base.mt.storage.load()
        await base.mt.storage.clear(true)

        while (true) {
            const phone = `999661${getAccountId()}`
            let user

            try {
                user = await tg.start({
                    phone,
                    code: () => '11111',
                })
            } catch (e) {
                if (
                    (e instanceof MtcuteError && e.message.match(/Signup is no longer supported|2FA is enabled/)) ||
                    tl.RpcError.is(e, 'SESSION_PASSWORD_NEEDED')
                ) {
                    // retry with another number
                    continue
                } else {
                    await tg.close()
                    throw e
                }
            }

            await tg.close()

            assertEquals(user.isSelf, true)
            assertEquals(user.phoneNumber, phone)
            break
        }
    })
})