Fix removing unrelated decorators from class

Author: simonihmig

lib/__tests__/__snapshots__/transform.js.snap

@@ -547,6 +547,38 @@ foo
 =========="
 `;
 
+exports[`native components keeps unrelated decorators in place 1`] = `
+"==========
+
+      import Component from '@ember/component';
+      import { classNames, layout } from '@ember-decorators/component';
+
+      @layout(template)
+      @classNames('foo')
+      export default class FooComponent extends Component {
+      }
+    
+~~~~~~~~~~
+foo
+~~~~~~~~~~
+ => tagName: div
+~~~~~~~~~~
+
+      import Component from '@ember/component';
+      import { layout, tagName } from '@ember-decorators/component';
+
+      @tagName(\\"\\")
+      @layout(template)
+      export default class FooComponent extends Component {
+      }
+    
+~~~~~~~~~~
+<div class=\\"foo\\" ...attributes>
+  foo
+</div>
+=========="
+`;
+
 exports[`native components multi-line template 1`] = `
 "==========
 

lib/__tests__/transform.js

@@ -452,6 +452,22 @@ describe('native components', function() {
     );
   });
 
+  test('keeps unrelated decorators in place', () => {
+    let source = `
+      import Component from '@ember/component';
+      import { classNames, layout } from '@ember-decorators/component';
+
+      @layout(template)
+      @classNames('foo')
+      export default class FooComponent extends Component {
+      }
+    `;
+
+    let template = `foo`;
+
+    expect(generateSnapshot(source, template)).toMatchSnapshot();
+  });
+
   test('skips tagless components', () => {
     let source = `
       import Component from '@ember/component';

lib/transform/native.js

@@ -120,7 +120,7 @@ module.exports = function transformNativeComponent(root, options) {
   debug('classNameBindings: %o', classNameBindings);
 
   // set `@tagName('')`
-  addClassDecorator(exportDefaultDeclaration, 'tagName', [j.stringLiteral('')]);
+  addClassDecorator(classDeclaration, 'tagName', [j.stringLiteral('')]);
   ensureImport(root, 'tagName', '@ember-decorators/component');
 
   // remove `elementId`, `attributeBindings`, `classNames` and `classNameBindings`