Ensure JSX elements under Program are transpiled

src/index.js

@@ -171,6 +171,13 @@ export default function ({ types: t, traverse: _traverse }) {
         },
 
         exit(path) {
+          path.traverse(elementVisitor, Object.assign({}, this, {
+            secondaryTree: true,
+            root: null,
+            replacedElements: new Set(),
+            closureVarsStack: [],
+          }));
+
           hoist(path, this);
           injectHelpers(this);
         }

test/fixtures/program/element/actual.js

@@ -0,0 +1 @@
+var div = <div></div>;

test/fixtures/program/element/expected.js

@@ -0,0 +1,12 @@
+var _jsxWrapper = function _jsxWrapper(func, args) {
+  var wrapper = args ? function wrapper() {
+    return func.apply(this, args);
+  } : func;
+  wrapper.__jsxDOMWrapper = true;
+  return wrapper;
+};
+
+var div = _jsxWrapper(function () {
+  elementOpen("div");
+  return elementClose("div");
+});

test/fixtures/program/expression-children/actual.js

@@ -0,0 +1 @@
+var div = <div key={i++}></div>;

test/fixtures/program/expression-children/expected.js

@@ -0,0 +1,12 @@
+var _jsxWrapper = function _jsxWrapper(func, args) {
+  var wrapper = args ? function wrapper() {
+    return func.apply(this, args);
+  } : func;
+  wrapper.__jsxDOMWrapper = true;
+  return wrapper;
+};
+
+var div = _jsxWrapper(function (_ref) {
+  elementOpen("div", _ref, ["key", _ref]);
+  return elementClose("div");
+}, [i++]);

test/fixtures/program/hoist/actual.js

@@ -0,0 +1 @@
+var div = <div key={i++}></div>;

test/fixtures/program/hoist/expected.js

@@ -0,0 +1,15 @@
+var _jsxWrapper = function _jsxWrapper(func, args) {
+  var wrapper = args ? function wrapper() {
+    return func.apply(this, args);
+  } : func;
+  wrapper.__jsxDOMWrapper = true;
+  return wrapper;
+};
+
+var _statics = ["key", ""],
+    _wrapper = function _wrapper(_ref) {
+  elementOpen("div", _ref, (_statics[1] = _ref, _statics));
+  return elementClose("div");
+};
+
+var div = _jsxWrapper(_wrapper, [i++]);

test/fixtures/program/hoist/options.json

@@ -0,0 +1,5 @@
+{
+  "options": {
+    "hoist": true
+  }
+}