TypescriptAPIUtils.isClassInstanceOf: return true for same classes

Author: davidkudera

packages/typescript-api-utils/classes/instanceOf.ts

@@ -7,6 +7,10 @@ import * as ts from 'typescript';
 
 export function isClassInstanceOf(classDeclaration: ts.ClassDeclaration, parentClass: ts.ClassDeclaration, compilerOptions: ts.CompilerOptions, moduleResolutionHost: ts.ModuleResolutionHost): boolean
 {
+	if (classDeclaration === parentClass) {
+		return true;
+	}
+
 	let parentClassSourceFile: ts.SourceFile = undefined;
 	function getParentClassSourceFile(): ts.SourceFile
 	{

packages/typescript-api-utils/tests/data/classes/instanceOf/valid_5.ts

@@ -0,0 +1 @@
+export class A {}

packages/typescript-api-utils/tests/tests/classes/instanceOf.ts

@@ -61,6 +61,13 @@ describe('#classes/instanceOf', () => {
 			expect(isClassInstanceOf(classB, classA)).to.be.equal(true);
 		});
 
+		it('should return true when test class is same as testing class', () => {
+			const sourceFile = getSourceFile('valid_5');
+			const classA = findNode<ts.ClassDeclaration>(ts.SyntaxKind.ClassDeclaration, 'A', sourceFile);
+
+			expect(isClassInstanceOf(classA, classA)).to.be.equal(true);
+		});
+
 	});
 
 });