From b1ad7eae8a74aab1a0bb293c10aa248a0908fbea Mon Sep 17 00:00:00 2001
From: James Turley <james.turley@signalmedia.co>
Date: Mon, 22 Mar 2021 15:27:13 +0000
Subject: [PATCH] Tests for TaskPageLinks

Signed-off-by: James Turley <james.turley@signalmedia.co>
Signed-off-by: James Turley <jamesturley1905@googlemail.com>
---
 .../core/src/components/IconLink/IconLink.tsx |  1 -
 .../TaskPage/TaskPageLinks.test.tsx           | 76 +++++++++++++++++++
 2 files changed, 76 insertions(+), 1 deletion(-)
 create mode 100644 plugins/scaffolder/src/components/TaskPage/TaskPageLinks.test.tsx

diff --git a/packages/core/src/components/IconLink/IconLink.tsx b/packages/core/src/components/IconLink/IconLink.tsx
index c2f0a5c5da..d7f9e5571b 100644
--- a/packages/core/src/components/IconLink/IconLink.tsx
+++ b/packages/core/src/components/IconLink/IconLink.tsx
@@ -18,7 +18,6 @@ import React from 'react';
 import { IconComponent } from '@backstage/core-api';
 import { Grid, LinkProps, makeStyles, Typography } from '@material-ui/core';
 import LanguageIcon from '@material-ui/icons/Language';
-import { omit } from 'lodash';
 import { Link } from '../Link';
 
 const useStyles = makeStyles({
diff --git a/plugins/scaffolder/src/components/TaskPage/TaskPageLinks.test.tsx b/plugins/scaffolder/src/components/TaskPage/TaskPageLinks.test.tsx
new file mode 100644
index 0000000000..5ea170eae2
--- /dev/null
+++ b/plugins/scaffolder/src/components/TaskPage/TaskPageLinks.test.tsx
@@ -0,0 +1,76 @@
+/*
+ * Copyright 2020 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 React from 'react';
+import { TaskPageLinks } from './TaskPageLinks';
+import { renderInTestApp } from '@backstage/test-utils';
+
+describe('TaskPageLinks', () => {
+  beforeEach(() => {});
+
+  afterEach(() => {
+    jest.resetAllMocks();
+  });
+
+  it('renders the entityRef link', async () => {
+    const output = { entityRef: 'Component:default/my-app' };
+    const { findByText } = await renderInTestApp(
+      <TaskPageLinks output={output} />,
+    );
+
+    const element = await findByText('Open in catalog');
+
+    expect(element).toBeInTheDocument();
+    expect(element).toHaveAttribute(
+      'href',
+      '/catalog/default/Component/my-app',
+    );
+  });
+
+  it('renders the remoteUrl link', async () => {
+    const output = { remoteUrl: 'https://remote.url' };
+    const { findByText } = await renderInTestApp(
+      <TaskPageLinks output={output} />,
+    );
+
+    const element = await findByText('Repo');
+
+    expect(element).toBeInTheDocument();
+    expect(element).toHaveAttribute('href', 'https://remote.url');
+  });
+
+  it('renders further links', async () => {
+    const output = {
+      links: [
+        { url: 'https://first.url', title: 'Cool link 1' },
+        { url: 'https://second.url', title: 'Cool link 2' },
+      ],
+    };
+    const { findByText } = await renderInTestApp(
+      <TaskPageLinks output={output} />,
+    );
+
+    let element = await findByText('Cool link 1');
+
+    expect(element).toBeInTheDocument();
+    expect(element).toHaveAttribute('href', 'https://first.url');
+
+    element = await findByText('Cool link 2');
+
+    expect(element).toBeInTheDocument();
+    expect(element).toHaveAttribute('href', 'https://second.url');
+  });
+});