From 1b1ddb0c4d99c7f0b7a935cb3d64a5ffe385eb8c Mon Sep 17 00:00:00 2001
From: Patrik Oldsberg <poldsberg@gmail.com>
Date: Wed, 2 Jun 2021 10:54:20 +0200
Subject: [PATCH] codemods: add tests for core-imports

Signed-off-by: Patrik Oldsberg <poldsberg@gmail.com>
---
 .../codemods/src/tests/core-imports.test.ts   | 68 +++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 packages/codemods/src/tests/core-imports.test.ts

diff --git a/packages/codemods/src/tests/core-imports.test.ts b/packages/codemods/src/tests/core-imports.test.ts
new file mode 100644
index 0000000000..90bc217d3e
--- /dev/null
+++ b/packages/codemods/src/tests/core-imports.test.ts
@@ -0,0 +1,68 @@
+/*
+ * Copyright 2021 Spotify AB
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import jscodeshift, { API } from 'jscodeshift';
+import coreImportTransform from '../../transforms/core-imports';
+
+function runTransform(source: string) {
+  const j = jscodeshift.withParser('tsx');
+  const api: API = {
+    j,
+    jscodeshift: j,
+    stats: () => undefined,
+    report: () => undefined,
+  };
+  return coreImportTransform({ source, file: 'test.ts' }, api);
+}
+
+describe('core-imports', () => {
+  it('should leave file unchanged', () => {
+    const input = `
+import something from 'somewhere';
+
+function nothing() {
+return something()
+}
+`;
+
+    expect(runTransform(input)).toBe(input);
+  });
+
+  it('should refactor imports', () => {
+    const input = `
+/* COPYRIGHT: ME */
+import { Button as MyButton, createApiRef, createApp } from '@backstage/core';
+
+const app = createApp();
+const apiRef = createApiRef();
+const button = <MyButton />
+`;
+
+    const output = `
+/* COPYRIGHT: ME */
+import { Button as MyButton } from '@backstage/core-components';
+
+import { createApiRef } from '@backstage/core-plugin-api';
+import { createApp } from '@backstage/core-app-api';
+
+const app = createApp();
+const apiRef = createApiRef();
+const button = <MyButton />
+`;
+
+    expect(runTransform(input)).toBe(output);
+  });
+});