chatgpt-api/apps/api/src/db/schema/common.test.ts

21 wiersze
594 B
TypeScript

import { expect, test } from 'vitest'
import { getIdSchemaForModelType } from '../schemas'
import {
createIdForModel,
idMaxLength,
idPrefixMap,
type ModelType
} from './common'
for (const modelType of Object.keys(idPrefixMap)) {
test(`${modelType} id`, () => {
for (let i = 0; i < 100; ++i) {
const id = createIdForModel(modelType as ModelType)
expect(id.startsWith(idPrefixMap[modelType as ModelType])).toBe(true)
expect(id.length).toBeLessThanOrEqual(idMaxLength)
expect(getIdSchemaForModelType(modelType as ModelType).parse(id)).toBe(id)
}
})
}