feat: add tests for return type partial type searching

stanimirovv · Nov 5, 2023 · 9792707 · 9792707
1 parent 21caa41
commit 9792707
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/src/evaluator.ts b/src/evaluator.ts
@@ -160,6 +160,7 @@ function removeDynamicImports (inputStr: string): string {
   return inputStr.replace(importRegex, '')
 }
 
+// TODO move to their own file ?
 // showcase how we might add partial type checking
 function doTypesMatch (searchParameterType: string, functionParameter: ParameterDeclaration): boolean {
   const mandatoryObjectKeys = parseTypeQuery(searchParameterType)

diff --git a/tests/evaluator.spec.ts b/tests/evaluator.spec.ts
@@ -43,7 +43,7 @@ describe('getMatchingFunctions everything defined', () => {
 
   it('should correctly parse ":?...,string"', () => {
     const result = evaluateSearchQuery('testproject.tsconfig.json', { kind: 'both', returnTypes: [''], parameterTypes: [['...'], ['string']] })
-    expect(result.length).toEqual(3)
+    expect(result.length).toEqual(4)
   })
 
   it('should correctly parse ":?{a&b}"', () => {
@@ -60,4 +60,9 @@ describe('getMatchingFunctions everything defined', () => {
     const result = evaluateSearchQuery('testproject.tsconfig.json', { kind: 'both', returnTypes: [''], parameterTypes: [['{a}']] })
     expect(result.length).toEqual(1)
   })
+
+  it('should correctly parse ":{a}?"', () => {
+    const result = evaluateSearchQuery('testproject.tsconfig.json', { kind: 'both', returnTypes: ['{a}'], parameterTypes: [] })
+    expect(result.length).toEqual(1)
+  })
 })
diff --git a/tests/testProject/class.ts b/tests/testProject/class.ts
@@ -12,4 +12,8 @@ export class Foo {
   public asd (a: TestType): string {
     return '42'
   }
+
+  public tt1 (a: string, ...b: string[]): TestType {
+    return { a: '42', b: '42' }
+  }
 }