chore(handler): add more tests for search and highlights

chemzqm · chemzqm · commit 3874573dfb69 · 2021-06-30T21:18:15.000+08:00
diff --git a/src/__tests__/handler/highlights.test.ts b/src/__tests__/handler/highlights.test.ts
@@ -2,6 +2,7 @@ import { Neovim } from '@chemzqm/neovim'
 import { Disposable, Position, Range } from 'vscode-languageserver-protocol'
 import Highlights from '../../handler/highlights'
 import languages from '../../languages'
+import workspace from '../../workspace'
 import { disposeAll } from '../../util'
 import helper from '../helper'
 
@@ -106,7 +107,21 @@ describe('document highlights', () => {
     expect(res).toBeNull()
   })
 
-  it('should not throw when document is not valid', async () => {
+  it('should not throw when document is command line', async () => {
+    await nvim.call('feedkeys', ['q:', 'in'])
+    let doc = await workspace.document
+    expect(doc.isCommandLine).toBe(true)
+    let err
+    try {
+      await highlights.highlight()
+    } catch (e) {
+      err = e
+    }
+    await nvim.input('<C-c>')
+    expect(err).toBeUndefined()
+  })
+
+  it('should not throw when provide not found', async () => {
     disposeAll(disposables)
     await helper.createDocument()
     await nvim.setLine('  oo')
diff --git a/src/__tests__/handler/search.test.ts b/src/__tests__/handler/search.test.ts
@@ -1,6 +1,6 @@
 import { Neovim } from '@chemzqm/neovim'
 import Refactor from '../../handler/refactor'
-import Search from '../../handler/search'
+import Search, { getPathFromArgs } from '../../handler/search'
 import helper from '../helper'
 import path from 'path'
 
@@ -25,17 +25,39 @@ afterEach(async () => {
   await helper.reset()
 })
 
+describe('getPathFromArgs', () => {
+  it('should get undefined path', async () => {
+    let res = getPathFromArgs(['a'])
+    expect(res).toBeUndefined()
+    res = getPathFromArgs(['a', 'b', '-c'])
+    expect(res).toBeUndefined()
+    res = getPathFromArgs(['a', '-b', 'c'])
+    expect(res).toBeUndefined()
+  })
+})
+
 describe('search', () => {
 
   it('should open refactor window', async () => {
     let search = new Search(nvim, cmd)
     let buf = await refactor.createRefactorBuffer()
     await search.run([], cwd, buf)
+    await helper.wait(50)
     let fileItems = buf.fileItems
     expect(fileItems.length).toBe(2)
     expect(fileItems[0].ranges.length).toBe(2)
   })
 
+  it('should abort task', async () => {
+    let search = new Search(nvim, cmd)
+    let buf = await refactor.createRefactorBuffer()
+    let p = search.run(['--sleep', '1000'], cwd, buf)
+    search.abort()
+    await p
+    let fileItems = buf.fileItems
+    expect(fileItems.length).toBe(0)
+  })
+
   it('should work with CocAction search', async () => {
     await helper.doAction('search', ['CocAction'])
     let bufnr = await nvim.call('bufnr', ['%'])
diff --git a/src/__tests__/rg b/src/__tests__/rg
@@ -27,4 +27,12 @@ let content = `\x1b[30mmodules/cursors.test.ts\x1b[m
 \x1b[32m46\x1b[m-    expect(res).toEqual({ line: 0, character: 3 })
 `
 
-process.stdout.write(content)
+let idx = process.argv.findIndex(s => s == '--sleep')
+if (idx !== -1) {
+  let ms = process.argv[idx + 1]
+  setTimeout(() => {
+    process.stdout.write(content)
+  }, ms)
+} else {
+  process.stdout.write(content)
+}
diff --git a/src/handler/search.ts b/src/handler/search.ts
@@ -185,10 +185,14 @@ export default class Search {
       })
     })
   }
+
+  public abort(): void {
+    this.task?.dispose()
+  }
 }
 
 // rg requires `-- [path]` at the end
-function getPathFromArgs(args: string[]): string | undefined {
+export function getPathFromArgs(args: string[]): string | undefined {
   if (args.length < 2) return undefined
   let len = args.length
   if (args[len - 1].startsWith('-')) return undefined
