77 lines
1.9 KiB
TypeScript
77 lines
1.9 KiB
TypeScript
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: Error[] = []
|
|
|
|
const errorHandler = (err: Error) => {
|
|
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: Error[] = []
|
|
|
|
const errorHandler = (err: Error) => {
|
|
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'),
|
|
)
|
|
})
|
|
})
|