Add functional component tests

test/__snapshots__/functional.test.ts.snap

@@ -0,0 +1,19 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`Connected Components React Plugin - Functional ComponentWithChildren.jsx snippet creation 1`] = `
+Object {
+  "description": "",
+  "lang": "jsx",
+  "snippet": "<MyComponent>
+  {children}
+</MyComponent>",
+}
+`;
+
+exports[`Connected Components React Plugin - Functional FunctionalComponents.jsx snippet creation 1`] = `
+Object {
+  "description": "",
+  "lang": "jsx",
+  "snippet": "<MyComponent />",
+}
+`;

test/functional.test.ts

@@ -0,0 +1,29 @@
+import Plugin from "../src";
+
+describe("Connected Components React Plugin - Functional", () => {
+    test("FunctionalComponents.jsx snippet creation", async () => {
+        const plugin = new Plugin();
+
+        const componentCode = await plugin.process(
+            {
+                path: "test/samples/jsx-functional/FunctionalComponent.jsx",
+                zeplinNames: []
+            }
+        );
+
+        expect(componentCode).toMatchSnapshot();
+    });
+
+    test("ComponentWithChildren.jsx snippet creation", async () => {
+        const processor = new Plugin();
+
+        const componentCode = await processor.process(
+            {
+                path: "test/samples/jsx-functional/FunctionalComponentWithChildren.jsx",
+                zeplinNames: []
+            }
+        );
+
+        expect(componentCode).toMatchSnapshot();
+    });
+});

test/propType.test.ts

@@ -6,7 +6,7 @@ describe("Connected Components React Plugin - PropTypes", () => {
 
         const componentCode = await plugin.process(
             {
-                path: "test/samples/Component.jsx",
+                path: "test/samples/jsx-class/Component.jsx",
                 zeplinNames: []
             }
         );
@@ -19,7 +19,7 @@ describe("Connected Components React Plugin - PropTypes", () => {
 
         const componentCode = await processor.process(
             {
-                path: "test/samples/ComponentWithChildren.jsx",
+                path: "test/samples/jsx-class/ComponentWithChildren.jsx",
                 zeplinNames: []
             }
         );
@@ -32,7 +32,7 @@ describe("Connected Components React Plugin - PropTypes", () => {
 
         const componentCode = await processor.process(
             {
-                path: "test/samples/ComponentWithProps.jsx",
+                path: "test/samples/jsx-class/ComponentWithProps.jsx",
                 zeplinNames: []
             }
         );
@@ -45,7 +45,7 @@ describe("Connected Components React Plugin - PropTypes", () => {
 
         const componentCode = await processor.process(
             {
-                path: "test/samples/ComponentWithChildrenAndProps.jsx",
+                path: "test/samples/jsx-class/ComponentWithChildrenAndProps.jsx",
                 zeplinNames: []
             }
         );
@@ -58,7 +58,7 @@ describe("Connected Components React Plugin - PropTypes", () => {
 
         const componentCode = await processor.process(
             {
-                path: "test/samples/ComponentWithMemoization.jsx",
+                path: "test/samples/jsx-class/ComponentWithMemoization.jsx",
                 zeplinNames: []
             }
         );

test/samples/jsx-functional/FunctionalComponent.jsx

@@ -0,0 +1,17 @@
+import React from 'react';
+
+const MyComponent = () => {
+  const thing = {
+    stuff: 'stuff',
+  };
+
+  const string = thing?.stuff;
+
+  return (
+    <div>
+      {string}
+    </div>
+  );
+};
+
+export default MyComponent;

test/samples/jsx-functional/FunctionalComponentWithChildren.jsx

@@ -0,0 +1,21 @@
+import React from 'react';
+
+const MyComponent = () => {
+  const thing = {
+    stuff: 'stuff',
+  };
+
+  const string = thing?.stuff;
+
+  return (
+    <div>
+      {string}
+    </div>
+  );
+};
+
+MyComponent.propTypes = {
+  children: PropTypes.element.isRequired
+};
+
+export default MyComponent;