format.test.ts

src/lib/format.test.ts

1. import { describe, expect, it } from 'vitest'
3. import { costNano, estimateTokens } from '../ipc/mock/mockClient'
4. import { perMillion, relativeDay, shortModel, snippetParts, tokens, usd } from './format'
6. describe('money and tokens', () => {
7. it('formats nano-dollars like cortex-core does', () => {
8. expect(usd(0)).toBe('$0')
9. expect(usd(12)).toBe('<$0.0001')
10. expect(usd(3_500_000)).toBe('$0.0035')
11. expect(usd(1_240_000_000)).toBe('$1.24')
12. })
14. it('prices per million', () => {
15. expect(perMillion(0)).toBe('free')
16. expect(perMillion(0.5)).toBe('$0.50')
17. expect(perMillion(0.075)).toBe('$0.075')
18. expect(perMillion(15)).toBe('$15.00')
19. })
21. it('token counts', () => {
22. expect(tokens(412)).toBe('412')
23. expect(tokens(1234)).toBe('1.2k')
24. expect(tokens(262144)).toBe('262k')
25. expect(tokens(1_048_576)).toBe('1.0M')
26. })
28. it('the mock bills exactly as cost.rs', () => {
29. // Same cases as cost.rs's tests.
30. const p = { input_per_m: 0.5, output_per_m: 1.5, cache_read_per_m: 0.1 }
31. expect(costNano({ prompt_tokens: 1000, completion_tokens: 2000, cached_tokens: 0 }, p)).toBe(3_500_000)
32. expect(costNano({ prompt_tokens: 1000, completion_tokens: 0, cached_tokens: 400 }, p)).toBe(340_000)
33. expect(costNano({ prompt_tokens: 10, completion_tokens: -5, cached_tokens: 99 }, p)).toBe(1_000)
34. expect(estimateTokens('abcde')).toBe(2)
35. expect(estimateTokens('你好')).toBe(2)
36. })
37. })
39. describe('text helpers', () => {
40. it('short model names', () => {
41. expect(shortModel('Qwen/Qwen3-235B-A22B-Instruct-2507')).toBe('Qwen3-235B-A22B-Instruct-2507')
42. expect(shortModel('llama3.2')).toBe('llama3.2')
43. })
45. it('search snippet marks', () => {
46. expect(snippetParts('use \u0001rsync\u0002 with \u0001delete\u0002.')).toEqual([
47. { text: 'use ', hit: false },
48. { text: 'rsync', hit: true },
49. { text: ' with ', hit: false },
50. { text: 'delete', hit: true },
51. { text: '.', hit: false },
52. ])
53. })
55. it('relative days', () => {
56. const now = new Date(2026, 8, 28, 15).getTime()
57. expect(relativeDay(new Date(2026, 8, 28, 1).getTime(), now)).toBe('Today')
58. expect(relativeDay(new Date(2026, 8, 27, 23).getTime(), now)).toBe('Yesterday')
59. expect(relativeDay(new Date(2026, 8, 24).getTime(), now)).toBe('This week')
60. expect(relativeDay(new Date(2026, 5, 1).getTime(), now)).toBe('Earlier')
61. })
62. })