additional tests

Author: barelyhuman

test/snapshots/test.ts.md

@@ -200,3 +200,30 @@ Generated by [AVA](https://avajs.dev).
         onClick: updateUser␊
       }, "Click Me"));␊
     }`
+
+## Mix and Match State
+
+> Snapshot 1
+
+    `import * as React from "react";␊
+    ␊
+    function App() {␊
+      const [users, setUsers] = React.useState([{␊
+        name: "reaper"␊
+      }]);␊
+      const [updateCount, setUpdateCount] = React.useState(1);␊
+    ␊
+      const updateUser = () => {␊
+        const _nextUsers = users.slice();␊
+    ␊
+        _nextUsers[0].name = "barelyhuman";␊
+        setUser(_nextUsers);␊
+        setUpdateCount(updateCount + 1);␊
+      };␊
+    ␊
+      return /*#__PURE__*/React.createElement(React.Fragment, null, users.map(user => {␊
+        return /*#__PURE__*/React.createElement("p", null, user.name);␊
+      }), /*#__PURE__*/React.createElement("p", null, updateCount), /*#__PURE__*/React.createElement("button", {␊
+        onClick: updateUser␊
+      }, "Click Me"));␊
+    }`

test/snapshots/test.ts.snap

@@ -1,6 +1,6 @@
 import test from 'ava'
 import {transform} from '@babel/core'
-import plugin from '../'
+import plugin from '../source/index'
 
 const compile = (code: string) =>
   transform(code, {
@@ -253,3 +253,37 @@ test('Array Update', (t) => {
   }
   t.snapshot(result.code)
 })
+
+test('Mix and Match State', (t) => {
+  const code = `
+  import * as React from "react";
+  
+  function App() {
+    let $users = [{ name: "reaper" }];
+    const [updateCount,setUpdateCount] = React.useState(1);
+
+    const updateUser = () => {
+      const _nextUsers =$users.slice();
+      _nextUsers[0].name = "barelyhuman"
+      $user = _nextUsers;
+      setUpdateCount(updateCount+1)
+    };
+
+    return (
+      <>
+      {$users.map(user=>{
+        return <p>{user.name}</p>
+      })}
+      <p>{updateCount}</p>
+      <button onClick={updateUser}>Click Me</button>
+      </>
+    );
+  }
+  `
+
+  const result = compile(code)
+  if (!result) {
+    return t.fail()
+  }
+  t.snapshot(result.code)
+})