From 8d479c93d28ce1cc973ef0820fff42c504e1f63f Mon Sep 17 00:00:00 2001
From: Alan Agius <alan.agius4@gmail.com>
Date: Wed, 5 Jun 2019 09:01:56 +0200
Subject: [PATCH] test: add test to verify that the new es2015 class wrapping
 logic handles wrapping of tslib and tsickle classes

Related to https://github.com/ngrx/platform/issues/1905 and https://github.com/ng-packagr/ng-packagr/issues/1307

Fixes #14613
---
 .../src/transforms/wrap-enums_spec.ts         | 34 +++++++++++++++++--
 1 file changed, 32 insertions(+), 2 deletions(-)

diff --git a/packages/angular_devkit/build_optimizer/src/transforms/wrap-enums_spec.ts b/packages/angular_devkit/build_optimizer/src/transforms/wrap-enums_spec.ts
index 7a82b4f886..c5cc64241e 100644
--- a/packages/angular_devkit/build_optimizer/src/transforms/wrap-enums_spec.ts
+++ b/packages/angular_devkit/build_optimizer/src/transforms/wrap-enums_spec.ts
@@ -14,9 +14,40 @@ import { getWrapEnumsTransformer } from './wrap-enums';
 const transform = (content: string) => transformJavascript(
   { content, getTransforms: [getWrapEnumsTransformer] }).content;
 
-// tslint:disable-next-line:no-big-function
+// tslint:disable:no-big-function
 describe('wrap enums and classes transformer', () => {
   describe('wraps class declarations', () => {
+    it('should wrap tsickle emitted classes which followed by metadata', () => {
+      const input = tags.stripIndent`
+       class CustomComponentEffects {
+            constructor(_actions) {
+              this._actions = _actions;
+              this.doThis = this._actions;
+            }
+        }
+        CustomComponentEffects.decorators = [{ type: Injectable }];
+        CustomComponentEffects.ctorParameters = () => [{ type: Actions }];
+        tslib_1.__decorate([
+            Effect(),
+            tslib_1.__metadata("design:type", Object)
+        ], CustomComponentEffects.prototype, "doThis", void 0);
+        tslib_1.__decorate([
+            Effect({ dispatch: false }),
+            tslib_1.__metadata("design:type", Object)
+        ], CustomComponentEffects.prototype, "doThat", void 0);
+      `;
+
+      const output = tags.stripIndent`
+        const CustomComponentEffects = /*@__PURE__*/ (() => {
+          ${input}
+
+          return CustomComponentEffects;
+        })();
+      `;
+
+      expect(tags.oneLine`${transform(input)}`).toEqual(tags.oneLine`${output}`);
+    });
+
     it('should ClassDeclarations that are referenced with in CallExpressions', () => {
       const input = tags.stripIndent`
         class ApplicationModule {
@@ -305,7 +336,6 @@ describe('wrap enums and classes transformer', () => {
     });
   });
 
-  // tslint:disable-next-line:no-big-function
   describe('wrap enums', () => {
     it('wraps ts 2.2 enums in IIFE', () => {
       const input = tags.stripIndent`