Marko v3: Added tests for onBeforeGenerateCode and onAfterGenerateCode

test/fixtures/codegen/autotest/code-before-after-if/expected.js

@@ -0,0 +1,6 @@
+if (a > b) {
+  before();
+  out.w("<div class=\"greeting\">Hello World</div>");
+
+  after();
+}

test/fixtures/codegen/autotest/code-before-after-if/index.js

@@ -0,0 +1,29 @@
+'use strict';
+
+module.exports = function(builder) {
+    var htmlElement = builder.htmlElement(
+        'div',
+        [
+            {
+                name: 'class',
+                value: builder.literal('greeting')
+            }
+        ],
+        [
+                builder.text(builder.literal('Hello World'))
+        ]);
+
+    htmlElement.onBeforeGenerateCode((event) => {
+        event.insertCode(builder.functionCall('before', []));
+    });
+
+    htmlElement.onAfterGenerateCode((event) => {
+        event.insertCode(builder.functionCall('after', []));
+    });
+
+    var ifStatement = builder.ifStatement('a > b', [
+        htmlElement
+    ]);
+
+    return ifStatement;
+};

test/fixtures/codegen/autotest/code-before-after-surrounded/expected.js

@@ -0,0 +1,9 @@
+if (a > b) {
+  var before;
+
+  before();
+  var foo;
+  after();
+
+  var after;
+}

test/fixtures/codegen/autotest/code-before-after-surrounded/index.js

@@ -0,0 +1,21 @@
+'use strict';
+
+module.exports = function(builder) {
+    var vars = builder.vars([ 'foo' ]);
+
+    vars.onBeforeGenerateCode((event) => {
+        event.insertCode(builder.functionCall('before', []));
+    });
+
+    vars.onAfterGenerateCode((event) => {
+        event.insertCode(builder.functionCall('after', []));
+    });
+
+    var ifStatement = builder.ifStatement('a > b', [
+        builder.vars([ 'before' ]),
+        vars,
+        builder.vars([ 'after' ])
+    ]);
+
+    return ifStatement;
+};

test/fixtures/codegen/autotest/code-before-after-var/expected.js

@@ -0,0 +1,5 @@
+if (a > b) {
+  before();
+  var foo;
+  after();
+}

test/fixtures/codegen/autotest/code-before-after-var/index.js

@@ -0,0 +1,19 @@
+'use strict';
+
+module.exports = function(builder) {
+    var vars = builder.vars([ 'foo' ]);
+
+    vars.onBeforeGenerateCode((event) => {
+        event.insertCode(builder.functionCall('before', []));
+    });
+
+    vars.onAfterGenerateCode((event) => {
+        event.insertCode(builder.functionCall('after', []));
+    });
+
+    var ifStatement = builder.ifStatement('a > b', [
+        vars
+    ]);
+
+    return ifStatement;
+};

test/fixtures/codegen/autotest/code-before-after/expected.js

@@ -0,0 +1,4 @@
+before();
+out.w("<div class=\"greeting\">Hello World</div>");
+
+after();

test/fixtures/codegen/autotest/code-before-after/index.js

@@ -0,0 +1,25 @@
+'use strict';
+
+module.exports = function(builder) {
+    var htmlElement = builder.htmlElement(
+        'div',
+        [
+            {
+                name: 'class',
+                value: builder.literal('greeting')
+            }
+        ],
+        [
+                builder.text(builder.literal('Hello World'))
+        ]);
+
+    htmlElement.onBeforeGenerateCode((event) => {
+        event.insertCode(builder.functionCall('before', []));
+    });
+
+    htmlElement.onAfterGenerateCode((event) => {
+        event.insertCode(builder.functionCall('after', []));
+    });
+
+    return htmlElement;
+};