mirror of
https://github.com/AgentSeal/codeburn.git
synced 2026-10-20 03:05:24 +00:00
2f2a14fb70
Some checks are pending
CI / semgrep (push) Waiting to run
Tests / test (22) (push) Waiting to run
Tests / test (24) (push) Waiting to run
Tests / test (26) (push) Waiting to run
Tests / test-platforms (macos-latest) (push) Waiting to run
Tests / test (22.13.0) (push) Waiting to run
Tests / test-platforms (windows-latest) (push) Waiting to run
Tests / desktop (push) Waiting to run
* Add regressions for redirected editor storage roots * Honor redirected data roots for Copilot and Cursor * Verify redirected Copilot OTel discovery on Linux * Reproduce Cursor cache collisions across storage roots * Bind Cursor result caches to their database and backfill redirected history * Verify redirected discovery against the latest daily cache * Update env guard and document editor storage overrides
64 lines
3.2 KiB
TypeScript
64 lines
3.2 KiB
TypeScript
import { afterEach, expect, it, vi } from 'vitest'
|
|
import { mkdir, mkdtemp, rm, writeFile } from 'node:fs/promises'
|
|
import { join } from 'node:path'
|
|
import { tmpdir } from 'node:os'
|
|
import { clearSessionCache, parseAllSessions } from '../src/parser.js'
|
|
import { currentTzKey, ensureCacheHydrated, toDateString } from '../src/daily-cache.js'
|
|
import { aggregateProjectsIntoDays } from '../src/day-aggregator.js'
|
|
import { calculateCost } from '../src/models.js'
|
|
|
|
afterEach(() => {
|
|
clearSessionCache()
|
|
vi.unstubAllEnvs()
|
|
})
|
|
|
|
it.skipIf(process.platform === 'darwin')('backfills redirected editor history in a finalized v50 cache while retaining other providers', async () => {
|
|
const root = await mkdtemp(join(tmpdir(), 'editor-discovery-cache-'))
|
|
try {
|
|
const cache = join(root, 'cache')
|
|
vi.stubEnv('CODEBURN_CACHE_DIR', cache)
|
|
vi.stubEnv('APPDATA', root)
|
|
vi.stubEnv('XDG_CONFIG_HOME', root)
|
|
vi.stubEnv('CODEBURN_COPILOT_DISABLE_OTEL', '1')
|
|
vi.stubEnv('CODEBURN_COPILOT_SESSION_STATE_DIR', join(root, 'no-cli'))
|
|
vi.stubEnv('CODEBURN_COPILOT_SESSION_STORE_DB', join(root, 'no-store'))
|
|
vi.stubEnv('CODEBURN_COPILOT_JETBRAINS_DIR', join(root, 'no-jb'))
|
|
const chats = join(root, 'Code', 'User', 'globalStorage', 'emptyWindowChatSessions')
|
|
await mkdir(chats, { recursive: true })
|
|
const timestamp = new Date(Date.now() - 4 * 86400000).toISOString()
|
|
const date = toDateString(new Date(timestamp))
|
|
await writeFile(join(chats, 'redirected.jsonl'), JSON.stringify({ kind: 0, v: {
|
|
sessionId: 'redirected', creationDate: Date.parse(timestamp), requests: [{
|
|
requestId: 'r', modelId: 'copilot/gpt-4o', completionTokens: 20,
|
|
result: { metadata: { promptTokens: 100 } },
|
|
}],
|
|
} }) + '\n')
|
|
clearSessionCache()
|
|
const parse = () => parseAllSessions(undefined, 'copilot')
|
|
const fresh = await parse()
|
|
expect(aggregateProjectsIntoDays(fresh)[0]).toMatchObject({ date, calls: 1, inputTokens: 100, outputTokens: 20 })
|
|
|
|
const historical = structuredClone(fresh)
|
|
for (const project of historical) for (const session of project.sessions) for (const turn of session.turns) for (const call of turn.assistantCalls) {
|
|
call.provider = 'claude'
|
|
call.model = 'claude-sonnet-4-6'
|
|
call.usage.inputTokens = 50
|
|
call.usage.outputTokens = 10
|
|
call.costUSD = calculateCost(call.model, 50, 10, 0, 0, 0)
|
|
}
|
|
await mkdir(cache, { recursive: true })
|
|
await writeFile(join(cache, 'daily-cache.v50.json'), JSON.stringify({
|
|
version: 50, savingsConfigHash: '', tzKey: currentTzKey(),
|
|
lastComputedDate: toDateString(new Date(Date.now() - 86400000)),
|
|
days: aggregateProjectsIntoDays(historical), complete: true, watermarkTrusted: true,
|
|
}))
|
|
const hydrated = await ensureCacheHydrated(parse, aggregateProjectsIntoDays)
|
|
const day = hydrated.days.find(d => d.date === date)!
|
|
expect(day).toMatchObject({ calls: 2, inputTokens: 150, outputTokens: 30 })
|
|
expect(day.providers.copilot).toMatchObject({ calls: 1, inputTokens: 100, outputTokens: 20 })
|
|
expect(day.providers.claude).toMatchObject({ calls: 1, inputTokens: 50, outputTokens: 10 })
|
|
} finally {
|
|
clearSessionCache()
|
|
await rm(root, { recursive: true, force: true })
|
|
}
|
|
})
|