fix(test-utils): Support re-export declarations

fixtures/test-utils/errors-no-wrapper-classes/tsconfig.json

@@ -1,4 +1,3 @@
 {
-  "extends": "../tsconfig.json",
-  "include": ["errors-no-wrapper-classes.ts"]
+  "extends": "../tsconfig.json"
 }

fixtures/test-utils/exports/alert.ts

@@ -0,0 +1,10 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+import { ElementWrapper } from './core';
+
+export class AlertWrapper extends ElementWrapper {
+  findContent() {
+    return new ElementWrapper();
+  }
+}

fixtures/test-utils/exports/button.ts

@@ -0,0 +1,10 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+import { ElementWrapper } from './core';
+
+export class ButtonWrapper extends ElementWrapper {
+  findText() {
+    return new ElementWrapper();
+  }
+}

fixtures/test-utils/exports/core.ts

@@ -0,0 +1,3 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+export class ElementWrapper {}

fixtures/test-utils/exports/index.ts

@@ -0,0 +1,4 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+export { AlertWrapper } from './alert';
+export { ButtonWrapper } from './button';

fixtures/test-utils/exports/tsconfig.json

@@ -0,0 +1,3 @@
+{
+  "extends": "../tsconfig.json"
+}

src/components/type-utils.ts

@@ -107,3 +107,10 @@ export function extractDeclaration(symbol: ts.Symbol) {
   }
   return declarations[0];
 }
+
+export function printFlags(flags: number, mapping: Record<number, string>) {
+  return Object.entries(mapping)
+    .filter(([key, value]) => Number(key) & flags)
+    .map(([key, value]) => value)
+    .join(' | ');
+}

src/test-utils-new/extractor.ts

@@ -43,11 +43,11 @@ export default function extractDocumentation(
     if (extraExports.includes(className)) {
       continue;
     }
-    if (!(symbol.flags & ts.SymbolFlags.Class)) {
+    const classType = checker.getDeclaredTypeOfSymbol(symbol);
+    if (!classType.isClass()) {
       throw new Error(`Exported symbol is not a class, got ${checker.symbolToString(symbol)}`);
     }
 
-    const classType = checker.getTypeAtLocation(extractDeclaration(symbol));
     const classDefinition: TestUtilsDoc = { name: className, methods: [] };
     for (const property of classType.getProperties()) {
       const declaration = property.valueDeclaration;

test/components/type-utils.test.ts

@@ -0,0 +1,21 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+import ts from 'typescript';
+import { test, expect } from 'vitest';
+import { printFlags } from '../../lib/components/type-utils';
+
+test('serialises node flags', () => {
+  expect(
+    printFlags(
+      ts.NodeFlags.HasAsyncFunctions + ts.NodeFlags.HasImplicitReturn + ts.NodeFlags.ContainsThis,
+      ts.NodeFlags
+    )
+  ).toEqual('ContainsThis | HasImplicitReturn | ReachabilityCheckFlags | HasAsyncFunctions | ReachabilityAndEmitFlags');
+});
+
+test('serialises type flags', () => {
+  expect(printFlags(ts.TypeFlags.Number + ts.TypeFlags.Enum, ts.TypeFlags)).toEqual(
+    'Number | Enum | NumberLike | EnumLike | PossiblyFalsy | Primitive | NotPrimitiveUnion | Singleton | Intrinsic | IncludesMask | DisjointDomains | DefinitelyNonNullable | Narrowable'
+  );
+});

test/test-utils/doc-generation.test.ts

@@ -83,6 +83,15 @@ describe('Generate documentation', () => {
     expect(childClassMethod?.inheritedFrom).toBeUndefined();
   });
 
+  test('deal with re-exports', () => {
+    const results = buildTestUtilsProject('exports');
+    expect(results.map(classDoc => classDoc.name)).toEqual(['AlertWrapper', 'ButtonWrapper']);
+    const alertWrapper = results.find(classDoc => classDoc.name === 'AlertWrapper')!;
+    expect(alertWrapper.methods.map(method => method.name)).toEqual(['findContent']);
+    const buttonWrapper = results.find(classDoc => classDoc.name === 'ButtonWrapper')!;
+    expect(buttonWrapper.methods.map(method => method.name)).toEqual(['findText']);
+  });
+
   test('default value rendering', () => {
     const results = buildTestUtilsProject('default-values');
     expect(results.length).toBe(1);

test/test-utils/usage.test.ts

@@ -16,6 +16,12 @@ describe('documentTestUtils throws error for ', () => {
     expect(() => buildTestUtilsProject('errors-empty')).toThrow('Failed to parse tsconfig.json');
   });
 
+  test('throws error on unknown export', () => {
+    expect(() => buildTestUtilsProject('errors-no-wrapper-classes')).toThrow(
+      'Exported symbol is not a class, got speakTruth'
+    );
+  });
+
   test('having no input files because of a non-matching glob', () => {
     expect(() =>
       buildTestUtilsProject('simple', {