add some regression tests

ionic-team · Jul 6, 2023 · eadaac1 · eadaac1
1 parent 9f3e4b5
commit eadaac1
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 0 deletions.
diff --git a/src/compiler/transformers/test/lazy-component.spec.ts b/src/compiler/transformers/test/lazy-component.spec.ts
@@ -33,4 +33,33 @@ describe('lazy-component', () => {
     expect(t.outputText).toContain(`import { registerInstance as __stencil_registerInstance } from "@stencil/core"`);
     expect(t.outputText).toContain(`__stencil_registerInstance(this, hostRef)`);
   });
+
+  it('adds a getter for an @Element() reference', () => {
+    const compilerCtx = mockCompilerCtx();
+    const transformOpts: d.TransformOptions = {
+      coreImportPath: '@stencil/core',
+      componentExport: 'lazy',
+      componentMetadata: null,
+      currentDirectory: '/',
+      proxy: null,
+      style: 'static',
+      styleImportData: null,
+    };
+
+    const code = `
+      @Component({
+        tag: 'cmp-a'
+      })
+      export class CmpA {
+        @Element() el: HtmlElement;
+      }
+    `;
+
+    const transformer = lazyComponentTransform(compilerCtx, transformOpts);
+
+    const t = transpileModule(code, null, compilerCtx, [], [transformer]);
+
+    expect(t.outputText).toContain(`get el() { return __stencil_getElement(this); } };`);
+    expect(t.outputText).not.toContain(`el;`);
+  });
 });
diff --git a/src/compiler/transformers/test/native-constructor.spec.ts b/src/compiler/transformers/test/native-constructor.spec.ts
@@ -67,5 +67,23 @@ describe('nativeComponentTransform', () => {
       );
       expect(transpiledModule.outputText).toContain(`this.__attachShadow()`);
     });
+
+    it('adds a getter for an @Element() reference', () => {
+      const code = `
+        @Component({
+          tag: 'cmp-a'
+        })
+        export class CmpA {
+          @Element() el: HtmlElement;
+        }
+      `;
+
+      const transformer = nativeComponentTransform(compilerCtx, transformOpts);
+
+      const transpiledModule = transpileModule(code, null, compilerCtx, [], [transformer]);
+
+      expect(transpiledModule.outputText).toContain(`get el() { return this; }`);
+      expect(transpiledModule.outputText).not.toContain(`el;`);
+    });
   });
 });