test: fix existing tests to be compatible with new features

Author: matheusdc

__tests__/pkg-usage.test.ts

@@ -37,7 +37,7 @@ describe('getPackagesUsages()', () => {
       expect(
         getPackagesUsages({
           packages: [pkg],
-          fileGlobs: `${MOCKS_DIR}/**.ts`,
+          fileGlobs: `${MOCKS_DIR}/**.tsx`,
           packageJsonCWD: MOCKS_DIR_CWD,
         })
       ).toStrictEqual([{ count: 0, files: [], name: pkg, version }]);
@@ -48,21 +48,20 @@ describe('getPackagesUsages()', () => {
     it('should return the right package usage', () => {
       const { fileName, imports, pkg, version } = mockPackageUsageFile();
 
-      expect(
-        getPackagesUsages({
-          packages: [pkg],
-          fileGlobs: `${MOCKS_DIR}/**.ts`,
-          packageJsonCWD: MOCKS_DIR_CWD,
-        })
-      ).toStrictEqual([
+      const result = getPackagesUsages({
+        packages: [pkg],
+        fileGlobs: `${MOCKS_DIR}/**.tsx`,
+        packageJsonCWD: MOCKS_DIR_CWD,
+      });
+
+      expect(result).toStrictEqual([
         {
           count: 1,
           files: [
             {
-              defaultImport: undefined,
-              filePath: `${MOCKS_DIR_CWD}/${fileName}.ts`,
-              name: `${fileName}.ts`,
-              namedImports: imports,
+              filePath: `${MOCKS_DIR_CWD}/${fileName}.tsx`,
+              name: `${fileName}.tsx`,
+              imports,
             },
           ],
           name: pkg,

__tests__/utils.ts

@@ -30,15 +30,89 @@ function mockUniqueList<T, U>(
     .filter((value, index, self) => self.indexOf(value) === index);
 }
 
+function word() {
+  return f.hacker.noun().replace(/ |-/g, '');
+}
+
+function capitalize(word: string) {
+  return `${word.charAt(0).toUpperCase()}${word.slice(1)}`;
+}
+
+function generatePropList(props: string[]) {
+  return props.map((prop) => `${prop}="${prop}"`).join(' ');
+}
+
+function jsxElement(value: string, line: number) {
+  const props = mockUniqueList(() => word().toLowerCase());
+  return [
+    {
+      name: value,
+      usages: [
+        {
+          line: line + 2,
+          props,
+          text: `<${value} ${generatePropList(props)}/>`,
+        },
+      ],
+    },
+    `function ${capitalize(word())}() { return (<${value} ${generatePropList(
+      props
+    )}/>)}`,
+  ];
+}
+
+function propertyAccess(value: string, line: number) {
+  const property = word();
+  return [
+    {
+      name: value,
+      usages: [{ line: line + 2, property, text: `\n${value}.${property}` }],
+    },
+    `${value}.${property}`,
+  ];
+}
+
+function callExpression(value: string, line: number) {
+  return [
+    { name: value, usages: [{ line: line + 2, text: `\n${value}()` }] },
+    `${value}()`,
+  ];
+}
+
+function valueUsage(value: string, line: number) {
+  return [
+    { name: value, usages: [{ line: line + 2, text: `\n${value};` }] },
+    `${value};`,
+  ];
+}
+
+const possibleUsages = [propertyAccess, callExpression, valueUsage, jsxElement];
+
+function mockTSFile(pkgName: string, imports: string[]) {
+  const importSection = `import { ${imports.join(', ')} } from '${pkgName}';`;
+
+  const usages = imports.map((entry, index) =>
+    possibleUsages[index % possibleUsages.length](entry, index)
+  );
+
+  const mockedImports = usages.map((data) => data[1]);
+
+  return {
+    file: [importSection, ...mockedImports].join('\n'),
+    imports: usages.map((data) => data[0]),
+  };
+}
+
 export function mockPackageUsageFile(customData?: string) {
-  const imports = mockUniqueList(() => f.hacker.noun().replace(/ |-/g, ''));
+  const importNames = mockUniqueList(() => capitalize(word()));
   const fileName = f.datatype.uuid();
   const pkg = f.hacker.noun();
   const version = f.system.semver();
 
+  const { file, imports } = mockTSFile(pkg, importNames);
   // Mocked with random data to generate a resilient test
-  const data = customData ?? `import { ${imports.join(', ')} } from '${pkg}';`;
-  writeFile(`${fileName}.ts`, data);
+  const data = customData ?? file;
+  writeFile(`${fileName}.tsx`, data);
 
   writeFile(
     'package.json',
@@ -55,7 +129,7 @@ export function mockPackageUsageFile(customData?: string) {
   );
 
   return {
-    imports,
+    imports: !customData ? imports : undefined,
     fileName,
     pkg,
     version,