mtcute/packages/sqlite/test/sqlite.test.ts

import { afterAll, beforeAll, describe, expect, it, vi } from 'vitest'

import { stubPeerUser, testStateStorage, testStorage } from '@mtcute/test'

import { SqliteStorage } from '../src/index.js'

describe('SqliteStorage', () => {
    testStorage(new SqliteStorage(), {
        // sqlite implements "unimportant" updates, which are batched once every 30sec (tested below)
        skipEntityOverwrite: true,
        customTests: (s) => {
            describe('batching', () => {
                beforeAll(() => void vi.useFakeTimers())
                afterAll(() => void vi.useRealTimers())

                it('should batch entity writes', async () => {
                    s.updatePeers([stubPeerUser])
                    s.updatePeers([{ ...stubPeerUser, username: 'test123' }])
                    s.save()

                    // eslint-disable-next-line
                    expect(Object.keys(s['_pendingUnimportant'])).toEqual([String(stubPeerUser.id)])
                    // not yet updated
                    expect(s.getPeerByUsername(stubPeerUser.username!)).not.toBeNull()
                    expect(s.getPeerByUsername('test123')).toBeNull()

                    await vi.advanceTimersByTimeAsync(30001)

                    expect(s.getPeerByUsername(stubPeerUser.username!)).toBeNull()
                    expect(s.getPeerByUsername('test123')).not.toBeNull()
                })

                it('should batch update state writes', () => {
                    s.setUpdatesPts(123)
                    s.setUpdatesQts(456)
                    s.setUpdatesDate(789)
                    s.setUpdatesSeq(999)

                    // not yet updated
                    expect(s.getUpdatesState()).toBeNull()

                    s.save()

                    expect(s.getUpdatesState()).toEqual([123, 456, 789, 999])
                })
            })
        },
    })
    testStateStorage(new SqliteStorage())
})