1 files changed, 214 insertions, 0 deletions
diff --git a/@linaria/packages/babel/__tests__/evaluators/extractor.test.ts b/@linaria/packages/babel/__tests__/evaluators/extractor.test.ts
new file mode 100644
index 0000000..b764a25
--- /dev/null
+++ b/@linaria/packages/babel/__tests__/evaluators/extractor.test.ts
@@ -0,0 +1,214 @@
+import path from 'path';
+import dedent from 'dedent';
+import type { TransformOptions } from '@babel/core';
+import exctract from '../../../extractor/src';
+
+function getFileName() {
+  return path.resolve(__dirname, `../__fixtures__/test.js`);
+}
+
+function _shake(opts?: TransformOptions, only: string[] = ['__linariaPreval']) {
+  return (
+    literal: TemplateStringsArray,
+    ...placeholders: string[]
+  ): [string, Map<string, string[]>] => {
+    const code = dedent(literal, ...placeholders);
+    const [shaken, deps] = exctract(
+      getFileName(),
+      {
+        babelOptions: opts || {},
+        displayName: true,
+        evaluate: true,
+        rules: [
+          {
+            action: require('../../../extractor/src').default,
+          },
+          {
+            test: /\/node_modules\//,
+            action: 'ignore',
+          },
+        ],
+      },
+      code,
+      only
+    );
+
+    return [shaken, deps!];
+  };
+}
+
+it('removes all', () => {
+  const [shaken] = _shake()`
+    const { whiteColor: color, anotherColor } = require('…');
+    const a = color || anotherColor;
+    color.green = '#0f0';
+
+    exports.__linariaPreval = [];
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});
+
+it('keeps only code which is related to `color`', () => {
+  const [shaken] = _shake()`
+    const { whiteColor: color, anotherColor } = require('…');
+    const wrap = '';
+    const a = color || anotherColor;
+    color.green = '#0f0';
+    module.exports = { color, anotherColor };
+    exports.__linariaPreval = [color];
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});
+
+it('keeps only code which is related to `anotherColor`', () => {
+  const [shaken] = _shake()`
+    const { whiteColor: color, anotherColor } = require('…');
+    const a = color || anotherColor;
+    color.green = '#0f0';
+    exports.__linariaPreval = [anotherColor];
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});
+
+it('keeps only code which is related to `a`', () => {
+  const [shaken] = _shake()`
+    const { whiteColor: color, anotherColor } = require('…');
+    const a = color || anotherColor;
+    color.green = '#0f0';
+    exports.__linariaPreval = [a];
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});
+
+it('shakes imports', () => {
+  const [shaken] = _shake()`
+    import { unrelatedImport } from '…';
+    import { whiteColor as color, anotherColor } from '…';
+    import defaultColor from '…';
+    import anotherDefaultColor from '…';
+    import '…';
+    require('…');
+    export default color;
+    exports.__linariaPreval = [color, defaultColor];
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});
+
+it('should keep member expression key', () => {
+  const [shaken] = _shake()`
+    const key = 'blue';
+    const obj = { blue: '#00F' };
+    const blue = obj[key];
+    exports.__linariaPreval = [blue];
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});
+
+it('shakes exports', () => {
+  const [shaken] = _shake()`
+    import { whiteColor as color, anotherColor } from '…';
+    export const a = color;
+    export { redColor } from "…";
+    export { anotherColor };
+    exports.__linariaPreval = [a];
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});
+
+it('shakes es5 exports', () => {
+  const [shaken] = _shake(undefined, ['redColor', 'greenColor'])`
+    "use strict";
+    Object.defineProperty(exports, "__esModule", {
+      value: true
+    });
+    exports.redColor = 'red';
+    Object.defineProperty(exports, "blueColor", {
+      enumerable: true,
+      get: function get() {
+        return 'blue';
+      }
+    });
+    Object.defineProperty(exports, "greenColor", {
+      enumerable: true,
+      get: function get() {
+        return 'green';
+      }
+    });
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});
+
+// TODO: this test will be disabled until the shaker is fully implemented
+// eslint-disable-next-line jest/no-disabled-tests
+it.skip('should throw away any side effects', () => {
+  const [shaken] = _shake()`
+    const objects = { key: { fontSize: 12 } };
+    const foo = (k) => {
+      const obj = objects[k];
+      console.log('side effect');
+      return obj;
+    };
+    exports.__linariaPreval = [foo];
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});
+
+it('keeps objects as is', () => {
+  const [shaken] = _shake()`
+    const fill1 = (top = 0, left = 0, right = 0, bottom = 0) => ({
+      position: 'absolute',
+      top,
+      right,
+      bottom,
+      left,
+    });
+
+    const fill2 = (top = 0, left = 0, right = 0, bottom = 0) => {
+      return {
+        position: 'absolute',
+        top,
+        right,
+        bottom,
+        left,
+      };
+    };
+
+    exports.__linariaPreval = [fill1, fill2];
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});
+
+it('shakes sequence expression', () => {
+  const [shaken] = _shake()`
+    import { external } from '…';
+    const color1 = (external, () => 'blue');
+    let local = '';
+    const color2 = (local = color1(), () => local);
+    exports.__linariaPreval = [color2];
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});
+
+it('shakes assignment patterns', () => {
+  const [shaken] = _shake()`
+    const [identifier = 1] = [2];
+    const [{...object} = {}] = [{ a: 1, b: 2 }];
+    const [[...array] = []] = [[1,2,3,4]];
+    const obj = { member: null };
+    ([obj.member = 42] = [1]);
+    exports.__linariaPreval = [identifier, object, array, obj];
+  `;
+
+  expect(shaken).toMatchSnapshot();
+});