mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2025-12-26 03:31:24 +08:00
665a62080b
* test: more unit tests - Adjust vitest configuration to handle main process and renderer process tests separately - Add unit tests for main process utils - Add unit tests for the renderer process - Add three component tests to verify vitest usage: `DragableList`, `Scrollbar`, `QuickPanelView` - Add an e2e startup test to verify playwright usage - Extract `splitApiKeyString` and add tests for it - Add and format some comments * fix: mock individual properties * test: add tests for CustomTag * test: add tests for ExpandableText * test: conditional rendering tooltip of tag * chore: update dependencies
72 lines
2.5 KiB
TypeScript
72 lines
2.5 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
||
|
||
import { decrypt, encrypt } from '../aes'
|
||
|
||
const key = '12345678901234567890123456789012' // 32字节
|
||
const iv = '1234567890abcdef1234567890abcdef' // 32字节hex，实际应16字节hex
|
||
|
||
function getIv16() {
|
||
// 取前16字节作为 hex
|
||
return iv.slice(0, 32)
|
||
}
|
||
|
||
describe('aes utils', () => {
|
||
it('should encrypt and decrypt normal string', () => {
|
||
const text = 'hello world'
|
||
const { iv: outIv, encryptedData } = encrypt(text, key, getIv16())
|
||
expect(typeof encryptedData).toBe('string')
|
||
expect(outIv).toBe(getIv16())
|
||
const decrypted = decrypt(encryptedData, getIv16(), key)
|
||
expect(decrypted).toBe(text)
|
||
})
|
||
|
||
it('should support unicode and special chars', () => {
|
||
const text = '你好，世界！🌟🚀'
|
||
const { encryptedData } = encrypt(text, key, getIv16())
|
||
const decrypted = decrypt(encryptedData, getIv16(), key)
|
||
expect(decrypted).toBe(text)
|
||
})
|
||
|
||
it('should handle empty string', () => {
|
||
const text = ''
|
||
const { encryptedData } = encrypt(text, key, getIv16())
|
||
const decrypted = decrypt(encryptedData, getIv16(), key)
|
||
expect(decrypted).toBe(text)
|
||
})
|
||
|
||
it('should encrypt and decrypt long string', () => {
|
||
const text = 'a'.repeat(100_000)
|
||
const { encryptedData } = encrypt(text, key, getIv16())
|
||
const decrypted = decrypt(encryptedData, getIv16(), key)
|
||
expect(decrypted).toBe(text)
|
||
})
|
||
|
||
it('should throw error for wrong key', () => {
|
||
const text = 'test'
|
||
const { encryptedData } = encrypt(text, key, getIv16())
|
||
expect(() => decrypt(encryptedData, getIv16(), 'wrongkeywrongkeywrongkeywrongkey')).toThrow()
|
||
})
|
||
|
||
it('should throw error for wrong iv', () => {
|
||
const text = 'test'
|
||
const { encryptedData } = encrypt(text, key, getIv16())
|
||
expect(() => decrypt(encryptedData, 'abcdefabcdefabcdefabcdefabcdefab', key)).toThrow()
|
||
})
|
||
|
||
it('should throw error for invalid key/iv length', () => {
|
||
expect(() => encrypt('test', 'shortkey', getIv16())).toThrow()
|
||
expect(() => encrypt('test', key, 'shortiv')).toThrow()
|
||
})
|
||
|
||
it('should throw error for invalid encrypted data', () => {
|
||
expect(() => decrypt('nothexdata', getIv16(), key)).toThrow()
|
||
})
|
||
|
||
it('should throw error for non-string input', () => {
|
||
// @ts-expect-error purposely pass wrong type to test error branch
|
||
expect(() => encrypt(null, key, getIv16())).toThrow()
|
||
// @ts-expect-error purposely pass wrong type to test error branch
|
||
expect(() => decrypt(null, getIv16(), key)).toThrow()
|
||
})
|
||
})
|