tests: add for scope based and multiple components

Author: barelyhuman

test/snapshots/test.ts.md

@@ -59,3 +59,23 @@ Generated by [AVA](https://avajs.dev).
         onClick: onPress␊
       }, "Press"));␊
     }`
+
+## Check Arrow Function Scope
+
+> Snapshot 1
+
+    `import * as React from "react";␊
+    let $b = 2;␊
+    ␊
+    const Component = () => {␊
+      const [a, setA] = React.useState(1);␊
+    ␊
+      const onPress = () => {␊
+        setA(a + 1);␊
+        $b = 3;␊
+      };␊
+    ␊
+      return /*#__PURE__*/React.createElement("div", null, /*#__PURE__*/React.createElement("p", null, a), /*#__PURE__*/React.createElement("button", {␊
+        onClick: onPress␊
+      }, "Press"));␊
+    };`

test/snapshots/test.ts.snap

@@ -59,3 +59,79 @@ test('Check Functional Scope', (t) => {
 	}
 	t.snapshot(result.code)
 })
+
+test('Check Arrow Function Scope', (t) => {
+	const code = `
+    import * as React from "react";
+
+    let $b = 2;
+    
+    const Component = () => {
+      let $a = 1;
+    
+      const onPress = () => {
+        $a += 1;
+        $b = 3;
+      };
+    
+      return (
+        <div>
+          <p>{$a}</p>
+          <button onClick={onPress}>Press</button>
+        </div>
+      );
+    };
+    `
+
+	const result = compile(code)
+	if (!result) {
+		return t.fail()
+	}
+	t.snapshot(result.code)
+})
+
+test('Multi Component Scope', (t) => {
+	const code = `
+import * as React from "react";
+
+let $b = 2;
+
+const Component = () => {
+  let $a = 1;
+
+  const onPress = () => {
+    $a += 1;
+    $b = 3;
+  };
+
+  return (
+    <div>
+      <p>{$a}</p>
+      <button onClick={onPress}>Press</button>
+    </div>
+  );
+};
+
+const ComponentTwo = () => {
+  let $a = 3;
+
+  const onPress = () => {
+    $a = 5;
+    $b = 3;
+  };
+
+  return (
+    <div>
+      <p>{$a}</p>
+      <button onClick={onPress}>Press</button>
+    </div>
+  );
+};
+`
+
+	const result = compile(code)
+	if (!result) {
+		return t.fail()
+	}
+	t.snapshot(result.code)
+})