test(transformer): 修改 evalClass

NervJS · Jul 4, 2018 · ef99f65 · ef99f65
1 parent a32661e
commit ef99f65
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 13 deletions.
diff --git a/packages/taro-transformer-wx/__tests__/template.spec.ts b/packages/taro-transformer-wx/__tests__/template.spec.ts
@@ -273,11 +273,13 @@ describe('Template', () => {
 
         expect(template).toMatch(prettyPrint(`
         <block>
-            <view class=\"page-body\" wx:if=\"{{tasks !== null}}\"></view>
-            <view class=\"page-body\" wx:else>
-                <text>Hello world!</text>
-            </view>
+        <block wx:if=\"{{tasks !== null}}\">
+            <view class=\"page-body\"></view>
         </block>
+        <view class=\"page-body\" wx:else>
+            <text>Hello world!</text>
+        </view>
+    </block>
         `))
       })
 
@@ -307,15 +309,19 @@ describe('Template', () => {
         })
 
         expect(template).toMatch(prettyPrint(`
-          <block>
-              <view class=\"page-body\" wx:if=\"{{tasks !== null}}\"></view>
-              <view class=\"page-body\" wx:elif=\"{{tasks.length === 0}}\">
-                  <text>{{tasks.length}}</text>
-              </view>
-              <view class=\"page-body\" wx:else>
-                  <text>Hello world!</text>
-              </view>
-          </block>
+        <block>
+        <block wx:if=\"{{tasks !== null}}\">
+            <view class=\"page-body\"></view>
+        </block>
+        <block wx:elif=\"{{tasks.length === 0}}\">
+            <view class=\"page-body\">
+                <text>{{tasks.length}}</text>
+            </view>
+        </block>
+        <view class=\"page-body\" wx:else>
+            <text>Hello world!</text>
+        </view>
+    </block>
         `))
       })
     })

diff --git a/packages/taro-transformer-wx/__tests__/utils.ts b/packages/taro-transformer-wx/__tests__/utils.ts
@@ -88,6 +88,16 @@ export function evalClass (ast: t.File, props = '') {
     // constructor 即便没有被定义也会被加上
     if (t.isClassMethod(method) && method.kind === 'constructor') {
       const index = method.body.body.findIndex(node => t.isSuper(node))
+      method.body.body.push(
+        t.expressionStatement(t.assignmentExpression(
+          '=',
+          t.memberExpression(
+            t.thisExpression(),
+            t.identifier('state')
+          ),
+          t.callExpression(t.memberExpression(t.thisExpression(), t.identifier('_createData')), [])
+        ))
+      )
       method.body.body.splice(index, 0, ...statements)
     }
   }