clean up class decorators #1185

test/core/fixtures/transformation/es7.decorators/class-modules/expected.js

@@ -9,11 +9,12 @@ var _foo = require("foo");
 var _foo2 = babelHelpers.interopRequireWildcard(_foo);
 
 var Foo = (function () {
+  var _Foo = Foo;
+
   function Foo() {
     babelHelpers.classCallCheck(this, _Foo);
   }
 
-  var _Foo = Foo;
   Foo = _foo2["default"](Foo) || Foo;
   return Foo;
 })();

test/core/fixtures/transformation/es7.decorators/class-super/actual.js

@@ -0,0 +1,12 @@
+@bar
+class Foo extends Bar {
+  constructor() {
+    super();
+  }
+}
+
+var Foo2 = @bar class extends Bar {
+  constructor() {
+    super();
+  }
+};

test/core/fixtures/transformation/es7.decorators/class-super/expected.js

@@ -0,0 +1,28 @@
+"use strict";
+
+var Foo = (function (_Bar) {
+  var _Foo = Foo;
+
+  function Foo() {
+    babelHelpers.classCallCheck(this, _Foo);
+
+    babelHelpers.get(Object.getPrototypeOf(_Foo.prototype), "constructor", this).call(this);
+  }
+
+  babelHelpers.inherits(Foo, _Bar);
+  Foo = bar(Foo) || Foo;
+  return Foo;
+})(Bar);
+
+var Foo2 = (function (_Bar2) {
+  var _class = function Foo2() {
+    babelHelpers.classCallCheck(this, _class2);
+
+    babelHelpers.get(Object.getPrototypeOf(_class2.prototype), "constructor", this).call(this);
+  };
+
+  var _class2 = _class;
+  babelHelpers.inherits(_class, _Bar2);
+  _class = bar(_class) || _class;
+  return _class;
+})(Bar);

test/core/fixtures/transformation/es7.decorators/class/expected.js

@@ -1,55 +1,60 @@
 "use strict";
 
 var Foo = (function () {
+  var _Foo = Foo;
+
   function Foo() {
     babelHelpers.classCallCheck(this, _Foo);
   }
 
-  var _Foo = Foo;
   Foo = foo(Foo) || Foo;
   return Foo;
 })();
 
 var Bar = (function () {
+  var _Bar = Bar;
+
   function Bar() {
     babelHelpers.classCallCheck(this, _Bar);
   }
 
-  var _Bar = Bar;
   Bar = bar(Bar) || Bar;
   Bar = foo(Bar) || Bar;
   return Bar;
 })();
 
 var Foo2 = (function () {
+  var _Foo2 = Foo;
+
   function Foo() {
     babelHelpers.classCallCheck(this, _Foo2);
   }
 
-  var _Foo2 = Foo;
   Foo = bar(Foo) || Foo;
   return Foo;
 })();
 
 var Bar2 = (function () {
+  var _Bar2 = Bar;
+
   function Bar() {
     babelHelpers.classCallCheck(this, _Bar2);
   }
 
-  var _Bar2 = Bar;
   Bar = bar(Bar) || Bar;
   Bar = foo(Bar) || Bar;
   return Bar;
 })();
 
 var Baz = (function () {
+  var _Baz = Baz;
+
   function Baz(baz) {
     babelHelpers.classCallCheck(this, _Baz);
 
     this.baz = baz;
   }
 
-  var _Baz = Baz;
   Baz = foo(Baz) || Baz;
   return Baz;
 })();