add tests for getTypeArgumentParamsFromDeclaration & getNativeComponentType fn's

Author: Pranav-yadav

packages/react-native-codegen/src/parsers/__tests__/parsers-test.js

@@ -165,6 +165,97 @@ describe('FlowParser', () => {
       ).toEqual(expected);
     });
   });
+
+  describe('getTypeArgumentParamsFromDeclaration', () => {
+    it('returns type arguments params from declaration', () => {
+      const declaration = {
+        type: 'TypeAlias',
+        typeArguments: {
+          type: 'TypeParameterDeclaration',
+          params: [
+            {
+              type: 'TypeParameter',
+              name: 'T',
+            },
+          ],
+        },
+      };
+
+      const expected = [
+        {
+          type: 'TypeParameter',
+          name: 'T',
+        },
+      ];
+
+      expect(parser.getTypeArgumentParamsFromDeclaration(declaration)).toEqual(
+        expected,
+      );
+    });
+
+    it('returns undefined if declaration type argument params is Invalid', () => {
+      const declaration = {
+        type: 'TypeAlias',
+        typeArguments: {
+          type: 'TypeParameterDeclaration',
+        },
+      };
+
+      expect(parser.getTypeArgumentParamsFromDeclaration(declaration)).toEqual(
+        undefined,
+      );
+    });
+  });
+
+  describe('getNativeComponentType', () => {
+    it('returns native component type when typeArgumentParams & funcArgumentParams are valid', () => {
+      const typeArgumentParams = [
+        {
+          type: 'TypeParameter',
+          name: 'T',
+          id: {
+            name: 'T',
+          },
+        },
+      ];
+
+      const funcArgumentParams = [
+        {
+          type: 'StringLiteral',
+          value: 'componentName',
+        },
+      ];
+
+      const expected = {
+        propsTypeName: 'T' /* typeArgumentParams[0].id.name */,
+        componentName: 'componentName' /* funcArgumentParams[0].value */,
+      };
+
+      expect(
+        parser.getNativeComponentType(typeArgumentParams, funcArgumentParams),
+      ).toEqual(expected);
+    });
+
+    it('returns undefined when typeArgumentParams & funcArgumentParams are invalid', () => {
+      const typeArgumentParams = [
+        {
+          type: 'TypeParameter',
+          name: 'T',
+          id: {},
+        },
+      ];
+      const funcArgumentParams = [{}];
+
+      const expected = {
+        propsTypeName: undefined /* typeArgumentParams[0].id.name */,
+        componentName: undefined /* funcArgumentParams[0].value */,
+      };
+
+      expect(
+        parser.getNativeComponentType(typeArgumentParams, funcArgumentParams),
+      ).toEqual(expected);
+    });
+  });
 });
 
 describe('TypeScriptParser', () => {
@@ -326,4 +417,92 @@ describe('TypeScriptParser', () => {
       ).toEqual(expected);
     });
   });
+
+  describe('getTypeArgumentParamsFromDeclaration', () => {
+    it('returns type argument params from declaration', () => {
+      const declaration = {
+        type: 'TypeAlias',
+        typeParameters: {
+          type: 'TypeParameterDeclaration',
+          params: [
+            {
+              type: 'TypeParameter',
+              name: 'T',
+            },
+          ],
+        },
+      };
+
+      const expected = [
+        {
+          type: 'TypeParameter',
+          name: 'T',
+        },
+      ];
+
+      expect(parser.getTypeArgumentParamsFromDeclaration(declaration)).toEqual(
+        expected,
+      );
+    });
+
+    it('returns undefined if declaration type arguments params is Invalid', () => {
+      const declaration = {
+        type: 'TypeAlias',
+        typeParameters: {},
+      };
+
+      expect(parser.getTypeArgumentParamsFromDeclaration(declaration)).toEqual(
+        undefined,
+      );
+    });
+  });
+
+  describe('getNativeComponentType', () => {
+    it('returns native component type when typeArgumentParams & funcArgumentParams are valid', () => {
+      const typeArgumentParams = [
+        {
+          typeName: {
+            type: 'Identifier',
+            name: 'T',
+          },
+        },
+      ];
+
+      const funcArgumentParams = [
+        {
+          type: 'ObjectTypeAnnotation',
+          value: 'StringTypeAnnotation',
+        },
+      ];
+
+      const expected = {
+        propsTypeName: 'T' /* typeArgumentParams[0].typeName.name */,
+        componentName: 'StringTypeAnnotation' /* funcArgumentParams[0].value */,
+      };
+
+      expect(
+        parser.getNativeComponentType(typeArgumentParams, funcArgumentParams),
+      ).toEqual(expected);
+    });
+
+    it('returns undefined when typeArgumentParams & funcArgumentParams are invalid', () => {
+      const typeArgumentParams = [
+        {
+          typeName: {
+            type: 'Invalid',
+          },
+        },
+      ];
+      const funcArgumentParams = [{}];
+
+      const expected = {
+        propsTypeName: undefined /* typeArgumentParams[0].typeName.name */,
+        componentName: undefined /* funcArgumentParams[0].value */,
+      };
+
+      expect(
+        parser.getNativeComponentType(typeArgumentParams, funcArgumentParams),
+      ).toEqual(expected);
+    });
+  });
 });