93 lines
2.5 KiB
TypeScript
93 lines
2.5 KiB
TypeScript
import {jest, expect, test, afterEach} from '@jest/globals'
|
|
import AsyncStorage from '@react-native-async-storage/async-storage'
|
|
|
|
import {defaults, schema} from '#/state/persisted/schema'
|
|
import {transform, migrate} from '#/state/persisted/legacy'
|
|
import * as store from '#/state/persisted/store'
|
|
import {logger} from '#/logger'
|
|
import * as fixtures from '#/state/persisted/__tests__/fixtures'
|
|
|
|
const write = jest.mocked(store.write)
|
|
const read = jest.mocked(store.read)
|
|
|
|
jest.mock('#/logger')
|
|
jest.mock('#/state/persisted/store', () => ({
|
|
write: jest.fn(),
|
|
read: jest.fn(),
|
|
}))
|
|
|
|
afterEach(() => {
|
|
jest.clearAllMocks()
|
|
AsyncStorage.clear()
|
|
})
|
|
|
|
test('migrate: fresh install', async () => {
|
|
await migrate()
|
|
|
|
expect(AsyncStorage.getItem).toHaveBeenCalledWith('root')
|
|
expect(read).toHaveBeenCalledTimes(1)
|
|
expect(logger.info).toHaveBeenCalledWith(
|
|
'persisted state: no migration needed',
|
|
)
|
|
})
|
|
|
|
test('migrate: fresh install, existing new storage', async () => {
|
|
read.mockResolvedValueOnce(defaults)
|
|
|
|
await migrate()
|
|
|
|
expect(AsyncStorage.getItem).toHaveBeenCalledWith('root')
|
|
expect(read).toHaveBeenCalledTimes(1)
|
|
expect(logger.info).toHaveBeenCalledWith(
|
|
'persisted state: no migration needed',
|
|
)
|
|
})
|
|
|
|
test('migrate: fresh install, AsyncStorage error', async () => {
|
|
const prevGetItem = AsyncStorage.getItem
|
|
|
|
const error = new Error('test error')
|
|
|
|
AsyncStorage.getItem = jest.fn(() => {
|
|
throw error
|
|
})
|
|
|
|
await migrate()
|
|
|
|
expect(AsyncStorage.getItem).toHaveBeenCalledWith('root')
|
|
expect(logger.error).toHaveBeenCalledWith(error, {
|
|
message: 'persisted state: error migrating legacy storage',
|
|
})
|
|
|
|
AsyncStorage.getItem = prevGetItem
|
|
})
|
|
|
|
test('migrate: has legacy data', async () => {
|
|
await AsyncStorage.setItem('root', JSON.stringify(fixtures.LEGACY_DATA_DUMP))
|
|
|
|
await migrate()
|
|
|
|
expect(write).toHaveBeenCalledWith(transform(fixtures.LEGACY_DATA_DUMP))
|
|
expect(logger.info).toHaveBeenCalledWith(
|
|
'persisted state: migrated legacy storage',
|
|
)
|
|
})
|
|
|
|
test('migrate: has legacy data, fails validation', async () => {
|
|
const legacy = fixtures.LEGACY_DATA_DUMP
|
|
// @ts-ignore
|
|
legacy.shell.colorMode = 'invalid'
|
|
await AsyncStorage.setItem('root', JSON.stringify(legacy))
|
|
|
|
await migrate()
|
|
|
|
const transformed = transform(legacy)
|
|
const validate = schema.safeParse(transformed)
|
|
|
|
expect(write).not.toHaveBeenCalled()
|
|
expect(logger.error).toHaveBeenCalledWith(
|
|
'persisted state: legacy data failed validation',
|
|
// @ts-ignore
|
|
{error: validate.error},
|
|
)
|
|
})
|