66 lines
2.9 KiB
TypeScript
66 lines
2.9 KiB
TypeScript
import type { BlockView } from '../blockView'
|
|
import type { RuleReference } from '../contract'
|
|
import { describe, expect, it } from 'vitest'
|
|
import { toBlockViews } from '../blockView'
|
|
import { createPythonAdapter } from '../languages/python'
|
|
import { celeryRule } from '../rules/celery'
|
|
import { readFixture, WASM_DIR, withBlocks } from '../testSupport'
|
|
import { pythonZhTranslator } from '../translators/pythonZh'
|
|
import { DefinitionSources, referencesIn } from './definitionSources'
|
|
|
|
const py = createPythonAdapter(WASM_DIR)
|
|
const SNAPSHOT = readFixture('celeryApp.py')
|
|
|
|
function views(source: string): Promise<BlockView[]> {
|
|
return withBlocks(py, 'python', source, blocks => toBlockViews(blocks, { source, language: py, markers: [] }, pythonZhTranslator, { rules: [celeryRule], report: () => {} }))
|
|
}
|
|
|
|
/** 记下问了哪边：规则（引用的键）还是语言扩展（下标）。 */
|
|
function recorder(): { asked: string[], sources: DefinitionSources<string> } {
|
|
const asked: string[] = []
|
|
const sources = new DefinitionSources<string>({
|
|
rule: async (reference: RuleReference) => {
|
|
asked.push(`规则 ${reference.rule} ${reference.key}`)
|
|
return ['tasks.py']
|
|
},
|
|
language: async (offset: number) => {
|
|
asked.push(`语言扩展 ${offset}`)
|
|
return []
|
|
},
|
|
})
|
|
return { asked, sources }
|
|
}
|
|
|
|
describe('名字的定义去哪儿查', () => {
|
|
it('视图数据里规则标了引用的名字：下标 → 引用（任务表里的任务名）', async () => {
|
|
const references = referencesIn(await views(SNAPSHOT))
|
|
expect(references.size).toBe(12)
|
|
const offset = SNAPSHOT.indexOf('commerce.retry_failed_fulfillments",\n "schedule"')
|
|
expect(references.get(offset)).toEqual({ rule: 'celery', key: 'commerce.retry_failed_fulfillments' })
|
|
})
|
|
|
|
it('规则标了引用的问规则，其余（同一张卡片里的应用名、别的块里的名字）问语言扩展', async () => {
|
|
const blocks = await views(SNAPSHOT)
|
|
const { asked, sources } = recorder()
|
|
sources.remember(3, blocks)
|
|
const task = SNAPSHOT.indexOf('commerce.retry_failed_fulfillments",\n "schedule"')
|
|
const app = SNAPSHOT.indexOf('celery_app.conf.beat_schedule')
|
|
|
|
expect(await sources.locate(3, task)).toEqual(['tasks.py'])
|
|
expect(await sources.locate(3, app)).toEqual([])
|
|
expect(asked).toEqual([`规则 celery commerce.retry_failed_fulfillments`, `语言扩展 ${app}`])
|
|
})
|
|
|
|
it('引用只对记下它的那个文档版本有效；没记过文档时都问语言扩展', async () => {
|
|
const blocks = await views(SNAPSHOT)
|
|
const task = SNAPSHOT.indexOf('commerce.retry_failed_fulfillments",\n "schedule"')
|
|
const { asked, sources } = recorder()
|
|
await sources.locate(1, task)
|
|
sources.remember(3, blocks)
|
|
await sources.locate(4, task)
|
|
sources.remember(4, [])
|
|
await sources.locate(4, task)
|
|
expect(asked).toEqual([`语言扩展 ${task}`, `语言扩展 ${task}`, `语言扩展 ${task}`])
|
|
})
|
|
})
|