mtcute/packages/tl-utils/src/calculator.test.ts

98 lines
3.4 KiB
TypeScript

import { describe, expect, it } from 'vitest'
import { calculateStaticSizes } from './calculator.js'
import { parseTlToEntries } from './parse.js'
describe('calculateStaticSizes', () => {
const test = (tl: string, expected: object) => {
expect(calculateStaticSizes(parseTlToEntries(tl))).toEqual(expected)
}
it('computes for constructors without parameters', () => {
test('auth.logOut = Bool;', { 'auth.logOut': 4 })
})
it('computes for constructors with static parameters', () => {
test('auth.exportAuthorization#e5bfffcd dc_id:int = auth.ExportedAuthorization;', {
'auth.exportAuthorization': 8,
})
})
it('correctly skips true fields', () => {
test('help.promoData#8c39793f flags:# proxy:flags.0?true expires:int = help.PromoData;', {
'help.promoData': 12,
})
})
it('correctly skips constructors with predicated fields', () => {
test(
'help.promoData#8c39793f flags:# expires:int psa_type:flags.1?int psa_message:flags.2?int = help.PromoData;',
{},
)
})
it('correctly skips constructors with generic fields', () => {
test('invokeWithLayer {T:X} = !X;', {})
})
it('correctly skips constructors with non-static fields', () => {
test('help.promoData#8c39793f psa_type:string psa_message:string = help.PromoData;', {})
})
it('correctly skips constructors with vector fields', () => {
test('help.promoData#8c39793f psa_type:Vector<int> = help.PromoData;', {})
})
it('correctly handles static-sized children', () => {
test(
'peerUser#9db1bc6d user_id:int53 = Peer;\n' +
'help.promoData#8c39793f flags:# proxy:flags.0?true expires:int peer:peerUser = help.PromoData;',
{
peerUser: 12,
'help.promoData': 20,
},
)
})
it('correctly handles static-sized union children', () => {
test(
'peerUser#9db1bc6d user_id:int53 = Peer;\n' +
'peerChannel#9db1bc6d channel_id:int53 = Peer;\n' +
'help.promoData#8c39793f flags:# proxy:flags.0?true expires:int peer:Peer = help.PromoData;',
{
peerUser: 12,
peerChannel: 12,
'help.promoData': 24,
},
)
})
it('correctly handles differently sized union children', () => {
test(
'peerUser user_id:int53 = Peer;\n' +
'peerChannel channel_id:int53 access_hash:long = Peer;\n' +
'help.promoData#8c39793f flags:# proxy:flags.0?true expires:int peer:Peer = help.PromoData;',
{
peerUser: 12,
peerChannel: 20,
},
)
})
it('correctly handles non static-sized union children', () => {
test(
'peerUser user_id:int53 = Peer;\n' +
'peerChannel channel_id:int53 access_hash:bytes = Peer;\n' +
'help.promoData#8c39793f flags:# proxy:flags.0?true expires:int peer:Peer = help.PromoData;',
{
peerUser: 12,
},
)
test(
'peerUser user_id:int53 access_hash:bytes = Peer;\n' +
'peerChannel channel_id:int53 access_hash:bytes = Peer;\n' +
'help.promoData#8c39793f flags:# proxy:flags.0?true expires:int peer:Peer = help.PromoData;',
{},
)
})
})